68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
from typing import Optional, List
|
|
|
|
from tldextract import extract
|
|
|
|
from app import db
|
|
from app.models import AbstractConfiguration, AbstractResource
|
|
from app.models.onions import Onion
|
|
|
|
|
|
class Origin(AbstractConfiguration):
|
|
group_id = db.Column(db.Integer, db.ForeignKey("group.id"), nullable=False)
|
|
domain_name = db.Column(db.String(255), unique=True, nullable=False)
|
|
auto_rotation = db.Column(db.Boolean, nullable=False)
|
|
|
|
group = db.relationship("Group", back_populates="origins")
|
|
mirrors = db.relationship("Mirror", back_populates="origin")
|
|
proxies = db.relationship("Proxy", back_populates="origin")
|
|
alarms = db.relationship("Alarm", back_populates="origin")
|
|
|
|
@classmethod
|
|
def csv_header(cls) -> List[str]:
|
|
return super().csv_header() + [
|
|
"group_id", "domain_name"
|
|
]
|
|
|
|
def destroy(self) -> None:
|
|
super().destroy()
|
|
for proxy in self.proxies:
|
|
proxy.destroy()
|
|
|
|
def onion(self) -> Optional[str]:
|
|
tld = extract(self.domain_name).registered_domain
|
|
onion = Onion.query.filter(Onion.domain_name == tld).first()
|
|
if not onion:
|
|
return None
|
|
domain_name: str = self.domain_name
|
|
return domain_name.replace(tld, f"{onion.onion_name}")
|
|
|
|
|
|
class Proxy(AbstractResource):
|
|
origin_id = db.Column(db.Integer, db.ForeignKey("origin.id"), nullable=False)
|
|
provider = db.Column(db.String(20), nullable=False)
|
|
psg = db.Column(db.Integer, nullable=True)
|
|
slug = db.Column(db.String(20), nullable=True)
|
|
terraform_updated = db.Column(db.DateTime(), nullable=True)
|
|
url = db.Column(db.String(255), nullable=True)
|
|
|
|
origin = db.relationship("Origin", back_populates="proxies")
|
|
alarms = db.relationship("Alarm", back_populates="proxy")
|
|
|
|
@classmethod
|
|
def csv_header(cls) -> List[str]:
|
|
return super().csv_header() + [
|
|
"origin_id", "provider", "psg", "slug", "terraform_updated", "url"
|
|
]
|
|
|
|
|
|
class Mirror(AbstractResource):
|
|
origin_id = db.Column(db.Integer, db.ForeignKey("origin.id"), nullable=False)
|
|
url = db.Column(db.String(255), unique=True, nullable=False)
|
|
|
|
origin = db.relationship("Origin", back_populates="mirrors")
|
|
|
|
@classmethod
|
|
def csv_header(cls) -> List[str]:
|
|
return super().csv_header() + [
|
|
"origin_id", "url"
|
|
]
|