set-up API key config

Justin Wong · Justin Wong · commit 213d41cb21ec · 2018-08-30T17:08:28.000-07:00
diff --git a/setup.py b/setup.py
@@ -0,0 +1,38 @@
+import os
+import getpass
+import base64
+import time
+
+
+CONFIG_FILE_PATH      = "config.py"
+
+CONFIG_FILE_TEMPLATE = """credentials = dict(api_key='{0}') """
+
+
+# get hashed credentials 
+try:
+    api_key = base64.b64encode(raw_input("enter/paste API Key: ").encode()).decode()
+except:
+    api_key = base64.b64encode(input("enter/paste API Key: ").encode()).decode()
+print("   Hashed: {}\n".format(api_key))
+
+new_config = CONFIG_FILE_TEMPLATE.format(api_key)
+
+# check if config file exists
+if not os.path.isfile(CONFIG_FILE_PATH):
+    # create new config file
+    with open(CONFIG_FILE_PATH, "w") as config_file:
+        config_file.write(new_config)
+        print("{} created successfully".format(CONFIG_FILE_PATH))
+else:
+    with open(CONFIG_FILE_PATH, "r") as config_file:
+        cur_config = config_file.read()
+    if new_config != cur_config:
+        # update config file
+        with open(CONFIG_FILE_PATH, "w") as config_file:
+            config_file.writelines(new_config)
+        print("{} updated successfully".format(CONFIG_FILE_PATH))
+    else:
+        print("{} already contains latest credentials".format(CONFIG_FILE_PATH))
+time.sleep(2)
+
