69 lines
3.4 KiB
Python
69 lines
3.4 KiB
Python
|
import uuid
|
||
|
import os
|
||
|
|
||
|
import dateutil.relativedelta
|
||
|
from datetime import datetime
|
||
|
from sqlalchemy import Enum, UniqueConstraint
|
||
|
from sqlalchemy.dialects.postgresql import UUID
|
||
|
|
||
|
from loguru import logger
|
||
|
from overseer.flask import db, SQLITE_MODE
|
||
|
|
||
|
uuid_column_type = lambda: UUID(as_uuid=True) if not SQLITE_MODE else db.String(36)
|
||
|
|
||
|
|
||
|
class Guarantee(db.Model):
|
||
|
__tablename__ = "guarantees"
|
||
|
id = db.Column(db.Integer, primary_key=True)
|
||
|
guarantor_id = db.Column(db.Integer, db.ForeignKey("instances.id", ondelete="CASCADE"), nullable=False)
|
||
|
guarantor_instance = db.relationship("Instance", back_populates="guarantees", foreign_keys=[guarantor_id])
|
||
|
guaranteed_id = db.Column(db.Integer, db.ForeignKey("instances.id", ondelete="CASCADE"), unique=True, nullable=False)
|
||
|
guaranteed_instance = db.relationship("Instance", back_populates="guarantors", foreign_keys=[guaranteed_id])
|
||
|
|
||
|
|
||
|
class Endorsement(db.Model):
|
||
|
__tablename__ = "endorsements"
|
||
|
__table_args__ = (UniqueConstraint('approving_id', 'endorsed_id', name='endoresements_approving_id_endorsed_id'),)
|
||
|
id = db.Column(db.Integer, primary_key=True)
|
||
|
approving_id = db.Column(db.Integer, db.ForeignKey("instances.id", ondelete="CASCADE"), nullable=False)
|
||
|
approving_instance = db.relationship("Instance", back_populates="approvals", foreign_keys=[approving_id])
|
||
|
endorsed_id = db.Column(db.Integer, db.ForeignKey("instances.id", ondelete="CASCADE"), nullable=False)
|
||
|
endorsed_instance = db.relationship("Instance", back_populates="endorsements", foreign_keys=[endorsed_id])
|
||
|
|
||
|
|
||
|
class Instance(db.Model):
|
||
|
__tablename__ = "instances"
|
||
|
|
||
|
id = db.Column(db.Integer, primary_key=True)
|
||
|
domain = db.Column(db.String(255), unique=True, nullable=False)
|
||
|
api_key = db.Column(db.String(100), unique=True, nullable=False, index=True)
|
||
|
created = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||
|
updated = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||
|
|
||
|
open_registrations = db.Column(db.Boolean, unique=False, nullable=False, index=True)
|
||
|
email_verify = db.Column(db.Boolean, unique=False, nullable=False, index=True)
|
||
|
|
||
|
approvals = db.relationship("Endorsement", back_populates="approving_instance", cascade="all, delete-orphan", foreign_keys=[Endorsement.approving_id])
|
||
|
endorsements = db.relationship("Endorsement", back_populates="endorsed_instance", cascade="all, delete-orphan", foreign_keys=[Endorsement.endorsed_id])
|
||
|
guarantees = db.relationship("Guarantee", back_populates="guarantor_instance", cascade="all, delete-orphan", foreign_keys=[Guarantee.guarantor_id])
|
||
|
guarantors = db.relationship("Guarantee", back_populates="guaranteed_instance", cascade="all, delete-orphan", foreign_keys=[Guarantee.guaranteed_id])
|
||
|
|
||
|
def create(self):
|
||
|
db.session.add(self)
|
||
|
db.session.commit()
|
||
|
|
||
|
def get_details(self):
|
||
|
ret_dict = {
|
||
|
"domain": self.domain,
|
||
|
"open_registrations": self.open_registrations,
|
||
|
"email_verify": self.email_verify,
|
||
|
"endorsements": len(self.endorsements),
|
||
|
"approvals": len(self.approvals),
|
||
|
"guarantor": self.get_guarantor().domain,
|
||
|
}
|
||
|
return ret_dict
|
||
|
|
||
|
def get_guarantor(self):
|
||
|
guarantee = self.guarantors[0]
|
||
|
return Instance.query.filter_by(id=guarantee.guarantor_id).first()
|