dmart 1.4.40.post19__py3-none-any.whl → 1.4.40.post21__py3-none-any.whl

dmart/dmart.py

@@ -17,12 +17,10 @@ from pathlib import Path
 
 
 script_dir = str(Path(__file__).resolve().parent)
-if script_dir not in sys.path:
-    sys.path.insert(0, script_dir)
 
 from hypercorn.config import Config
 from hypercorn.run import run
-from utils.settings import settings
+from dmart.utils.settings import settings
 
 
 commands = """
@@ -48,7 +46,8 @@ commands = """
 sentinel = object()
 
 
-def hypercorn_main() -> int:
+def hypercorn_main(args_list: list[str]) -> int:
+    from dmart.utils.settings import settings
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "application",
@@ -254,13 +253,17 @@ def hypercorn_main() -> int:
         default=sentinel,
         type=int,
     )
-    args = parser.parse_args(sys.argv[1:])
+    args = parser.parse_args(args_list)
     
     if args.config == "hypercorn_config.toml" and not os.path.exists(args.config):
         config = Config()
         config.backlog = 2000
         config.workers = 1
         config.bind = ["localhost:8282"]
+        # Allow child processes to find the application
+        script_dir = os.path.dirname(os.path.abspath(__file__))
+        if script_dir not in sys.path:
+            sys.path.insert(0, script_dir)
     else:
         config = Config.from_toml(args.config)
         
@@ -409,23 +412,25 @@ def print_formatted(data):
 
 
 def main():
-    sys.argv = sys.argv[1:]
-    if len(sys.argv) == 0:
+    if len(sys.argv) < 2:
         print("You must provide a command to run:")
         print(commands)
         sys.exit(1)
 
-    match sys.argv[0]:
+    command = sys.argv[1]
+    args_list = sys.argv[2:]
+
+    match command:
         case "hyper":
-            hypercorn_main()
+            hypercorn_main(args_list)
         case "cli":
             config_file = None
-            if "--config" in sys.argv:
-                idx = sys.argv.index("--config")
-                if idx + 1 < len(sys.argv):
-                    config_file = sys.argv[idx + 1]
-                    sys.argv.pop(idx + 1)
-                    sys.argv.pop(idx)
+            if "--config" in args_list:
+                idx = args_list.index("--config")
+                if idx + 1 < len(args_list):
+                    config_file = args_list[idx + 1]
+                    args_list.pop(idx + 1)
+                    args_list.pop(idx)
 
             if not config_file:
                 if os.path.exists("cli.ini"):
@@ -483,10 +488,7 @@ def main():
             
             last_import_error = None
             try:
-                dmart_dir = Path(__file__).resolve().parent
-                if str(dmart_dir) not in sys.path:
-                    sys.path.append(str(dmart_dir))
-                import cli # type: ignore
+                from dmart import cli # type: ignore
                 cli.main()
                 return
             except ImportError as e:
@@ -520,17 +522,19 @@ def main():
                  print("Error: cli.py not found.")
             sys.exit(1)
         case "serve":
+            from dmart.utils.settings import settings
+            from dmart.main import main as server
             open_cxb = False
-            if "--open-cxb" in sys.argv:
+            if "--open-cxb" in args_list:
                 open_cxb = True
-                sys.argv.remove("--open-cxb")
+                args_list.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_list:
+                idx = args_list.index("--cxb-config")
+                if idx + 1 < len(args_list):
+                    os.environ["DMART_CXB_CONFIG"] = args_list[idx + 1]
+                    args_list.pop(idx + 1)
+                args_list.pop(idx)
                 
             if open_cxb:
                 host = settings.listening_host
@@ -544,10 +548,9 @@ def main():
                 import threading
                 threading.Thread(target=open_browser, daemon=True).start()
 
-            from main import main as server
             asyncio.run(server())
         case "health-check":
