jarvis-core/jarvis_core/db/models/actions.py

70 lines
2.2 KiB
Python

"""User action models."""
from datetime import datetime
from umongo import Document, fields
from jarvis_core.db import JARVIS_INST
@JARVIS_INST.register
class Ban(Document):
active = fields.BooleanField(default=True)
admin = fields.IntegerField(required=True)
user = fields.IntegerField(required=True)
username = fields.StringField(required=True)
discrim = fields.IntegerField(required=True)
duration = fields.IntegerField(required=False)
guild = fields.IntegerField(required=True)
type = fields.StringField(default="perm")
reason = fields.StringField(required=True)
created_at = fields.DateTimeField(default=datetime.utcnow)
@JARVIS_INST.register
class Kick(Document):
"""Kick database object."""
admin = fields.IntegerField(required=True)
guild = fields.IntegerField(required=True)
reason = fields.StringField(required=True)
user = fields.IntegerField(required=True)
created_at = fields.DateTimeField(default=datetime.utcnow)
@JARVIS_INST.register
class Mute(Document):
"""Mute database object."""
active = fields.BooleanField(default=True)
user = fields.IntegerField(required=True)
admin = fields.IntegerField(required=True)
duration = fields.IntegerField(default=10)
guild = fields.IntegerField(required=True)
reason = fields.StringField(required=True)
created_at = fields.DateTimeField(default=datetime.utcnow)
@JARVIS_INST.register
class Unban(Document):
"""Unban database object."""
user = fields.IntegerField(required=True)
username = fields.StringField(required=True)
discrim = fields.IntegerField(required=True)
guild = fields.IntegerField(required=True)
admin = fields.IntegerField(required=True)
reason = fields.StringField(required=True)
created_at = fields.DateTimeField(default=datetime.utcnow)
@JARVIS_INST.register
class Warning(Document):
"""Warning database object."""
active = fields.BooleanField(default=True)
admin = fields.IntegerField(required=True)
user = fields.IntegerField(required=True)
guild = fields.IntegerField(required=True)
duration = fields.IntegerField(default=24)
reason = fields.StringField(required=True)
created_at = fields.DateTimeField(default=datetime.utcnow)