From 56c2275145054f5e914215b399f5c6929586f089 Mon Sep 17 00:00:00 2001
From: "Autumn.home" <autumn@home.com>
Date: Mon, 17 Jun 2024 22:08:38 +0800
Subject: [PATCH] fix

---
 main.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/main.py b/main.py
index d92c29d..aba209f 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,7 @@ from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.staticfiles import StaticFiles
 
 from core.config import *
+from core.db import get_mongo_db
 
 set_config()
 from starlette.requests import Request
@@ -27,9 +28,27 @@ app = FastAPI()
 
 from core.apscheduler_handler import scheduler
 
+# async def update():
+#     if float(VERSION) <= 1.2:
+#         async for db in get_mongo_db():
+#             cursor = db.project.find({"root_domains": ""},{"_id":1, "root_domains": 1})
+#             async for document in cursor:
+#                 logger.info("Update found empty root_domains")
+#                 root_domain = []
+#                 for root in document["root_domains"]:
+#                     if root != "":
+#                         root_domain.append(root)
+#                 update_document = {
+#                     "$set": {
+#                         "root_domains": root_domain,
+#                     }
+#                 }
+#                 await db.project.update_one({"_id": document['_id']}, update_document)
+
 
 @app.on_event("startup")
 async def startup_db_client():
+    # await update()
     file_path = os.path.join(os.getcwd(), 'file')
     if not os.path.exists(file_path):
         os.makedirs(file_path)