fastgenerateapi 1.1.9__py2.py3-none-any.whl → 1.1.11__py2.py3-none-any.whl

fastgenerateapi/__version__.py

@@ -8,7 +8,7 @@
 #                                                  d8888P
 
 
-VERSION = (1, 1, 9)
+VERSION = (1, 1, 11)
 
 __version__ = '.'.join(map(str, VERSION))
 

fastgenerateapi/channel/connection_manager.py

@@ -9,15 +9,22 @@ class ConnectionManager:
 
     @classmethod
     def add_connection(cls, user_id: str, websocket: WebSocket):
+        conn = cls.active_connections.get(user_id)
+        if conn:
+            conn.close()
         cls.active_connections[user_id] = websocket
 
     @classmethod
     def del_connection(cls, user_id: str):
         if user_id in cls.active_connections:
+            cls.active_connections[user_id].close()
             del cls.active_connections[user_id]
 
     @classmethod
     def add_group_connection(cls, group_id: str, user_id: str, connection: WebSocket):
+        conn = cls.active_group_connections.get(group_id, {}).get(user_id)
+        if conn:
+            conn.close()
         cls.active_group_connections.setdefault(group_id, {})[user_id] = connection
 
     @classmethod
@@ -25,6 +32,7 @@ class ConnectionManager:
         if group_id in cls.active_group_connections:
             user_dict = cls.active_group_connections[group_id]
             if user_id in user_dict:
+                user_dict[user_id].close()
                 del user_dict[user_id]
                 if not user_dict:
                     del cls.active_group_connections[group_id]

fastgenerateapi/channel/consumer.py

@@ -36,7 +36,7 @@ class Consumer(WebSocketEndpoint):
     # 断开 删除
     async def on_disconnect(self, websocket, close_code):
         if self.group_id and self.user_id:
-            await ConnectionManager.del_group_connection(self.group_id, self.user_id)
+            ConnectionManager.del_group_connection(self.group_id, self.user_id)
         pass
 
     async def check_args(self, group_id, user_id):
@@ -56,7 +56,7 @@ class Consumer(WebSocketEndpoint):
             await self.error(websocket, code=422, msg="未获取到用户信息")
         if not await self.check_args(group_id, user_key):
             await self.error(websocket, code=422, msg="无连接权限")
-        await ConnectionManager.add_group_connection(group_id, user_key, websocket)
+        ConnectionManager.add_group_connection(group_id, user_key, websocket)
 
     # 群发消息
     @classmethod

fastgenerateapi/settings/all_settings.py

@@ -1,5 +1,10 @@
+from functools import lru_cache
+from pathlib import Path
+from typing import Union
 
+import yaml
 from pydantic import BaseModel
+from pydantic.fields import FieldInfo
 
 from fastgenerateapi.settings.redis_settings import RedisSettings
 from fastgenerateapi.settings.app_settings import AppSettings
@@ -14,78 +19,80 @@ class SettingsModel(BaseModel):
     # 缓存配置
     redis_settings: RedisSettings = RedisSettings()
 
