Add a config loader and add output option

Author: lnishan

SQLGitHub.py

@@ -1,16 +1,10 @@
 #!/usr/bin/env python
 
+import config_loader
 from components import top_level
 
-try:
-    import config
-except ImportError:
-    token = None
-else:
-    token = config.token
-
 
 if __name__ == "__main__":
-    token = token or raw_input("Please enter your GitHub token (which can be obtained from https://github.com/settings/tokens):")
-    sqlserv = top_level.SQLGitHub(token)
+    token, output = config_loader.Load("config")
+    sqlserv = top_level.SQLGitHub(token, output)
     sqlserv.Start()

components/top_level.py

@@ -20,8 +20,9 @@ class SQLGitHub:
 
     _PROMPT_STR = u"SQLGitHub> "
 
-    def __init__(self, token):
+    def __init__(self, token, output="str"):
         self._github = Github(token)
+        self._output = output
         self._parser = parser.SgParser(self._github)
         self._completer = WordCompleter(definition.ALL_TOKENS,
                                         ignore_case=True)
@@ -45,7 +46,10 @@ def Execute(self, sql, measure_time=True):
             except AttributeError:
                 sys.stderr.write("One or more of the specified fields doesn't exist.\n")
             else:
-                print(result)
+                if self._output == "str":
+                    print(result)
+                elif self._output == "csv":
+                    print(result.InCsv())
                 print("-")
                 print("Total rows: %d" % (len(result)))
                 if measure_time:

config_loader.py

@@ -0,0 +1,19 @@
+"""Config loader for SQLGitHub."""
+
+import importlib
+
+
+def Load(module):
+    try:
+        mod = importlib.import_module(module)
+    except ImportError:
+        token = None
+        output = "str"
+    else:
+        token = mod.token if hasattr(mod, "token") else None
+        output = mod.output if hasattr(mod, "output") else "str"
+    
+    token = token or raw_input("Please enter your GitHub token (which can be obtained from https://github.com/settings/tokens): ")
+
+    return (token,
+            output)