mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
import typing
|
|
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.orm import backref, relationship
|
|
|
|
from aurweb import schema, time
|
|
from aurweb.models.declarative import Base
|
|
from aurweb.models.user import User as _User
|
|
|
|
|
|
class TUVoteInfo(Base):
|
|
__table__ = schema.TU_VoteInfo
|
|
__tablename__ = __table__.name
|
|
__mapper_args__ = {"primary_key": [__table__.c.ID]}
|
|
|
|
Submitter = relationship(
|
|
_User,
|
|
backref=backref("tu_voteinfo_set", lazy="dynamic"),
|
|
foreign_keys=[__table__.c.SubmitterID],
|
|
)
|
|
|
|
def __init__(self, **kwargs):
|
|
# Default Quorum, Yes, No and Abstain columns to 0.
|
|
for col in ("Quorum", "Yes", "No", "Abstain"):
|
|
if col not in kwargs:
|
|
kwargs.update({col: 0})
|
|
|
|
super().__init__(**kwargs)
|
|
|
|
if self.Agenda is None:
|
|
raise IntegrityError(
|
|
statement="Column Agenda cannot be null.",
|
|
orig="TU_VoteInfo.Agenda",
|
|
params=("NULL"),
|
|
)
|
|
|
|
if self.User is None:
|
|
raise IntegrityError(
|
|
statement="Column User cannot be null.",
|
|
orig="TU_VoteInfo.User",
|
|
params=("NULL"),
|
|
)
|
|
|
|
if self.Submitted is None:
|
|
raise IntegrityError(
|
|
statement="Column Submitted cannot be null.",
|
|
orig="TU_VoteInfo.Submitted",
|
|
params=("NULL"),
|
|
)
|
|
|
|
if self.End is None:
|
|
raise IntegrityError(
|
|
statement="Column End cannot be null.",
|
|
orig="TU_VoteInfo.End",
|
|
params=("NULL"),
|
|
)
|
|
|
|
if not self.Submitter:
|
|
raise IntegrityError(
|
|
statement="Foreign key SubmitterID cannot be null.",
|
|
orig="TU_VoteInfo.SubmitterID",
|
|
params=("NULL"),
|
|
)
|
|
|
|
def __setattr__(self, key: str, value: typing.Any):
|
|
"""Customize setattr to stringify any Quorum keys given."""
|
|
if key == "Quorum":
|
|
value = str(value)
|
|
return super().__setattr__(key, value)
|
|
|
|
def __getattribute__(self, key: str):
|
|
"""Customize getattr to floatify any fetched Quorum values."""
|
|
attr = super().__getattribute__(key)
|
|
if key == "Quorum":
|
|
return float(attr)
|
|
return attr
|
|
|
|
def is_running(self):
|
|
return self.End > time.utcnow()
|
|
|
|
def total_votes(self):
|
|
return self.Yes + self.No + self.Abstain
|