From b81ea66d12b1a32c8291cbe9e14fe17b8e020d08 Mon Sep 17 00:00:00 2001 From: zevaryx Date: Sun, 20 Mar 2022 12:53:17 -0600 Subject: [PATCH] Support new Lock and Lockdown methods --- jarvis_core/db/models/__init__.py | 23 +++++++++++++++++++++++ pyproject.toml | 2 +- 2 files changed, 24 insertions(+), 1 deletion(-) diff --git a/jarvis_core/db/models/__init__.py b/jarvis_core/db/models/__init__.py index 901e034..3128bc3 100644 --- a/jarvis_core/db/models/__init__.py +++ b/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) diff --git a/pyproject.toml b/pyproject.toml index 7a2e5d2..8c993cc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "jarvis-core" -version = "0.6.1" +version = "0.7.0" description = "" authors = ["Your Name "]