mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-12-05 01:04:21 +00:00
34 lines
906 B
Python
34 lines
906 B
Python
|
from sqlalchemy import *
|
||
|
|
from sqlalchemy.orm import *
|
||
|
|
from sqlalchemy.ext.declarative import declared_attr
|
||
|
|
from ..types import PollMood
|
||
|
|
from sqlalchemy.dialects.postgresql import UUID
|
||
|
|
|
||
|
|
|
||
|
|
class PollVote:
|
||
|
|
__tablename__ = "pollvotes"
|
||
|
|
|
||
|
|
@declared_attr
|
||
|
|
def caster_id(self):
|
||
|
|
return Column(Integer, ForeignKey("users.uid"), primary_key=True)
|
||
|
|
|
||
|
|
@declared_attr
|
||
|
|
def caster(self):
|
||
|
|
return relationship("User", backref=backref("poll_votes_cast"))
|
||
|
|
|
||
|
|
@declared_attr
|
||
|
|
def poll_id(self):
|
||
|
|
return Column(UUID(as_uuid=True), ForeignKey("polls.id"), primary_key=True)
|
||
|
|
|
||
|
|
@declared_attr
|
||
|
|
def poll(self):
|
||
|
|
return relationship("Poll", backref=backref("votes"))
|
||
|
|
|
||
|
|
@declared_attr
|
||
|
|
def posted(self):
|
||
|
|
return Column(DateTime, nullable=False)
|
||
|
|
|
||
|
|
@declared_attr
|
||
|
|
def vote(self):
|
||
|
|
return Column(Enum(PollMood), nullable=False, default=PollMood.NEUTRAL)
|