Просмотр исходного кода

Merge pull request #63 from pjw91/fix-59

Fixes #59
Cammy 5 лет назад
Родитель
Сommit
4d3156a44b
1 измененных файлов с 11 добавлено и 11 удалено
  1. 11 11
      pyinstalive/startup.py

+ 11 - 11
pyinstalive/startup.py

@@ -32,12 +32,17 @@ def validate_inputs(config, args, unknown_args):
     error_arr = []
     try:
         if args.configpath:
-            pil.config_path = args.configpath
-            if not os.path.isfile(pil.config_path):
-                pil.config_path = os.path.join(os.getcwd(), "pyinstalive.ini")
-                logger.warn("Custom config path is invalid, falling back to default path: {:s}".format(pil.config_path))
-                logger.separator()
-                
+            if os.path.isfile(args.configpath):
+                pil.config_path = args.configpath
+            else:
+                if os.path.isfile(pil.config_path):
+                    logger.warn("Custom config path is invalid, falling back to default path: {:s}".format(pil.config_path))
+                    logger.separator()
+                else:  # Create new config if it doesn't exist
+                    logger.banner()
+                    helpers.new_config()
+                    return False
+        pil.config_path = os.path.realpath(pil.config_path)
         config.read(pil.config_path)
 
         if args.download:
@@ -311,11 +316,6 @@ def run():
 
     args, unknown_args = parser.parse_known_args()  # Parse arguments
 
-    if not os.path.exists(pil.config_path):  # Create new config if it doesn't exist
-        logger.banner()
-        helpers.new_config()
-        return
-
     if validate_inputs(config, args, unknown_args):
         if not args.username and not args.password:
             pil.ig_api = auth.authenticate(username=pil.ig_user, password=pil.ig_pass)