From fe393ff95c67240d6131690f905e22926dc50752 Mon Sep 17 00:00:00 2001
From: 283375 <bilibili283375@163.com>
Date: Fri, 25 Aug 2023 02:26:19 +0800
Subject: [PATCH] feat: add common model

---
 src/arcaea_offline/models/common.py | 18 ++++++++++++++++++
 src/arcaea_offline/models/scores.py |  8 ++++++++
 src/arcaea_offline/models/songs.py  | 18 +++++++++++-------
 3 files changed, 37 insertions(+), 7 deletions(-)
 create mode 100644 src/arcaea_offline/models/common.py

diff --git a/src/arcaea_offline/models/common.py b/src/arcaea_offline/models/common.py
new file mode 100644
index 0000000..435c075
--- /dev/null
+++ b/src/arcaea_offline/models/common.py
@@ -0,0 +1,18 @@
+from sqlalchemy import TEXT
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+
+__all__ = [
+    "CommonBase",
+    "Property",
+]
+
+
+class CommonBase(DeclarativeBase):
+    pass
+
+
+class Property(CommonBase):
+    __tablename__ = "property"
+
+    id: Mapped[str] = mapped_column(TEXT(), primary_key=True)
+    value: Mapped[str] = mapped_column(TEXT())
diff --git a/src/arcaea_offline/models/scores.py b/src/arcaea_offline/models/scores.py
index f0b9706..d7e3393 100644
--- a/src/arcaea_offline/models/scores.py
+++ b/src/arcaea_offline/models/scores.py
@@ -6,6 +6,14 @@ from sqlalchemy_utils import create_view
 
 from .songs import Chart, ChartInfo
 
+__all__ = [
+    "ScoresBase",
+    "Score",
+    "Calculated",
+    "Best",
+    "CalculatedPotential",
+]
+
 
 class ScoresBase(DeclarativeBase):
     pass
diff --git a/src/arcaea_offline/models/songs.py b/src/arcaea_offline/models/songs.py
index 81e79eb..29afc59 100644
--- a/src/arcaea_offline/models/songs.py
+++ b/src/arcaea_offline/models/songs.py
@@ -3,18 +3,22 @@ from typing import Optional
 from sqlalchemy import TEXT, ForeignKey
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
+__all__ = [
+    "SongsBase",
+    "Pack",
+    "PackLocalized",
+    "Song",
+    "SongLocalized",
+    "Chart",
+    "ChartLocalized",
+    "ChartInfo",
+]
+
 
 class SongsBase(DeclarativeBase):
     pass
 
 
-class Property(SongsBase):
-    __tablename__ = "property"
-
-    id: Mapped[str] = mapped_column(TEXT(), primary_key=True)
-    value: Mapped[str] = mapped_column(TEXT())
-
-
 class Pack(SongsBase):
     __tablename__ = "pack"