init

SourceCode/IntelligentRailwayCosting/app/ai/openai.py

@@ -20,12 +20,12 @@ class OpenAi:
             self._max_tokens = int(max_tokens)
 
     def call_openai(
-            self,
-            system_prompt: str,
-            user_prompt: str,
-            api_url: str = None,
-            api_key: str = None,
-            api_model: str = None,
+        self,
+        system_prompt: str,
+        user_prompt: str,
+        api_url: str = None,
+        api_key: str = None,
+        api_model: str = None,
     ) -> json:
         self.check_api(api_key, api_model, api_url)
         utils.get_logger().info(
@@ -134,24 +134,24 @@ class OpenAi:
                 raise Exception(f"解析 AI 响应错误: {response} {e}")
 
     def call_openai_with_image(
-            self,
-            image_path,
-            system_prompt: str,
-            user_prompt: str,
-            api_url: str = None,
-            api_key: str = None,
-            api_model: str = None,
+        self,
+        image_path,
+        system_prompt: str,
+        user_prompt: str,
+        api_url: str = None,
+        api_key: str = None,
+        api_model: str = None,
     ) -> json:
         pass
 
     def call_openai_with_file(
-            self,
-            file_path,
-            system_prompt: str,
-            user_prompt: str,
-            api_url: str = None,
-            api_key: str = None,
-            api_model: str = None,
+        self,
+        file_path,
+        system_prompt: str,
+        user_prompt: str,
+        api_url: str = None,
+        api_key: str = None,
+        api_model: str = None,
     ) -> json:
         self.check_api(api_key, api_model, api_url)
         utils.get_logger().info(

SourceCode/IntelligentRailwayCosting/app/core/api/response.py

@@ -34,7 +34,7 @@ class ResponseBase:
 
     @staticmethod
     def error(
-            message: str = "操作失败", code: int = 400, data: Optional[Any] = None
+        message: str = "操作失败", code: int = 400, data: Optional[Any] = None
     ) -> Response:
         """错误响应
         Args:
@@ -52,10 +52,10 @@ class ResponseBase:
 
     @staticmethod
     def json_response(
-            success: bool = True,
-            code: int = 200,
-            message: str = "",
-            data: Optional[Any] = None,
+        success: bool = True,
+        code: int = 200,
+        message: str = "",
+        data: Optional[Any] = None,
     ) -> Response:
         """自定义响应
         Args:

SourceCode/IntelligentRailwayCosting/app/core/api/table_response.py

@@ -10,7 +10,7 @@ class TableResponse(ResponseBase):
 
     @staticmethod
     def success(
-            rows: List[Dict] = None, total: int = 0, message: str = "操作成功"
+        rows: List[Dict] = None, total: int = 0, message: str = "操作成功"
     ) -> Response:
         """表格数据成功响应
         Args:

SourceCode/IntelligentRailwayCosting/app/core/dtos/chapter.py

@@ -66,9 +66,9 @@ class ChapterDto(BaseModel):
 
     @classmethod
     def from_model(
-            cls,
-            chapter_model: ChapterModel,
-            budget_item_model: Optional[TotalBudgetItemModel] = None,
+        cls,
+        chapter_model: ChapterModel,
+        budget_item_model: Optional[TotalBudgetItemModel] = None,
     ) -> "ChapterDto":
         """从数据库模型创建DTO对象"""
         dto = cls(

SourceCode/IntelligentRailwayCosting/app/core/user_session/current_user.py

@@ -16,11 +16,11 @@ class CurrentUser(UserMixin):
     # _project_supplement: Optional[str] = None
 
     def __init__(
-            self,
-            user_id: Optional[int] = None,
-            username: Optional[str] = None,
-            item_range: Optional[str] = None,
-            specialty: Optional[str] = None,
+        self,
+        user_id: Optional[int] = None,
+        username: Optional[str] = None,
+        item_range: Optional[str] = None,
+        specialty: Optional[str] = None,
     ):
         self._user_id = user_id
         self._username = username

SourceCode/IntelligentRailwayCosting/app/stores/quota_input.py

@@ -20,13 +20,13 @@ class QuotaInputStore:
         return self._current_user
 
     def get_quotas_paginated(
-            self,
-            project_id: str,
-            budget_id: int,
-            item_id: int,
-            page: int = 1,
-            page_size: int = 10,
-            keyword: Optional[str] = None,
+        self,
+        project_id: str,
+        budget_id: int,
+        item_id: int,
+        page: int = 1,
+        page_size: int = 10,
+        keyword: Optional[str] = None,
     ):
         """分页查询定额输入列表
 
@@ -147,7 +147,7 @@ class QuotaInputStore:
             return QuotaInputDto.from_model(model)
 
     def update_quota(
-            self, project_id: str, dto: QuotaInputDto
+        self, project_id: str, dto: QuotaInputDto
     ) -> Optional[QuotaInputDto]:
         """更新定额输入
 

SourceCode/IntelligentRailwayCosting/app/stores/railway_costing_sqlserver/log.py

@@ -12,15 +12,15 @@ class LogStore:
         self._database = "Iwb_RailwayCosting"
 
     def query_logs_paginated(
-            self,
-            page: int = 1,
-            page_size: int = 10,
-            username: Optional[str] = None,
-            operation_type: Optional[str] = None,
-            operation_module: Optional[str] = None,
-            operation_result: Optional[int] = None,
-            start_time: Optional[datetime] = None,
-            end_time: Optional[datetime] = None,
+        self,
+        page: int = 1,
+        page_size: int = 10,
+        username: Optional[str] = None,
+        operation_type: Optional[str] = None,
+        operation_module: Optional[str] = None,
+        operation_result: Optional[int] = None,
+        start_time: Optional[datetime] = None,
+        end_time: Optional[datetime] = None,
     ) -> Dict[str, Any]:
         """
         分页查询日志记录
@@ -68,15 +68,15 @@ class LogStore:
             return {"total": total, "data": logs}
 
     def insert_log(
-            self,
-            username: str,
-            operation_type: str,
-            operation_desc: Optional[str] = None,
-            operation_result: Optional[int] = None,
-            operation_module: Optional[str] = None,
-            operation_data: Optional[str] = None,
-            data_changes: Optional[str] = None,
-            operation_ip: Optional[str] = None,
+        self,
+        username: str,
+        operation_type: str,
+        operation_desc: Optional[str] = None,
+        operation_result: Optional[int] = None,
+        operation_module: Optional[str] = None,
+        operation_data: Optional[str] = None,
+        data_changes: Optional[str] = None,
+        operation_ip: Optional[str] = None,
     ) -> LogModel:
         """
         插入单条日志记录

SourceCode/IntelligentRailwayCosting/app/test/mysqy_test.py

@@ -120,14 +120,14 @@ class TestMySQLHelper(unittest.TestCase):
         try:
             # 测试成功的事务
             with self.db_helper.session_scope(
-                    self.test_db, self.test_config
+                self.test_db, self.test_config
             ) as session:
                 user = TestUser(name="test_user", email="test@example.com")
                 session.add(user)
 
             # 验证数据已保存
             with self.db_helper.session_scope(
-                    self.test_db, self.test_config
+                self.test_db, self.test_config
             ) as session:
                 saved_user = session.query(TestUser).filter_by(name="test_user").first()
                 self.assertIsNotNone(saved_user)
@@ -136,7 +136,7 @@ class TestMySQLHelper(unittest.TestCase):
             # 测试事务回滚
             with self.assertRaises(Exception):
                 with self.db_helper.session_scope(
-                        self.test_db, self.test_config
+                    self.test_db, self.test_config
                 ) as session:
                     user = TestUser(name="rollback_user", email="invalid_email")
                     session.add(user)
@@ -144,7 +144,7 @@ class TestMySQLHelper(unittest.TestCase):
 
             # 验证数据已回滚
             with self.db_helper.session_scope(
-                    self.test_db, self.test_config
+                self.test_db, self.test_config
             ) as session:
                 rollback_user = (
                     session.query(TestUser).filter_by(name="rollback_user").first()

SourceCode/IntelligentRailwayCosting/app/tools/db_helper/__init__.py

@@ -64,7 +64,7 @@ def mysql_session(database: str = None, config: Optional[Dict[str, Any]] = None)
 
 @contextmanager
 def sqlserver_query_session(
-        database: str = None, config: Optional[Dict[str, Any]] = None
+    database: str = None, config: Optional[Dict[str, Any]] = None
 ):
     """SQLServer数据库会话的上下文管理器（只读查询专用）