-            from data_adapters.file.health_check import main as health_check
+            from dmart.data_adapters.file.health_check import main as health_check
             parser = argparse.ArgumentParser(
                 description="This created for doing health check functionality",
                 formatter_class=argparse.ArgumentDefaultsHelpFormatter,
@@ -561,7 +564,7 @@ def main():
             asyncio.run(health_check(args.type, args.space, args.schemas))
             print(f'total time: {"{:.2f}".format(time.time() - before_time)} sec')
         case "create-index":
-            from data_adapters.file.create_index import main as create_index
+            from dmart.data_adapters.file.create_index import main as create_index
             parser = argparse.ArgumentParser(
                 description="Recreate Redis indices based on the available schema definitions",
                 formatter_class=argparse.ArgumentDefaultsHelpFormatter,
@@ -581,7 +584,7 @@ def main():
 
             asyncio.run(create_index(args.space, args.schemas, args.subpaths, args.flushall))
         case "export":
-            from utils.exporter import main as exporter, exit_with_error, OUTPUT_FOLDER_NAME, validate_config, extract
+            from dmart.utils.exporter import main as exporter, exit_with_error, OUTPUT_FOLDER_NAME, validate_config, extract
             parser = argparse.ArgumentParser()
             parser.add_argument(
                 "--config", required=True, help="Json config relative path from the script"
@@ -634,11 +637,12 @@ def main():
                 f"Output path: {os.path.abspath(os.path.join(output_path, OUTPUT_FOLDER_NAME))}"
             )
         case "settings":
+            from dmart.utils.settings import settings
             print_formatted(settings.model_dump_json())
         case "set_password":
-            import set_admin_passwd  # noqa: F401
+            from dmart import set_admin_passwd  # noqa: F401
         case "archive":
-            from data_adapters.file.archive import archive
+            from dmart.data_adapters.file.archive import archive
             parser = argparse.ArgumentParser(
                 description="Script for archiving records from different spaces and subpaths."
             )
@@ -665,10 +669,10 @@ def main():
             asyncio.run(archive(space, subpath, schema, olderthan))
             print("Done.")
         case "json_to_db":
-            from data_adapters.sql.json_to_db_migration import main as json_to_db_migration
+            from dmart.data_adapters.sql.json_to_db_migration import main as json_to_db_migration
             asyncio.run(json_to_db_migration())
         case "db_to_json":
-            from data_adapters.sql.db_to_json_migration import main as db_to_json_migration
+            from dmart.data_adapters.sql.db_to_json_migration import main as db_to_json_migration
             db_to_json_migration()
         case "help":
             print("Available commands:")
@@ -739,6 +743,7 @@ def main():
             import configparser
             import tempfile
             
+            from dmart.utils.settings import settings
             dmart_root = Path(__file__).resolve().parent
             alembic_ini_path = dmart_root / "alembic.ini"
             
@@ -764,7 +769,7 @@ def main():
                     config.write(temp_config_file)
                     temp_config_path = temp_config_file.name
 
-                alembic_cli_args = sys.argv[1:]
+                alembic_cli_args = args_list
                 if not alembic_cli_args:
                     alembic_cli_args = ["upgrade", "head"]
 
@@ -819,4 +824,10 @@ def main():
                 sys.exit(e.returncode)
 
 if __name__ == "__main__":
+    # If installed as a package, we need to ensure the parent of dmart package
+    # is in sys.path for child processes to find 'dmart.dmart'
+    dmart_pkg_parent = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    if dmart_pkg_parent not in sys.path:
+        sys.path.append(dmart_pkg_parent)
+
     main()

{dmart-1.4.40.post19.dist-info → dmart-1.4.40.post21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dmart
-Version: 1.4.40.post19
+Version: 1.4.40.post21
 Requires-Python: >=3.11
 Requires-Dist: fastapi
 Requires-Dist: pydantic

{dmart-1.4.40.post19.dist-info → dmart-1.4.40.post21.dist-info}/RECORD

@@ -8,7 +8,7 @@ dmart/conftest.py,sha256=0ry_zeCmdBNLbm5q115b-pkOrUFYxdsOUXbIMkr7E5Y,362
 dmart/curl.pypi.sh,sha256=KQ-kgV3_d5mwqoLd4XOwz5s2wRQ7LDVX3z-kvjvp9hA,15631
 dmart/curl.sh,sha256=lmHSFVr5ft-lc5Aq9LfvKyWfntrfYbnirhzx1EGjp_A,15743
 dmart/data_generator.py,sha256=CnE-VHEeX7-lAXtqCgbRqR9WHjTuOgeiZcviYrHAmho,2287
-dmart/dmart.py,sha256=q57oMsV93YfdqgJFOxeYhd0eTAwJ_QH8V18qN2hUmt8,32115
+dmart/dmart.py,sha256=eaaggp05-3F0Z-rHgW01TZfFbuk7GhOSbNRtDIEtrWQ,32735
 dmart/get_settings.py,sha256=Sbe2WCoiK398E7HY4SNLfDN_GmE8knR4M-YJWF31jcg,153
 dmart/hypercorn_config.toml,sha256=-eryppEG8HBOM_KbFc4dTQePnpyfoW9ZG5aUATU_6yM,50
 dmart/info.json,sha256=6Kb7MXzVS0PzDZOUXtBoAA3zpxvtRBR0PJnedjBE-v4,123
@@ -483,8 +483,8 @@ dmart/utils/ticket_sys_utils.py,sha256=9QAlW2iiy8KyxQRBDj_WmzS5kKb0aYJmGwd4qzmGV
 dmart/utils/web_notifier.py,sha256=QM87VVid2grC5lK3NdS1yzz0z1wXljr4GChJOeK86W4,843
 dmart/utils/templates/activation.html.j2,sha256=XAMKCdoqONoc4ZQucD0yV-Pg5DlHHASZrTVItNS-iBE,640
 dmart/utils/templates/reminder.html.j2,sha256=aoS8bTs56q4hjAZKsb0jV9c-PIURBELuBOpT_qPZNVU,639
-dmart-1.4.40.post19.dist-info/METADATA,sha256=9-7jAI0wwDqco2DUug9vUfTgHoryxFOqTPk0NX70RVU,839
-dmart-1.4.40.post19.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-dmart-1.4.40.post19.dist-info/entry_points.txt,sha256=N832M4wG8d2GDw1xztKRVM3TnxpY2QDzvdFE8XaWaG8,43
-dmart-1.4.40.post19.dist-info/top_level.txt,sha256=zJo4qk9fUW0BGIR9f4DCfpxaXbvQXH9izIOom6JsyAo,6
-dmart-1.4.40.post19.dist-info/RECORD,,
+dmart-1.4.40.post21.dist-info/METADATA,sha256=tDXPuJEXMjftV_fpEETDfHb9kgcYvG_WoPOtBb1_k1M,839
+dmart-1.4.40.post21.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+dmart-1.4.40.post21.dist-info/entry_points.txt,sha256=N832M4wG8d2GDw1xztKRVM3TnxpY2QDzvdFE8XaWaG8,43
+dmart-1.4.40.post21.dist-info/top_level.txt,sha256=zJo4qk9fUW0BGIR9f4DCfpxaXbvQXH9izIOom6JsyAo,6
+dmart-1.4.40.post21.dist-info/RECORD,,