Update sync api

backend/analytics_server/mhq/api/sync.py

@@ -1,12 +1,23 @@
 from flask import Blueprint
 
+from mhq.service.query_validator import get_query_validator
 from mhq.service.sync_data import trigger_data_sync
+from mhq.utils.lock import get_redis_lock_service
+from mhq.utils.log import LOG
 from mhq.utils.time import time_now
 
 app = Blueprint("sync", __name__)
 
 
 @app.route("/sync", methods=["POST"])
 def sync():
-    trigger_data_sync()
+    default_org = get_query_validator().get_default_org()
+    if not default_org:
+        raise Exception("Default org not found")
+    org_id = str(default_org.id)
+    with get_redis_lock_service().acquire_lock("{org}:" + f"{str(org_id)}:data_sync"):
+        try:
+            trigger_data_sync(org_id)
+        except Exception as e:
+            LOG.error(f"Error syncing data for org {org_id}: {str(e)}")
     return {"message": "sync started", "time": time_now().isoformat()}

backend/analytics_server/mhq/service/sync_data.py

@@ -26,15 +26,3 @@ def trigger_data_sync(org_id: str):
             )
             continue
     LOG.info(f"Data sync for org {org_id} completed successfully")
-
-
-if __name__ == "__main__":
-    default_org = get_query_validator().get_default_org()
-    if not default_org:
-        raise Exception("Default org not found")
-    org_id = str(default_org.id)
-    with get_redis_lock_service().acquire_lock("{org}:" + f"{str(org_id)}:data_sync"):
-        try:
-            trigger_data_sync(org_id)
-        except Exception as e:
-            LOG.error(f"Error syncing data for org {org_id}: {str(e)}")