feat(kz_dash): 数据库公用部分迁移

app.py

@@ -11,7 +11,11 @@
 from components.sidebar import create_sidebar
 from config import APP_NAME, DEBUG, LOG_CONFIG, ROOT_DIR, SERVER_CONFIG, THEME_CONFIG
 from data_source import init_data_source
-from models.database import init_database
+from kz_dash.models.database import init_database
+from models.account import ModelAccount, ModelPortfolio
+from models.fund import ModelFund, ModelFundNav
+from models.fund_user import ModelFundPosition, ModelFundTransaction
+from models.task import ModelTask
 from pages.account import render_account_page
 from pages.home import render_home_page
 from pages.task import render_task_page
@@ -50,7 +54,17 @@ def init_application():
     logger = init_log()
 
     # 初始化数据库
-    init_database()
+    init_database(
+        [
+            ModelAccount,
+            ModelPortfolio,
+            ModelFundPosition,
+            ModelFundTransaction,
+            ModelFundNav,
+            ModelFund,
+            ModelTask,
+        ]
+    )
     logger.info("数据库初始化成功")
 
     # 初始化数据源

backend/api/account.py

@@ -1,10 +1,8 @@
-import uuid
-
 from flask_restx import Namespace, Resource, fields
 
 from kz_dash.backend.api.common import create_list_response_model, create_response_model
 from models.account import ModelAccount, update_account
-from models.database import delete_record, get_record, get_record_list, update_record
+from kz_dash.models.database import delete_record, get_record, get_record_list, update_record
 from kz_dash.utility.response import format_response
 from kz_dash.utility.string_helper import get_uuid
 

backend/api/fund.py

@@ -1,12 +1,14 @@
 from flask_restx import Namespace, Resource, fields
 
 from kz_dash.backend.api.common import create_list_response_model, create_response_model
+from kz_dash.models.database import (
+    delete_record,
+    update_record,
+)
 from models.database import (
     add_fund_position,
-    delete_record,
     get_fund_positions,
     get_fund_transactions,
-    update_record,
 )
 from models.fund_user import ModelFundPosition
 from kz_dash.utility.response import format_response

backend/api/portfolio.py

@@ -3,8 +3,8 @@
 from flask_restx import Namespace, Resource, fields
 
 from kz_dash.backend.api.common import create_list_response_model, create_response_model
+from kz_dash.models.database import delete_record, get_record, get_record_list, update_record
 from models.account import ModelPortfolio
-from models.database import delete_record, get_record, get_record_list, update_record
 from kz_dash.utility.response import format_response
 from kz_dash.utility.string_helper import get_uuid
 

backend/api/task.py

@@ -7,7 +7,7 @@
 from flask_restx import Namespace, Resource, fields
 
 from kz_dash.backend.api.common import create_list_response_model, create_response_model
-from models.database import get_record_list
+from kz_dash.models.database import get_record_list
 from models.task import ModelTask
 from scheduler.job_manager import JobManager
 from scheduler.tasks import TaskFactory

models/account.py

@@ -2,7 +2,7 @@
 
 from peewee import BooleanField, CharField, ForeignKeyField
 
-from models.base import BaseModel
+from kz_dash.models.base import BaseModel
 from kz_dash.utility.string_helper import get_uuid