Modify configuration file parsing

NicoG96 · Sep 17, 2019 · dba16ee · dba16ee
1 parent caaf75f
commit dba16ee
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/demeter/demeter.py b/demeter/demeter.py
@@ -2,6 +2,7 @@
 from pyfiglet import Figlet
 from git import Repo, Git
 from github import Github
+import configparser
 import os.path
 import logging
 import github
@@ -175,8 +176,9 @@ def cherrypick(pull_requests, release_name):
 
 
 if __name__ == "__main__":
-    if not os.path.isfile("../config.py"):
-        config = open("../config.py", "w")
+    config = configparser.ConfigParser()
+
+    if not os.path.isfile("config.ini"):
         print(colored("Please enter your personal GitHub access token:\t", "yellow"))
         GITHUB_TOKEN = input()
         print(colored("Please enter the repository as it appears on Github(e.g. {User}/{Repository}:\t", "yellow"))
@@ -185,15 +187,19 @@ def cherrypick(pull_requests, release_name):
                       "\t", "yellow"))
         REPO_PATH = input()
 
-        config.write('GITHUB_TOKEN = "' + GITHUB_TOKEN + '"\n')
-        config.write('GITHUB_REPO = "' + GITHUB_REPO + '"\n')
-        config.write('REPO_PATH = r"' + REPO_PATH + '"\n')
-        config.close()
+        config['CREDENTIALS'] = {
+            'GITHUB_TOKEN': GITHUB_TOKEN,
+            'GITHUB_REPO': GITHUB_REPO,
+            'REPO_PATH': REPO_PATH
+        }
+
+        with open('config.ini', 'w') as settings:
+            config.write(settings)
 
-    import config
-    git = Git(config.REPO_PATH)
-    repo = Repo(config.REPO_PATH)
-    g = Github(config.GITHUB_TOKEN)
-    r = g.get_repo(config.GITHUB_REPO)
+    config.read('config.ini')
+    git = Git(config.get('CREDENTIALS', 'repo_path'))
+    repo = Repo(config.get('CREDENTIALS', 'repo_path'))
+    g = Github(config.get('CREDENTIALS', 'github_token'))
+    r = g.get_repo(config.get('CREDENTIALS', 'github_repo'))
 
     demeter_cli()