setup peewee ORM

Author: imperosol

.gitignore

@@ -66,6 +66,7 @@ cover/
 *.pot
 
 local_settings.py
+*.sqlite
 *.sqlite3
 *.sqlite3-journal
 

db/__init__.py

@@ -0,0 +1,42 @@
+from datetime import UTC, datetime
+
+import peewee
+from peewee import Model, SqliteDatabase
+
+from src.settings import BASE_DIR
+
+# SQLite may not be suited for production, but is
+# perfectly for the initial development phase
+db = SqliteDatabase(BASE_DIR / "db.sqlite3")
+
+
+class DbBaseModel(Model):
+    created_at = peewee.DateTimeField(default=datetime.now(tz=UTC))
+    updated_at = peewee.DateTimeField()
+
+    def save(self, *args, **kwargs):
+        self.updated_at = datetime.now(tz=UTC)
+        return super().save(*args, **kwargs)
+
+    class Meta:
+        database = db
+
+
+class User(DbBaseModel):
+    discord_id = peewee.IntegerField(primary_key=True)
+    sith_id = peewee.IntegerField(null=True)
+    username = peewee.CharField()
+
+
+class Club(DbBaseModel):
+    name = peewee.CharField(max_length=64, unique=True)
+    category_id = peewee.IntegerField(unique=True)
+    sith_id = peewee.IntegerField(unique=True)
+    president_role_id = peewee.IntegerField(unique=True)
+    treasurer_role_id = peewee.IntegerField(unique=True)
+    member_role_id = peewee.IntegerField(unique=True)
+    former_member_role_id = peewee.IntegerField(unique=True)
+
+
+def init():
+    db.create_tables([User, Club])

db/models.py

@@ -2,5 +2,5 @@ services:
   discord:
     build: .
     volumes:
-      - /data:/app/data
+      - ./db.sqlite3:/app/db.sqlite3
       - ./log:/app/log

docker-compose.yml

@@ -7,6 +7,7 @@ requires-python = ">=3.12"
 dependencies = [
     "aiohttp>=3.13.0",
     "discord-py[speed]>=2.6.4",
+    "peewee>=3.18.2",
     "pydantic-settings>=2.11.0",
     "pydantic>=2.12.0",
     "pytz>=2025.2",