Support new Lock and Lockdown methods

jarvis_core/db/models/__init__.py

@@ -51,6 +51,15 @@ class Guess(Document):
     user: int = fields.IntegerField(required=True)
 
 
+@JARVIS_INST.register
+class Permission(EmbeddedDocument):
+    """Embedded Permissions document."""
+
+    id: int = fields.IntegerField(required=True)
+    allow: int = fields.IntegerField(default=0)
+    deny: int = fields.IntegerField(default=0)
+
+
 @JARVIS_INST.register
 class Lock(Document):
     """Lock database object."""
@@ -61,6 +70,20 @@ class Lock(Document):
     duration: int = fields.IntegerField(default=10)
     guild: int = fields.IntegerField(required=True)
     reason: str = fields.StringField(required=True)
+    original_perms: Permission = fields.EmbeddedField(Permission, required=False)
+    created_at: datetime = fields.DateTimeField(default=datetime.utcnow)
+
+
+@JARVIS_INST.register
+class Lockdown(Document):
+    """Lockdown database object."""
+
+    active: bool = fields.BooleanField(default=True)
+    admin: int = fields.IntegerField(required=True)
+    duration: int = fields.IntegerField(default=10)
+    guild: int = fields.IntegerField(required=True)
+    reason: str = fields.StringField(required=True)
+    original_perms: int = fields.IntegerField(required=True)
     created_at: datetime = fields.DateTimeField(default=datetime.utcnow)
 
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "jarvis-core"
-version = "0.6.1"
+version = "0.7.0"
 description = ""
 authors = ["Your Name <you@example.com>"]