migrate to SQLModel

app/models/backend.py

@@ -0,0 +1,58 @@
+from typing import Optional
+
+from sqlmodel import Field, SQLModel
+from python_aid import aidx
+
+class BEConfig(SQLModel, table=True):
+    id: Optional[str] = Field(default=aidx.genAidx(), primary_key=True)
+    host: str = Field(unique=True)
+
+    name: str = Field(default="Holo")
+    description: str = Field(default="An Interconnected Extensible Microblogging Platform🪐")
+
+    repositoryUrl: str = Field(default="https://github.com/hol0-dev/backend")
+    feedbackUrl: str = Field(default="https://github.com/hol0-dev/backend/issues")
+
+    admin: Optional[str] = Field(default=None)
+    adminEmail: Optional[str] = Field(default=None)
+
+    maintainer: Optional[str] = Field(default=None)
+    maintainerEmail: Optional[str] = Field(default=None)
+
+    impressumUrl: Optional[str] = Field(default=None)
+    tosUrl: Optional[str] = Field(default=None)
+    privacyPolicyUrl: Optional[str] = Field(default=None)
+
+    pushNotification: bool = Field(default=False)
+    swPublicKey: Optional[str] = Field(default=None)
+    swPrivateKey: Optional[str] = Field(default=None)
+
+    enableMail: bool = Field(default=False)
+    mailAddress: Optional[str] = Field(default=None)
+    smtpHost: Optional[str] = Field(default=None)
+    smtpPort: Optional[str] = Field(default=None)
+    smtpUser: Optional[str] = Field(default=None)
+    smtpPass: Optional[str] = Field(default=None)
+    smtpSSL: bool = Field(default=False)
+
+    useObjectStrage: bool = Field(default=False)
+    s3BaseUrl: Optional[str] = Field(default=None)
+    s3Bucket: Optional[str] = Field(default=None)
+    s3Prefix: Optional[str] = Field(default=None)
+    s3Endpoint: Optional[str] = Field(default=None)
+    s3Region: Optional[str] = Field(default="us-east-1")
+    s3AccessKey: Optional[str] = Field(default=None)
+    s3SecretKey: Optional[str] = Field(default=None)
+    s3useSSL: bool = Field(default=False)
+    s3ForcePathStyle: bool = Field(default=False)
+    s3setPublicRead: bool = Field(default=False)
+
+    enableTurnstile: bool = Field(default=False)
+    turnstileSiteKey: Optional[str] = Field(default=None)
+    turnstileSecretKey: Optional[str] = Field(default=None)
+
+    faviconUrl: Optional[str] = Field(default=None)
+    appleTouchIconUrl: Optional[str] = Field(default=None)
+    androidTouchIconUrl: Optional[str] = Field(default=None)
+
+    turnstileSecretKey: str = Field(default="#B0F7DD")

app/models/channel.py

@@ -0,0 +1,36 @@
+from sqlmodel import Field, SQLModel, Relationship
+from datetime import datetime
+from typing import List, Optional, TYPE_CHECKING
+
+from ..utils.model_func import utcnow
+
+if TYPE_CHECKING:
+    from .instance import Instance
+    from .note import Note
+    from .user import User
+
+class ChannelFollow(SQLModel, table=True):
+    followee_id: str = Field(primary_key=True, index=True)
+    channel_id: str = Field(foreign_key="channel.id", index=True)
+    created_at: datetime = Field(default_factory=utcnow)
+
+    channel: "Channel" = Relationship(sa_relationship_kwargs={"cascade": "all, delete"})
+    followee: "User" = Relationship(back_populates="channel_followees", sa_relationship_kwargs={"cascade": "all, delete"})
+
+
+class Channel(SQLModel, table=True):
+    id: str = Field(primary_key=True, index=True)
+
+    name: str
+    description: str
+    remote: Optional[str] = Field(default=None)
+
+    created_at: datetime = Field(default_factory=utcnow, sa_column_kwargs={"name": "created_at"})
+    updated_at: datetime = Field(default_factory=utcnow, sa_column_kwargs={"name": "updated_at", "onupdate": utcnow})
+
+    host_id: Optional[str] = Field(default=None, foreign_key="instance.host")
+
+    followers: List[ChannelFollow] = Relationship(back_populates="channel")
+    notes: List["Note"] = Relationship(back_populates="channel")
+
+# 補足としてUserとInstance、Noteのクラスも定義する必要がありますが、ここでは省略します。

