generated from Luis/nextjs-python-web-template
112 lines
3.2 KiB
Python
112 lines
3.2 KiB
Python
from sqlalchemy import (
|
|
JSON,
|
|
Boolean,
|
|
Column,
|
|
DateTime,
|
|
ForeignKey,
|
|
Integer,
|
|
String,
|
|
Text,
|
|
UniqueConstraint,
|
|
)
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.sql import func
|
|
|
|
from .sql_db import Base
|
|
|
|
# Relationsship example
|
|
# https://dev.to/freddiemazzilli/flask-sqlalchemy-relationships-exploring-relationship-associations-igo
|
|
|
|
|
|
class Entity(Base):
|
|
__tablename__ = "entities"
|
|
|
|
## Queryable body ##
|
|
did = Column(String, primary_key=True, index=True)
|
|
name = Column(String, index=True)
|
|
ip = Column(String, index=True)
|
|
attached = Column(Boolean, index=True)
|
|
visible = Column(Boolean, index=True)
|
|
|
|
## Non queryable body ##
|
|
# In here we deposit: Network, Roles, Visible, etc.
|
|
other = Column(JSON)
|
|
|
|
## Relations ##
|
|
services = relationship("Service", back_populates="entity")
|
|
clients = relationship("Client", back_populates="entity")
|
|
repository = relationship("Repository", back_populates="entity")
|
|
# TODO maby refactor to repositories
|
|
|
|
|
|
class ServiceAbstract(Base):
|
|
__abstract__ = True
|
|
|
|
# Queryable body
|
|
uuid = Column(Text(length=36), primary_key=True, index=True)
|
|
service_name = Column(String, index=True)
|
|
service_type = Column(String, index=True)
|
|
endpoint_url = Column(String, index=True)
|
|
status = Column(String, index=True)
|
|
|
|
## Non queryable body ##
|
|
# In here we deposit: Action
|
|
other = Column(JSON)
|
|
|
|
|
|
class Service(ServiceAbstract):
|
|
__tablename__ = "services"
|
|
|
|
# Usage is the clients column
|
|
|
|
## Relations ##
|
|
# One entity can have many services
|
|
entity = relationship("Entity", back_populates="services")
|
|
entity_did = Column(String, ForeignKey("entities.did"))
|
|
|
|
# One service has many clients
|
|
clients = relationship("Client", back_populates="service")
|
|
|
|
|
|
class Client(Base):
|
|
__tablename__ = "clients"
|
|
|
|
## Queryable body ##
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
|
|
## Non queryable body ##
|
|
other = Column(JSON)
|
|
|
|
## Relations ##
|
|
# one entity can have many clients
|
|
entity = relationship("Entity", back_populates="clients")
|
|
entity_did = Column(String, ForeignKey("entities.did"))
|
|
|
|
# one client has one service
|
|
service = relationship("Service", back_populates="clients")
|
|
service_uuid = Column(String, ForeignKey("services.uuid"))
|
|
|
|
__table_args__ = (UniqueConstraint("service_uuid", "entity_did"),)
|
|
|
|
|
|
class Repository(ServiceAbstract):
|
|
__tablename__ = "repositories"
|
|
|
|
time_created = Column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
# one repository has one entity
|
|
entity = relationship("Entity", back_populates="repository")
|
|
entity_did = Column(Integer, ForeignKey("entities.did"))
|
|
|
|
|
|
# TODO: Ask how this works exactly
|
|
class Resolution(Base):
|
|
__tablename__ = "resolutions"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
requester_name = Column(String, index=True)
|
|
requester_did = Column(String, index=True)
|
|
resolved_did = Column(String, index=True)
|
|
other = Column(JSON)
|
|
timestamp = Column(DateTime(timezone=True), server_default=func.now())
|