Added argparse for query command line calls, support default language and query text - #5

Author: davidmezzetti

src/python/codequestion/query.py

@@ -2,10 +2,10 @@
 Query module
 """
 
+import argparse
 import os
 import os.path
 import sqlite3
-import sys
 
 import html2text
 import mdv
@@ -182,14 +182,33 @@ def close(db):
         db.close()
 
     @staticmethod
-    def run(query):
+    def args():
         """
-        Executes a query against a codequestion index.
+        Parses command line arguments.
 
-        Args:
-            query: input query
+        Returns:
+            command line arguments
         """
 
+        parser = argparse.ArgumentParser(description="codequestion query")
+
+        parser.add_argument("-l", "--lang", help="Sets default language")
+        parser.add_argument("-q", "--query", required=True, help="Query to execute")
+
+        return parser.parse_args()
+
+    @staticmethod
+    def run():
+        """
+        Executes a query against a codequestion index.
+        """
+
+        # Parse command line arguments
+        args = Query.args()
+
+        # Build query
+        query = "%s %s" % (args.lang, args.query) if args.lang else args.query
+
         # Load model
         embeddings, db = Query.load()
 
@@ -200,5 +219,4 @@ def run(query):
         Query.close(db)
 
 if __name__ == "__main__":
-    if len(sys.argv) > 1:
-        Query.run(sys.argv[1])
+    Query.run()