app/models/instance.py

@@ -0,0 +1,42 @@
+from datetime import datetime
+from typing import List, Optional, TYPE_CHECKING
+from sqlmodel import Field, Relationship, SQLModel
+
+if TYPE_CHECKING:
+    from .user import User
+
+class Instance(SQLModel, table=True):
+    id: str = Field(primary_key=True)
+    host: str = Field(unique=True)
+    usersCount: int = Field(default=0)
+    notesCount: int = Field(default=0)
+    name: Optional[str]
+    description: Optional[str]
+    iconUrl: str
+    faviconUrl: str
+    themeColor: str = Field(default="#ffffff")
+    firstRetrievedAt: Optional[datetime] = Field(default=datetime.now())
+    isNotResponding: Optional[bool] = Field(default=False)
+    isSuspended: Optional[bool] = Field(default=False)
+    isBlocked: Optional[bool] = Field(default=False)
+    isSilenced: Optional[bool] = Field(default=False)
+    softwareName: Optional[str]
+    softwareVersion: Optional[str]
+    softwareHomepage: Optional[str]
+    openRegistrations: bool
+    adminName: Optional[str]
+    adminEmail: Optional[str]
+    maintainerName: Optional[str]
+    maintainerEmail: Optional[str]
+    infoUpdatedAt: datetime = Field(default=datetime.now())
+    latestRequestReceivedAt: datetime = Field(default=datetime.now())
+    moderationNote: Optional[str]
+    langs: List[str]
+    tosUrl: Optional[str]
+    privacyPolicyUrl: Optional[str]
+    inquiryUrl: Optional[str]
+    impressumUrl: Optional[str]
+    repositoryUrl: Optional[str]
+    feedbackUrl: Optional[str]
+
+    users: List["User"] = Relationship(back_populates="instance", field="host", remote_model="User")

app/models/note.py

@@ -0,0 +1,49 @@
+from datetime import datetime
+from typing import List, Optional, TYPE_CHECKING
+
+from sqlmodel import Field, Relationship, SQLModel
+from python_aid import aidx
+
+from ..utils.model_func import utcnow
+
+if TYPE_CHECKING:
+    from .channel import Channel
+    from .instance import Instance
+    from .user import User
+
+class Reactions(SQLModel, table=True):
+    id: str = Field(default=None, primary_key=True)
+    noteId: str = Field(foreign_key="note.id")
+    userId: str = Field(foreign_key="user.id")
+
+    note: "Note" = Relationship(back_populates="reactions")
+    user: "User" = Relationship(back_populates="reactions", sa_relationship_kwargs={"cascade": "all, delete"})
+
+class File(SQLModel, table=True):
+    id: str = Field(default_factory=aidx.genAidx, primary_key=True)
+    hash: str = Field(unique=True)
+    url: str
+    isSensitive: bool
+    fileSize: Optional[int] = None
+    fileType: str
+    caption: Optional[str] = None
+    createdAt: datetime = Field(default_factory=utcnow)
+    attachedNotes: List["Note"] = Relationship(back_populates="attachments", link_model="AttachmentsToNote")
+
+class Note(SQLModel, table=True):
+    id: str = Field(default_factory=aidx.genAidx, primary_key=True)
+    content: Optional[str] = None
+    content_html: Optional[str] = None
+    username: str
+    created_at: datetime = Field(default_factory=utcnow)
+    updated_at: datetime = Field(default_factory=utcnow)
+    visibility: str
+    visibleUserIds: List[str] = []
+    replyId: Optional[str] = None
+    renoteId: Optional[str] = None
+    channelId: Optional[str] = Field(default=None, foreign_key="channel.id")
+    channel: Optional["Channel"] = Relationship(back_populates="notes")
+    authorId: str = Field(foreign_key="user.id")
+    author: "User" = Relationship(back_populates="notes")
+    reactions: List["Reactions"] = Relationship(back_populates="note")
+    attachments: List["File"] = Relationship(back_populates="note")