mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
888cf5118a
This rewrites the entire model base as declarative models. This allows us to more easily customize overlay fields in tables and is more common. This effort also brought some DB violations to light which this commit addresses. Signed-off-by: Kevin Morris <kevr@0cost.org>
58 lines
2 KiB
Python
58 lines
2 KiB
Python
from sqlalchemy import Column, ForeignKey, Integer
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.orm import backref, relationship
|
|
|
|
import aurweb.db
|
|
import aurweb.models.package
|
|
import aurweb.models.relation_type
|
|
|
|
from aurweb.models.declarative import Base
|
|
|
|
|
|
class PackageRelation(Base):
|
|
__tablename__ = "PackageRelations"
|
|
|
|
PackageID = Column(
|
|
Integer, ForeignKey("Packages.ID", ondelete="CASCADE"),
|
|
nullable=False)
|
|
Package = relationship(
|
|
"Package", backref=backref("package_relations", lazy="dynamic"),
|
|
foreign_keys=[PackageID])
|
|
|
|
RelTypeID = Column(
|
|
Integer, ForeignKey("RelationTypes.ID", ondelete="CASCADE"),
|
|
nullable=False)
|
|
RelationType = relationship(
|
|
"RelationType", backref=backref("package_relations", lazy="dynamic"),
|
|
foreign_keys=[RelTypeID])
|
|
|
|
__mapper_args__ = {"primary_key": [PackageID, RelTypeID]}
|
|
|
|
def __init__(self,
|
|
Package: aurweb.models.package.Package = None,
|
|
RelationType: aurweb.models.relation_type.RelationType = None,
|
|
RelName: str = None, RelCondition: str = None,
|
|
RelArch: str = None):
|
|
self.Package = Package
|
|
if not self.Package:
|
|
raise IntegrityError(
|
|
statement="Foreign key PackageID cannot be null.",
|
|
orig="PackageRelations.PackageID",
|
|
params=("NULL"))
|
|
|
|
self.RelationType = RelationType
|
|
if not self.RelationType:
|
|
raise IntegrityError(
|
|
statement="Foreign key RelTypeID cannot be null.",
|
|
orig="PackageRelations.RelTypeID",
|
|
params=("NULL"))
|
|
|
|
self.RelName = RelName # nullable=False
|
|
if not self.RelName:
|
|
raise IntegrityError(
|
|
statement="Column RelName cannot be null.",
|
|
orig="PackageRelations.RelName",
|
|
params=("NULL"))
|
|
|
|
self.RelCondition = RelCondition
|
|
self.RelArch = RelArch
|