-    # pydantic v2 不支持 ModelField, v.type_
-    # @classmethod
-    # def generate_file(cls, path='./.env'):
-    #     """
-    #     生成配置文件
-    #     .env 会增加前缀
-    #     .yaml 忽略前缀
-    #     :param path: 可选 .env / application.yaml
-    #     :return:
-    #     """
-    #     content = ''
-    #     setting_models = cls.__fields__.copy()
-    #     if path.__contains__('.env'):
-    #         for k, v in setting_models.items():
-    #             if issubclass(type(v), ModelField):
-    #                 sub_setting_class = v.type_
-    #                 content += f"[{sub_setting_class.__name__}]\n"
-    #                 fields = sub_setting_class.__fields__.copy()
-    #                 for k, v in fields.items():
-    #                     field_name = f'{sub_setting_class.Config.env_prefix}{k}'
-    #                     if v.field_info.description is None:
-    #                         content += f"{field_name}={v.default}\n"
-    #                     else:
-    #                         content += f"# {v.field_info.description}\n{field_name}={v.default}\n"
-    #                 content += "\n"
-    #     elif path.__contains__('.yaml'):
-    #         for k, v in setting_models.items():
-    #             if issubclass(type(v), ModelField):
-    #                 sub_setting_class = v.type_
-    #                 content += f"{sub_setting_class.__name__}:\n"
-    #                 fields = sub_setting_class.__fields__.copy()
-    #                 for k, v in fields.items():
-    #                     content += f"  {k}: {v.default}"
-    #                     if v.field_info.description is None:
-    #                         content += f"\n"
-    #                     else:
-    #                         content += f"  # {v.field_info.description}\n"
-    #                 content += "\n"
-    #     with open(file=path, mode='w', encoding='utf-8') as f:
-    #         f.writelines(content)
-    #
-    # @classmethod
-    # @lru_cache
-    # def get_global_settings(cls, path: Union[types.Path, str, None] = '.env'):
-    #     """
-    #     get global settings and set into cache
-    #     :param path: 可选 .env / application.yaml
-    #     :return:
-    #     """
-    #     setting_models = cls.__fields__.copy()
-    #
-    #     setting_dict = {}
-    #     if str(path).__contains__('.env'):
-    #         for k, v in setting_models.items():
-    #             if issubclass(type(v), ModelField):
-    #                 sub_setting_class = v.type_
-    #                 setting_dict[k] = sub_setting_class(_env_file=str(path))
-    #     elif str(path).__contains__('.yaml'):
-    #         with open(path, 'r', encoding='utf-8') as file:
-    #             data_dict = yaml.safe_load(file)
-    #         for k, v in setting_models.items():
-    #             if issubclass(type(v), ModelField):
-    #                 sub_setting_class = v.type_
-    #                 setting_dict[k] = sub_setting_class(**data_dict.get(sub_setting_class.__name__, {}))
-    #
-    #     setting_data = cls(**setting_dict)
-    #     global settings
-    #     settings.app_settings = setting_data.app_settings
-    #     settings.system_settings = setting_data.system_settings
-    #     settings.redis_settings = setting_data.redis_settings
-    #     return setting_data
+    @classmethod
+    def generate_file(cls, path='./.env'):
+        """
+        生成配置文件
+        .env 会增加前缀
+        .yaml 忽略前缀
+        :param path: 可选 .env / application.yaml
+        :return:
+        """
+        content = ''
+        setting_models = cls.model_fields.copy()
+        if path.__contains__('.env'):
+            for k, v in setting_models.items():
+                if isinstance(v, FieldInfo):
+                    content += f"[{v.annotation.__name__}]\n"
+                    env_prefix = v.annotation.model_config.get("env_prefix", "")
+                    fields = v.annotation.model_fields.copy()
+                    for k, v in fields.items():
+                        field_name = f'{env_prefix}{k}'
+                        if v.description is None:
+                            content += f"{field_name}={v.default}\n"
+                        else:
+                            content += f"# {v.description}\n{field_name}={v.default}\n"
+                    content += "\n"
+        elif path.__contains__('.yaml'):
+            for k, v in setting_models.items():
+                if isinstance(v, FieldInfo):
+                    content += f"[{v.annotation.__name__}]\n"
+                    env_prefix = v.annotation.model_config.get("env_prefix", "")
+                    fields = v.annotation.model_fields.copy()
+                    for k, v in fields.items():
+                        field_name = f'{env_prefix}{k}'
+                        content += f"  {field_name}: {v.default}"
+                        if v.description is None:
+                            content += f"\n"
+                        else:
+                            content += f"  # {v.description}\n"
+                    content += "\n"
+        with open(file=path, mode='w', encoding='utf-8') as f:
+            f.writelines(content)
+
+    @classmethod
+    @lru_cache
+    def get_global_settings(cls, path: Union[Path, str, None] = '.env'):
+        """
+        get global settings and set into cache
+        :param path: 可选 .env / application.yaml
+        :return:
+        """
+        setting_models = cls.model_fields.copy()
+
+        setting_dict = {}
+        if str(path).__contains__('.env'):
+            for k, v in setting_models.items():
+                if isinstance(v, FieldInfo):
+                    setting_dict[k] = v.annotation(_env_file=str(path))
+        elif str(path).__contains__('.yaml'):
+            with open(path, 'r', encoding='utf-8') as file:
+                data_dict = yaml.safe_load(file)
+            for k, v in setting_models.items():
+                if isinstance(v, FieldInfo):
+                    setting_dict[k] = v.annotation(**data_dict.get(v.annotation.__name__, {}))
+
+        setting_data = cls(**setting_dict)
+        global settings
+        settings.app_settings = setting_data.app_settings
+        settings.system_settings = setting_data.system_settings
+        settings.redis_settings = setting_data.redis_settings
+        return setting_data
 
 
 settings = SettingsModel()
 
+
+if __name__ == '__main__':
+    settings.generate_file()
+

