dmart 1.4.20__py3-none-any.whl → 1.4.22__py3-none-any.whl

{dmart-1.4.20.dist-info → dmart-1.4.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dmart
-Version: 1.4.20
+Version: 1.4.22
 Requires-Python: >=3.11
 Requires-Dist: fastapi
 Requires-Dist: pydantic

{dmart-1.4.20.dist-info → dmart-1.4.22.dist-info}/RECORD

@@ -1,6 +1,6 @@
 bundler.py,sha256=MDTUTVS0WL7gb-i1rDcS4i37KTqaW88TluL4L3rwjRw,1817
 data_generator.py,sha256=CnE-VHEeX7-lAXtqCgbRqR9WHjTuOgeiZcviYrHAmho,2287
-dmart.py,sha256=mG4TCkITYoUajlHsPJ_BZUBMu2G8gxvB7bidYfKoSHA,24498
+dmart.py,sha256=T63eUgBZXSWYIi6GRaRBdHQvnZ4HGam46naY0gTlI4c,24705
 get_settings.py,sha256=Sbe2WCoiK398E7HY4SNLfDN_GmE8knR4M-YJWF31jcg,153
 info.json,sha256=hXQWl19lfMkEj_zXdehGeKjiKGNJ7emY4S7d4pIqJ1E,123
 main.py,sha256=KZGhIL6AnEm5ZAPy4IvhBDpzSTjuodilV7NafNOyhzM,19676
@@ -272,8 +272,8 @@ utils/ticket_sys_utils.py,sha256=9QAlW2iiy8KyxQRBDj_WmzS5kKb0aYJmGwd4qzmGVqo,700
 utils/web_notifier.py,sha256=QM87VVid2grC5lK3NdS1yzz0z1wXljr4GChJOeK86W4,843
 utils/templates/activation.html.j2,sha256=XAMKCdoqONoc4ZQucD0yV-Pg5DlHHASZrTVItNS-iBE,640
 utils/templates/reminder.html.j2,sha256=aoS8bTs56q4hjAZKsb0jV9c-PIURBELuBOpT_qPZNVU,639
-dmart-1.4.20.dist-info/METADATA,sha256=MTGSY6tBlgUmFtLj4RP40CIZbMsFGbMGG2Jo41hv_hY,2149
-dmart-1.4.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dmart-1.4.20.dist-info/entry_points.txt,sha256=GjfoGh1bpxuU9HHGJzbtCFPNptHv9TryxHMN3uBSKpg,37
-dmart-1.4.20.dist-info/top_level.txt,sha256=S-gfX1pLerapNXiHZ8lvPYoV7sgwSX2_NCZ6xfzDUHM,267
-dmart-1.4.20.dist-info/RECORD,,
+dmart-1.4.22.dist-info/METADATA,sha256=Et0n5lf20vuttHiL3AOs-psMuNhC883LbbQQGZ0yMTE,2149
+dmart-1.4.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dmart-1.4.22.dist-info/entry_points.txt,sha256=GjfoGh1bpxuU9HHGJzbtCFPNptHv9TryxHMN3uBSKpg,37
+dmart-1.4.22.dist-info/top_level.txt,sha256=S-gfX1pLerapNXiHZ8lvPYoV7sgwSX2_NCZ6xfzDUHM,267
+dmart-1.4.22.dist-info/RECORD,,

dmart.py

@@ -18,10 +18,9 @@ from pathlib import Path
 from hypercorn.config import Config
 from hypercorn.run import run
 
+# Try to import alembic
 try:
-    # Try to import alembic
     import alembic
-    
     # Check if we are importing the local alembic directory which is likely a leftover
     if hasattr(alembic, '__path__'):
         local_alembic_path = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'alembic'))
@@ -38,7 +37,10 @@ try:
                         import alembic
                     except Exception:
                         pass
+except ImportError:
+    pass
 
+try:
     from alembic import command as alembic_command
     from alembic.config import Config as AlembicConfig
     HAS_ALEMBIC = True
@@ -412,27 +414,29 @@ def hypercorn_main() -> int:
 
 
 def main():
-    sys.argv = sys.argv[1:]
-    if len(sys.argv) == 0:
+    global sys
+    
+    args = sys.argv[1:]
+    if len(args) == 0:
         print("You must provide a command to run:")
         print(commands)
         sys.exit(1)
 
-    match sys.argv[0]:
+    match args[0]:
         case "hyper":
             hypercorn_main()
         case "server" | "serve":
             open_cxb = False
-            if "--open-cxb" in sys.argv:
+            if "--open-cxb" in args:
                 open_cxb = True
-                sys.argv.remove("--open-cxb")
+                args.remove("--open-cxb")
             
