feat: Add VoteMod, ref jarvis-bot#178

jarvis_core/db/models/__init__.py

@@ -9,6 +9,7 @@ from jarvis_core.db.models.actions import Ban, Kick, Mute, Unban, Warning
 from jarvis_core.db.models.modlog import Action, Modlog, Note
 from jarvis_core.db.models.reddit import Subreddit, SubredditFollow
 from jarvis_core.db.models.twitter import TwitterAccount, TwitterFollow
+from jarvis_core.db.models.votemod import Karma, Vote, VoteMod
 from jarvis_core.db.utils import NowField
 
 __all__ = [
@@ -42,6 +43,9 @@ __all__ = [
     "Unban",
     "UserSetting",
     "Warning",
+    "Karma",
+    "Vote",
+    "VoteMod",
     "all_models",
 ]
 
@@ -246,6 +250,7 @@ all_models = [
     Config,
     Filter,
     Guess,
+    Karma,
     Kick,
     Lock,
     Lockdown,
@@ -266,5 +271,7 @@ all_models = [
     TwitterFollow,
     Unban,
     UserSetting,
+    Vote,
+    VoteMod,
     Warning,
 ]

jarvis_core/db/models/votemod.py

@@ -0,0 +1,26 @@
+"""VoteMod models."""
+from typing import Optional
+
+from beanie import Document
+from pydantic import BaseModel
+
+
+class Vote(BaseModel):
+    user: int
+    score: int
+
+
+class Karma(Document):
+    user: int
+    karma: int = 1
+    revoked: bool = False
+
+
+class VoteMod(Document):
+    initiator: int
+    target: int
+    message: int
+    content: str
+    attachments: list[str]
+    score: int
+    votes: list[Vote]