fastgenerateapi/settings/db_settings.py

@@ -21,7 +21,6 @@ class DBSettings(BaseSettings):
         return f"{self.DB_TYPE.lower()}://{self.DB_USERNAME}:{self.DB_PASSWORD}@{self.DB_HOST}:{self.DB_PORT}/{self.DB_DATABASE}"
 
     model_config = SettingsConfigDict(
-        env_prefix="",
         env_file=".env",
         case_sensitive=True,
         extra='ignore'

fastgenerateapi/settings/file_settings.py

@@ -4,7 +4,7 @@ from pydantic import Field
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
-class FileServerSettings(BaseSettings):
+class FileSettings(BaseSettings):
     """
     文件配置
     """
@@ -13,7 +13,6 @@ class FileServerSettings(BaseSettings):
     FILE_ROOT: Optional[str] = Field(default='static', description="文件储存路径", title="file storage path")
 
     model_config = SettingsConfigDict(
-        env_prefix="FILESERVER_",
         env_file=".env",
         case_sensitive=True,
         extra='ignore'

{fastgenerateapi-1.1.9.dist-info → fastgenerateapi-1.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fastgenerateapi
-Version: 1.1.9
+Version: 1.1.11
 Summary: FastAPIView Class View
 Author: ShiLiang
 Author-email: 2509144896@qq.com

{fastgenerateapi-1.1.9.dist-info → fastgenerateapi-1.1.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 fastgenerateapi/__init__.py,sha256=eh2qtp7FbjrgaRX5b3H3UWpasFdNHbVFqLUhpqVbaTU,1287
-fastgenerateapi/__version__.py,sha256=sI8rW3wVqSh9IB7senA2sM8kxpevmXiyJUcRPE0EWQk,496
+fastgenerateapi/__version__.py,sha256=-n7dhyb2qQZHtAwcRPpSC9Z4U1Vo6n0LrAe9Qiooq-I,497
 fastgenerateapi/api_view/__init__.py,sha256=zkaTX5JxsNfjF-dFeEbHfUB58vhPMjm6Iiqx9HgJOrY,14
 fastgenerateapi/api_view/api_view.py,sha256=mfD8GB-hnyI2XO3tkSOlEa2FfBLdq0_Wqvp8gFrXFKU,1160
 fastgenerateapi/api_view/base_view.py,sha256=s8f6zdoRqPRmo6o2FqHZG2MKcn1gc8JELix8VetVo3k,9810
@@ -32,8 +32,8 @@ fastgenerateapi/cache/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 fastgenerateapi/cache/cache_decorator.py,sha256=u6dfanX2GMkjsJANgT8qXuEgw1fSPXFsmpL05b0PI44,579
 fastgenerateapi/cache/key_builder.py,sha256=6Vudd19EGg1IUGT5ftPDqLUNCCqXzmQNV69zBfjGf9A,503
 fastgenerateapi/channel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastgenerateapi/channel/connection_manager.py,sha256=BrsS0TMfgMBu3AQEb2RJwgfEUW27-e3bdZC0lM3qCls,1092
-fastgenerateapi/channel/consumer.py,sha256=1AZFAoeUZHZYZYJbN9s4pzI4xTG8uD0FbiSvmwjRGD8,3221
+fastgenerateapi/channel/connection_manager.py,sha256=Z59pFnQSsFWSPvdH0b_8m1KTFVYtqREXzkJvgpej4gc,1405
+fastgenerateapi/channel/consumer.py,sha256=9zZ3ctN88ncMHRrWndVSRTeqvMExyU3O3hq38ku7lzw,3209
 fastgenerateapi/channel/websocket_view.py,sha256=lG9rYMO_FTdULi--oEdqCtdguR5ZM8Uzrk6f3z-86zo,2656
 fastgenerateapi/controller/__init__.py,sha256=eM5a3k_G0bf-k_WcwfcBbI-M3kLFJIITBWX85lVyClE,316
 fastgenerateapi/controller/filter_controller.py,sha256=PX-2oOo7WGZWePjm318RCbIlLIdeYwHKG-u1onpjB24,5030
@@ -83,10 +83,10 @@ fastgenerateapi/schemas_factory/response_factory.py,sha256=C-E3lqIdfIJCIZhvmSaAi
 fastgenerateapi/schemas_factory/sql_get_all_schema_factory.py,sha256=dhgMvXZIq4av7puIMiQOTuVqNOuBoMvBZVd98GWJTRg,1810
 fastgenerateapi/schemas_factory/update_schema_factory.py,sha256=E-86vF98Ed92Q2VFqtS35xqxr_c34du9ok42JBfnj1g,3729
 fastgenerateapi/settings/__init__.py,sha256=dlyJb7BNh8Svau-UOrV4pxfoSYnNTA6eE41fu43vlT8,173
-fastgenerateapi/settings/all_settings.py,sha256=riAFFyOVRszZ3DZUVcudHgCgYrICnusuuFcth9zfeDM,3889
+fastgenerateapi/settings/all_settings.py,sha256=ylABCKJzh_US4ZfZQgDEYFYm75Zz-0mL8pqziFdgudY,3795
 fastgenerateapi/settings/app_settings.py,sha256=FgRVIrWUmfvfDohP2pP4JO7nFevUPEG3MNuUyfjM9BA,7294
-fastgenerateapi/settings/db_settings.py,sha256=ClmQg00bpn-EWnrdKjjJ5_emPoa1eEDf0oBoNGp7g3M,1594
-fastgenerateapi/settings/file_settings.py,sha256=Vtzp_LV4tBYQDwNOJfY3QyoQGvDEv_KZ3N_HR9md1-o,898
+fastgenerateapi/settings/db_settings.py,sha256=duTgLjMHO6MTDfEZ2tkAGiSXrTKuHSLBJW00l5_rv7g,1570
+fastgenerateapi/settings/file_settings.py,sha256=8Ra5H0nxRp1r_Ib_RGbNO0rUFaxqYMjuJT9A4fHE3Oo,857
 fastgenerateapi/settings/jwt_settings.py,sha256=mRtzLZz1bntrl6UXAnIq6xilEEN8NS8oJSv62CGXOEs,971
 fastgenerateapi/settings/otlp_settings.py,sha256=4haWpSdJBeor-rq_rBOgQl6oYRv9eppdwm1yWGt6GyE,3161
 fastgenerateapi/settings/redis_settings.py,sha256=6T-KqCj2Jva7SarDCExZZbdvr54u6n08mEJ0gQhlUoc,700
@@ -103,8 +103,8 @@ fastgenerateapi/utils/snowflake.py,sha256=GiFVkE10sPiqJ094sMfrPsaV7Y9Y3c1djrSfgs
 fastgenerateapi/utils/str_util.py,sha256=c-jUlCFw-Dz4W1W9Jc1TqGZw3JXu-qN5ovnE6fjc9j0,3445
 fastgenerateapi/utils/swagger_to_js.py,sha256=pPPTag6TYtxdbKMHD3m8lJvc8Gv9HC97CGHt4esU1-E,530
 script/__init__.py,sha256=26UWatnbm6ZIwQMuu9NNzQ0IW1ACO4Oe9caModuTpWM,4
-fastgenerateapi-1.1.9.dist-info/LICENSE,sha256=gcuuhKKc5-dwvyvHsXjlC9oM6N5gZ6umYbC8ewW1Yvg,35821
-fastgenerateapi-1.1.9.dist-info/METADATA,sha256=dfLtlXB1ICGWrrJIb2aHRbe0j0yOYHbjZcajqSbpWxU,5960
-fastgenerateapi-1.1.9.dist-info/WHEEL,sha256=iYlv5fX357PQyRT2o6tw1bN-YcKFFHKqB_LwHO5wP-g,110
-fastgenerateapi-1.1.9.dist-info/top_level.txt,sha256=CW2SlpYjTRdacF-5ufnPMtwpYcR0XYn_bDxa2ZrrTBI,23
-fastgenerateapi-1.1.9.dist-info/RECORD,,
+fastgenerateapi-1.1.11.dist-info/LICENSE,sha256=gcuuhKKc5-dwvyvHsXjlC9oM6N5gZ6umYbC8ewW1Yvg,35821
+fastgenerateapi-1.1.11.dist-info/METADATA,sha256=jFaw2DlNnKiSf98Co48jlHrprvW8sKQjehfXfStj3to,5961
+fastgenerateapi-1.1.11.dist-info/WHEEL,sha256=iYlv5fX357PQyRT2o6tw1bN-YcKFFHKqB_LwHO5wP-g,110
+fastgenerateapi-1.1.11.dist-info/top_level.txt,sha256=CW2SlpYjTRdacF-5ufnPMtwpYcR0XYn_bDxa2ZrrTBI,23
+fastgenerateapi-1.1.11.dist-info/RECORD,,