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>
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
from sqlalchemy import Column, ForeignKey, Integer
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.orm import backref, relationship
|
|
|
|
import aurweb.models.package
|
|
|
|
from aurweb.models import dependency_type
|
|
from aurweb.models.declarative import Base
|
|
|
|
|
|
class PackageDependency(Base):
|
|
__tablename__ = "PackageDepends"
|
|
|
|
PackageID = Column(
|
|
Integer, ForeignKey("Packages.ID", ondelete="CASCADE"),
|
|
nullable=False)
|
|
Package = relationship(
|
|
"Package", backref=backref("package_dependencies", lazy="dynamic"),
|
|
foreign_keys=[PackageID], lazy="select")
|
|
|
|
DepTypeID = Column(
|
|
Integer, ForeignKey("DependencyTypes.ID", ondelete="NO ACTION"),
|
|
nullable=False)
|
|
DependencyType = relationship(
|
|
"DependencyType",
|
|
backref=backref("package_dependencies", lazy="dynamic"),
|
|
foreign_keys=[DepTypeID], lazy="select")
|
|
|
|
__mapper_args__ = {"primary_key": [PackageID, DepTypeID]}
|
|
|
|
def __init__(self,
|
|
Package: aurweb.models.package.Package = None,
|
|
DependencyType: dependency_type.DependencyType = None,
|
|
DepName: str = None,
|
|
DepDesc: str = None,
|
|
DepCondition: str = None,
|
|
DepArch: str = None):
|
|
self.Package = Package
|
|
if not self.Package:
|
|
raise IntegrityError(
|
|
statement="Foreign key PackageID cannot be null.",
|
|
orig="PackageDependencies.PackageID",
|
|
params=("NULL"))
|
|
|
|
self.DependencyType = DependencyType
|
|
if not self.DependencyType:
|
|
raise IntegrityError(
|
|
statement="Foreign key DepTypeID cannot be null.",
|
|
orig="PackageDependencies.DepTypeID",
|
|
params=("NULL"))
|
|
|
|
self.DepName = DepName
|
|
if not self.DepName:
|
|
raise IntegrityError(
|
|
statement="Column DepName cannot be null.",
|
|
orig="PackageDependencies.DepName",
|
|
params=("NULL"))
|
|
|
|
self.DepDesc = DepDesc
|
|
self.DepCondition = DepCondition
|
|
self.DepArch = DepArch
|