mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
446a082352
We don't want to depend on the database to load up data about the models we define. We now leverage the existing `aurweb.schema` module for table definitions and set __table_args__["autoload"] to False. Signed-off-by: Kevin Morris <kevr@0cost.org>
40 lines
1.4 KiB
Python
40 lines
1.4 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.license import License as _License
|
|
from aurweb.models.package import Package as _Package
|
|
|
|
|
|
class PackageLicense(Base):
|
|
__table__ = schema.PackageLicenses
|
|
__tablename__ = __table__.name
|
|
__mapper_args__ = {
|
|
"primary_key": [__table__.c.PackageID, __table__.c.LicenseID]
|
|
}
|
|
|
|
Package = relationship(
|
|
_Package, backref=backref("package_licenses", lazy="dynamic",
|
|
cascade="all, delete"),
|
|
foreign_keys=[__table__.c.PackageID])
|
|
|
|
License = relationship(
|
|
_License, backref=backref("package_licenses", lazy="dynamic",
|
|
cascade="all, delete"),
|
|
foreign_keys=[__table__.c.LicenseID])
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
if not self.Package and not self.PackageID:
|
|
raise IntegrityError(
|
|
statement="Primary key PackageID cannot be null.",
|
|
orig="PackageLicenses.PackageID",
|
|
params=("NULL"))
|
|
|
|
if not self.License and not self.LicenseID:
|
|
raise IntegrityError(
|
|
statement="Primary key LicenseID cannot be null.",
|
|
orig="PackageLicenses.LicenseID",
|
|
params=("NULL"))
|