mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 13:12:24 +00:00
Wu Clan
b28a4d7673
* Update the primary database to postgresql * Delete the prints * Fix lint * Update models in code generation * Delete print * Fix the salt column type
29 lines
740 B
Python
29 lines
740 B
Python
import os.path
|
|
|
|
from backend.core.path_conf import BASE_PATH
|
|
from backend.utils.import_parse import get_model_objects
|
|
|
|
|
|
def get_app_models() -> list[type]:
|
|
"""获取 app 所有模型类"""
|
|
app_path = BASE_PATH / 'app'
|
|
list_dirs = os.listdir(app_path)
|
|
|
|
apps = [d for d in list_dirs if os.path.isdir(os.path.join(app_path, d)) and d != '__pycache__']
|
|
|
|
objs = []
|
|
for app in apps:
|
|
module_path = f'backend.app.{app}.model'
|
|
obj = get_model_objects(module_path)
|
|
if obj:
|
|
objs.extend(obj)
|
|
|
|
return objs
|
|
|
|
|
|
# import all app models for auto create db tables
|
|
for cls in get_app_models():
|
|
class_name = cls.__name__
|
|
if class_name not in globals():
|
|
globals()[class_name] = cls
|