-            if "--cxb-config" in sys.argv:
-                idx = sys.argv.index("--cxb-config")
-                if idx + 1 < len(sys.argv):
-                    os.environ["DMART_CXB_CONFIG"] = sys.argv[idx + 1]
-                    sys.argv.pop(idx + 1)
-                sys.argv.pop(idx)
+            if "--cxb-config" in args:
+                idx = args.index("--cxb-config")
+                if idx + 1 < len(args):
+                    os.environ["DMART_CXB_CONFIG"] = args[idx + 1]
+                    args.pop(idx + 1)
+                args.pop(idx)
                 
             if open_cxb:
                 url = f"http://{settings.listening_host}:{settings.listening_port}/cxb/"
@@ -443,6 +447,7 @@ def main():
                 import threading
                 threading.Thread(target=open_browser, daemon=True).start()
 
+            sys.argv = [sys.argv[0]] + args
             asyncio.run(server())
         case "health-check":
             parser = argparse.ArgumentParser(
@@ -453,9 +458,9 @@ def main():
             parser.add_argument("-s", "--space", help="hit the target space or pass (all) to make the full health check")
             parser.add_argument("-m", "--schemas", nargs="*", help="hit the target schema inside the space")
 
-            args = parser.parse_args()
+            args_parsed = parser.parse_args(args[1:])
             before_time = time.time()
-            asyncio.run(health_check(args.type, args.space, args.schemas))
+            asyncio.run(health_check(args_parsed.type, args_parsed.space, args_parsed.schemas))
             print(f'total time: {"{:.2f}".format(time.time() - before_time)} sec')
         case "create-index":
             parser = argparse.ArgumentParser(
@@ -473,9 +478,9 @@ def main():
                 "--flushall", action='store_true', help="FLUSHALL data on Redis"
             )
 
-            args = parser.parse_args()
+            args_parsed = parser.parse_args(args[1:])
 
-            asyncio.run(create_index(args.space, args.schemas, args.subpaths, args.flushall))
+            asyncio.run(create_index(args_parsed.space, args_parsed.schemas, args_parsed.subpaths, args_parsed.flushall))
         case "export":
             parser = argparse.ArgumentParser()
             parser.add_argument(
@@ -492,24 +497,24 @@ def main():
                 "--since",
                 help="Export entries created/updated since the provided timestamp",
             )
-            args = parser.parse_args()
+            args_parsed = parser.parse_args(args[1:])
             since = None
             output_path = ""
-            if args.output:
-                output_path = args.output
+            if args_parsed.output:
+                output_path = args_parsed.output
 
-            if args.since:
-                since = int(round(float(args.since) * 1000))
+            if args_parsed.since:
+                since = int(round(float(args_parsed.since) * 1000))
 
-            if not os.path.isdir(args.spaces):
-                exit_with_error(f"The spaces folder {args.spaces} is not found.")
+            if not os.path.isdir(args_parsed.spaces):
+                exit_with_error(f"The spaces folder {args_parsed.spaces} is not found.")
 
             out_path = os.path.join(output_path, OUTPUT_FOLDER_NAME)
             if os.path.isdir(out_path):
                 shutil.rmtree(out_path)
 
             tasks = []
-            with open(args.config, "r") as f:
+            with open(args_parsed.config, "r") as f:
                 config_objs = json.load(f)
 
             for config_obj in config_objs:
@@ -521,7 +526,7 @@ def main():
                                      config_obj.get("schema_shortname", ""),
                                      config_obj.get("included_meta_fields", {}),
                                      config_obj.get("excluded_payload_fields", {}),
-                                     args.spaces, output_path, since))
+                                     args_parsed.spaces, output_path, since))
 
             asyncio.run(exporter(tasks))
 
@@ -550,11 +555,11 @@ def main():
                 help="The number of day, older than which, the entries will be archived (based on updated_at)",
             )
 
-            args = parser.parse_args()
-            space = args.space
-            subpath = args.subpath
-            olderthan = args.olderthan
-            schema = args.schema or "meta"
+            args_parsed = parser.parse_args(args[1:])
+            space = args_parsed.space
+            subpath = args_parsed.subpath
+            olderthan = args_parsed.olderthan
+            schema = args_parsed.schema or "meta"
 
             asyncio.run(archive(space, subpath, schema, olderthan))
             print("Done.")
@@ -585,7 +590,6 @@ def main():
                     print("Error: 'alembic' library not found. Please install it with 'pip install alembic'.")
                     if ALEMBIC_ERROR:
                         print(f"Details: {ALEMBIC_ERROR}")
-                    import sys
                     print(f"Python executable: {sys.executable}")
                     print(f"sys.path: {sys.path}")
                     sys.exit(1)