mongodb增加用户名密码

Context/common.py

@@ -30,7 +30,10 @@ common_db = SqlalchemyConnect(common_db_base, host=conf['mysql']['host'],
                               db=conf['mysql']['db'])
 
 # mongo db
-mg_db = MongoConnect(conf["mongo"]["host"], int(conf["mongo"]["port"]), conf["mongo"]["db"])
+mg_db = MongoConnect(conf["mongo"]["host"], int(conf["mongo"]["port"]),
+                     conf["mongo"]["db"],
+                     conf["mongo"].get('user'),
+                     conf["mongo"].get("password"))
 
 # 邮箱验证码工具类
 email_code_verify = EmailCodeVerify(redis_pool, conf['email']['api'], conf['email']['sender'])
@@ -52,6 +55,6 @@ phone_msg_send = PhoneMsgSend(conf['hw_msg']['url'],
 phone_verify_code = PhoneVerifyCode(redis_pool, phone_msg_send)
 
 # 天眼查接口
-tianyancha_api = TianyanchaApi(conf['tianyancha']['token'])
+tianyancha_api = TianyanchaApi(conf['tianyancha']['token'], mg_db)
 # 预警通接口
-yujingtong_api = YujingtongApi(conf['yujingtong']['token'])
+yujingtong_api = YujingtongApi(conf['yujingtong']['token'], mg_db)

Mods/RegionalEconomies/Mods/AreaIndustrialStructure/Models.py

@@ -10,8 +10,9 @@ class AreaIndustrialStructure(common_db.Base):
     __tablename__ = "area_industrial_structure"
     id = Column(Integer, primary_key=True)
     area_id = Column(Integer, comment="区域ID")
-    count_type = Column(String(128), comment="统计类型")
-    count_value = Column(String(128), comment="统计时间")
+    # count_type = Column(String(128), comment="统计类型")
+    # count_value = Column(String(128), comment="统计时间")
+    year = Column(Integer, comment="统计时间")
     first_Industry_development = Column(Text, comment="第一产业发展情况")
     second_Industry_development = Column(Text, comment="第二产业发展情况")
     tertiary_Industry_development = Column(Text, comment="第三产业发展情况")

Mods/RegionalEconomies/Mods/AreaIndustrialStructure/Schemas.py

@@ -6,8 +6,9 @@ from typing import Optional, List
 class AreaIndustrialStructureInfo(BaseModel):
     id: Optional[int]
     area_id: Optional[int]
-    count_type: Optional[str]
-    count_value: Optional[str]
+    year: Optional[int]
+    # count_type: Optional[str]
+    # count_value: Optional[str]
     first_Industry_development: Optional[str]
     second_Industry_development: Optional[str]
     tertiary_Industry_development: Optional[str]
@@ -15,8 +16,9 @@ class AreaIndustrialStructureInfo(BaseModel):
 
 class AreaIndustrialStructureAddInfo(BaseModel):
     area_id: int
-    count_type: Optional[str]
-    count_value: Optional[str]
+    # count_type: Optional[str]
+    # count_value: Optional[str]
+    year: Optional[int]
     first_Industry_development: Optional[str]
     second_Industry_development: Optional[str]
     tertiary_Industry_development: Optional[str]

Utils/MongoUtils.py

@@ -14,11 +14,17 @@ def pydantic_to_mg_dict(data: BaseModel):
 
 
 class MongoConnect:
-    def __init__(self, host="localhost", port: int = 27017, db: str = None):
+    def __init__(self, host="localhost", port: int = 27017, db: str = None, user: str = None, password: str = None):
         self.host = host
         self.port = port
         self.db = db
-        self.client = MongoClient(host=self.host, port=self.port)
+        self.user = user
+        self.password = password
+        if self.user:
+            self.client = MongoClient(host=self.host, port=self.port,
+                                      username=self.user, password=self.password)
+        else:
+            self.client = MongoClient(host=self.host, port=self.port)
         self.db = self.client[self.db]
 
     def get_db(self):