35 lines
1 KiB
Python
35 lines
1 KiB
Python
"""Twitter database models."""
|
|
from datetime import datetime
|
|
|
|
from umongo import Document, fields
|
|
|
|
from jarvis_core.db import JARVIS_INST
|
|
|
|
|
|
@JARVIS_INST.register
|
|
class TwitterAccount(Document):
|
|
"""Twitter Account object."""
|
|
|
|
handle = fields.StringField(required=True)
|
|
twitter_id = fields.IntegerField(required=True)
|
|
last_tweet = fields.IntegerField(required=True)
|
|
last_sync = fields.DateTimeField(default=datetime.utcnow)
|
|
|
|
class Meta:
|
|
collection_name = "twitteraccount"
|
|
|
|
|
|
@JARVIS_INST.register
|
|
class TwitterFollow(Document):
|
|
"""Twitter Follow object."""
|
|
|
|
active = fields.BooleanField(default=True)
|
|
twitter_id = fields.IntegerField(required=True)
|
|
channel = fields.IntegerField(required=True)
|
|
guild = fields.IntegerField(required=True)
|
|
retweets = fields.BooleanField(default=True)
|
|
admin = fields.IntegerField(required=True)
|
|
created_at = fields.DateTimeField(default=datetime.utcnow)
|
|
|
|
class Meta:
|
|
collection_name = "twitterfollow"
|