mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
d2d47254b4
TU_VoteInfo -> VoteInfo TU_Votes -> Votes TU_VoteInfo.ActiveTUs -> VoteInfo.ActiveUsers script: tuvotereminder -> votereminder Signed-off-by: moson <moson@archlinux.org>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.orm import backref, relationship
|
|
|
|
from aurweb import schema
|
|
from aurweb.models.declarative import Base
|
|
from aurweb.models.user import User as _User
|
|
from aurweb.models.voteinfo import VoteInfo as _VoteInfo
|
|
|
|
|
|
class Vote(Base):
|
|
__table__ = schema.Votes
|
|
__tablename__ = __table__.name
|
|
__mapper_args__ = {"primary_key": [__table__.c.VoteID, __table__.c.UserID]}
|
|
|
|
VoteInfo = relationship(
|
|
_VoteInfo,
|
|
backref=backref("votes", lazy="dynamic"),
|
|
foreign_keys=[__table__.c.VoteID],
|
|
)
|
|
|
|
User = relationship(
|
|
_User,
|
|
backref=backref("votes", lazy="dynamic"),
|
|
foreign_keys=[__table__.c.UserID],
|
|
)
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
if not self.VoteInfo and not self.VoteID:
|
|
raise IntegrityError(
|
|
statement="Foreign key VoteID cannot be null.",
|
|
orig="Votes.VoteID",
|
|
params=("NULL"),
|
|
)
|
|
|
|
if not self.User and not self.UserID:
|
|
raise IntegrityError(
|
|
statement="Foreign key UserID cannot be null.",
|
|
orig="Votes.UserID",
|
|
params=("NULL"),
|
|
)
|