Prechádzať zdrojové kódy

Improve error if db is wrongly passed

Oz N Tiram 6 rokov pred
rodič
commit
eb229364cc
1 zmenil súbory, kde vykonal 5 pridanie a 0 odobranie
  1. 5 0
      pwman/data/factory.py

+ 5 - 0
pwman/data/factory.py

@@ -62,6 +62,11 @@ def check_db_version(dburi):
 
     dburi = urlparse(dburi)
     dbtype = dburi.scheme
+
+    if not dbtype:
+        print("Your URI seems incorrect ...")
+        sys.exit(0)
+
     try:
         cls = getattr(drivers, class_db_map[dbtype][0])
         ver = cls.check_db_version(class_db_map[dbtype][1](dburi))