add PackageSource SQLAlchemy ORM model

Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris 2021-06-10 17:46:29 -07:00
parent d18cfad63e
commit 11c4926502
2 changed files with 75 additions and 0 deletions

View file

@ -0,0 +1,31 @@
from sqlalchemy import Column, ForeignKey, Integer
from sqlalchemy.exc import IntegrityError
from sqlalchemy.orm import backref, relationship
import aurweb.models.package
from aurweb.models.declarative import Base
class PackageSource(Base):
__tablename__ = "PackageSources"
PackageID = Column(Integer, ForeignKey("Packages.ID", ondelete="CASCADE"),
nullable=False)
Package = relationship(
"Package", backref=backref("package_sources", lazy="dynamic"),
foreign_keys=[PackageID])
__mapper_args__ = {"primary_key": [PackageID]}
def __init__(self,
Package: aurweb.models.package.Package = None,
**kwargs):
super().__init__(**kwargs)
self.Package = Package
if not self.Package:
raise IntegrityError(
statement="Foreign key PackageID cannot be null.",
orig="PackageSources.PackageID",
params=("NULL"))