From 3e188a54c3dec49a9f8f75072a66c50bba35debf Mon Sep 17 00:00:00 2001 From: tonyrewin Date: Sun, 25 Dec 2022 08:45:13 +0300 Subject: [PATCH] fix-migration-call --- migration/__init__.py | 10 +++++++--- server.py | 4 ++-- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/migration/__init__.py b/migration/__init__.py index c41ced99..0293edc8 100644 --- a/migration/__init__.py +++ b/migration/__init__.py @@ -258,17 +258,21 @@ def data_load(): return storage -async def main(): +async def handling_migration(): init_tables() await all_handle(data_load(), sys.argv) -if __name__ == "__main__": +def process(): if "migrate" in sys.argv: loop = asyncio.get_event_loop() - loop.run_until_complete(main()) + loop.run_until_complete(handling_migration()) elif "bson" in sys.argv: import bson2json bson2json.json_tables() else: print("[migration] usage: python server.py migrate") + + +if __name__ == "__main__": + process() diff --git a/server.py b/server.py index 28b1150e..510a0257 100644 --- a/server.py +++ b/server.py @@ -89,10 +89,10 @@ if __name__ == "__main__": reload=want_reload ) # , ssl_keyfile="discours.key", ssl_certfile="discours.crt") elif x == "migrate": - from migration import migrate + from migration import process print("MODE: MIGRATE") - migrate() + process() elif x == "bson": from migration.bson2json import json_tables print("MODE: BSON")