mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
51320ab22a
Now, we allow the direct relationships and their foreign keys to be set in all of our models. Previously, we constrained this to direct relationships, and this forced users to perform a query in most situations to satisfy that requirement. Now, IDs can be passed directly. Additionally, this change removes the need for extraneous imports when users which to use relationships. We now import and use models directly instead of passing string-references to them. Signed-off-by: Kevin Morris <kevr@0cost.org>
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import Column, ForeignKey, Integer
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.orm import backref, relationship
|
|
|
|
from aurweb.models.declarative import Base
|
|
from aurweb.models.user import User as _User
|
|
|
|
|
|
class PackageBase(Base):
|
|
__tablename__ = "PackageBases"
|
|
|
|
FlaggerUID = Column(Integer,
|
|
ForeignKey("Users.ID", ondelete="SET NULL"))
|
|
Flagger = relationship(
|
|
_User, backref=backref("flagged_bases", lazy="dynamic"),
|
|
foreign_keys=[FlaggerUID])
|
|
|
|
SubmitterUID = Column(Integer,
|
|
ForeignKey("Users.ID", ondelete="SET NULL"))
|
|
Submitter = relationship(
|
|
_User, backref=backref("submitted_bases", lazy="dynamic"),
|
|
foreign_keys=[SubmitterUID])
|
|
|
|
MaintainerUID = Column(Integer,
|
|
ForeignKey("Users.ID", ondelete="SET NULL"))
|
|
Maintainer = relationship(
|
|
_User, backref=backref("maintained_bases", lazy="dynamic"),
|
|
foreign_keys=[MaintainerUID])
|
|
|
|
PackagerUID = Column(Integer, ForeignKey("Users.ID", ondelete="SET NULL"))
|
|
Packager = relationship(
|
|
_User, backref=backref("package_bases", lazy="dynamic"),
|
|
foreign_keys=[PackagerUID])
|
|
|
|
# A set used to check for floatable values.
|
|
TO_FLOAT = {"Popularity"}
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
if self.Name is None:
|
|
raise IntegrityError(
|
|
statement="Column Name cannot be null.",
|
|
orig="PackageBases.Name",
|
|
params=("NULL"))
|
|
|
|
# If no SubmittedTS/ModifiedTS is provided on creation, set them
|
|
# here to the current utc timestamp.
|
|
now = datetime.utcnow().timestamp()
|
|
if not self.SubmittedTS:
|
|
self.SubmittedTS = now
|
|
if not self.ModifiedTS:
|
|
self.ModifiedTS = now
|
|
|
|
if not self.FlaggerComment:
|
|
self.FlaggerComment = str()
|
|
|
|
def __getattribute__(self, key: str):
|
|
attr = super().__getattribute__(key)
|
|
if key in PackageBase.TO_FLOAT and not isinstance(attr, float):
|
|
return float(attr)
|
|
return attr
|