mmisp-lib 0.1.0__py3-none-any.whl

mmisp/db/models/attribute.py

@@ -0,0 +1,92 @@
+from typing import Self, Type
+
+from sqlalchemy import BigInteger, Boolean, Column, ForeignKey, Integer, String, Text, inspect
+from sqlalchemy.ext.hybrid import hybrid_property
+from sqlalchemy.orm import relationship
+from sqlalchemy.orm.decl_api import DeclarativeMeta
+
+from mmisp.lib.attributes import categories, default_category, mapper_safe_clsname_val, to_ids
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+from .event import Event
+from .tag import Tag
+
+
+class DictMixin:
+    def asdict(self: Self) -> dict:
+        d = {}
+        for key in self.__mapper__.c.keys():
+            if not key.startswith("_"):
+                d[key] = getattr(self, key)
+
+        for key, prop in inspect(self.__class__).all_orm_descriptors.items():
+            if isinstance(prop, hybrid_property):
+                d[key] = getattr(self, key)
+        return d
+
+
+class Attribute(Base, DictMixin):
+    __tablename__ = "attributes"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(40), unique=True, default=uuid, index=True)
+    event_id = Column(Integer, ForeignKey("events.id", ondelete="CASCADE"), nullable=False, index=True)
+    object_id = Column(Integer, nullable=False, default=0, index=True)
+    object_relation = Column(String(255), index=True)
+    category = Column(String(255), nullable=False, index=True)
+    type = Column(String(100), nullable=False, index=True)
+    value1 = Column(Text, nullable=False)
+    value2 = Column(Text, nullable=False, default="")
+    to_ids = Column(Boolean, default=True, nullable=False)
+    timestamp = Column(Integer, nullable=False, default=0)
+    distribution = Column(Integer, nullable=False, default=0)
+    sharing_group_id = Column(Integer, index=True, default=0)
+    comment = Column(Text)
+    deleted = Column(Boolean, nullable=False, default=False)
+    disable_correlation = Column(Boolean, nullable=False, default=False)
+    first_seen = Column(BigInteger, index=True)
+    last_seen = Column(BigInteger, index=True)
+
+    event = relationship("Event", back_populates="attributes", lazy="joined")
+
+    @property
+    def event_uuid(self: "Attribute") -> str:
+        return self.event.uuid
+
+    @hybrid_property
+    def value(self: Self) -> str:
+        if self.value2 == "":
+            return self.value1
+        return f"{self.value1}|{self.value2}"
+
+    @value.setter
+    def value(self: Self, value: str) -> None:
+        split = value.split("|", 1)
+        self.value1 = split[0]
+        if len(split) == 2:
+            self.value2 = split[1]
+
+
+class AttributeTag(Base):
+    __tablename__ = "attribute_tags"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    attribute_id = Column(Integer, ForeignKey(Attribute.id, ondelete="CASCADE"), nullable=False, index=True)
+    event_id = Column(Integer, ForeignKey(Event.id, ondelete="CASCADE"), nullable=False, index=True)
+    tag_id = Column(Integer, ForeignKey(Tag.id, ondelete="CASCADE"), nullable=False, index=True)
+    local = Column(Boolean, nullable=False, default=False)
+
+
+class AttributeMeta(DeclarativeMeta):
+    def __new__(cls: Type[type], clsname: str, bases: tuple, dct: dict) -> "AttributeMeta":
+        key = clsname[len("Attribute") :]
+        dct["default_category"] = default_category[mapper_safe_clsname_val[key]]
+        dct["categories"] = categories[mapper_safe_clsname_val[key]]
+        dct["to_ids"] = to_ids[mapper_safe_clsname_val[key]]
+        dct["__mapper_args__"] = {"polymorphic_identity": mapper_safe_clsname_val[key]}
+        return super().__new__(cls, clsname, bases, dct)
+
+
+for k, _ in mapper_safe_clsname_val.items():
+    vars()["Attribute" + k] = AttributeMeta("Attribute" + k, (Attribute,), dict())

mmisp/db/models/auth_key.py

@@ -0,0 +1,24 @@
+from time import time
+
+from sqlalchemy import Boolean, Column, ForeignKey, Integer, String
+
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+
+
+class AuthKey(Base):
+    __tablename__ = "auth_keys"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(255), unique=True, default=uuid, nullable=False)
+    authkey = Column(String(255), nullable=False)
+    authkey_start = Column(String(255), nullable=False)
+    authkey_end = Column(String(255), nullable=False)
+    created = Column(Integer, nullable=False, default=time)
+    expiration = Column(Integer, nullable=False, default=0)
+    read_only = Column(Boolean, nullable=False, default=0)
+    comment = Column(String(255))
+    allowed_ips = Column(String(255))
+    unique_ips = Column(String(255))
+    user_id = Column(Integer, ForeignKey("users.id"), nullable=False)

mmisp/db/models/event.py

@@ -0,0 +1,62 @@
+from datetime import datetime
+
+from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String, Text
+from sqlalchemy.orm import relationship
+
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+from .organisation import Organisation
+from .tag import Tag
+from .user import User
+
+
+class Event(Base):
+    __tablename__ = "events"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(40), unique=True, default=uuid, nullable=False, index=True)
+    org_id = Column(Integer, ForeignKey(Organisation.id), nullable=False, index=True)
+    date = Column(DateTime, default=datetime.utcnow, nullable=False)
+    info = Column(Text, nullable=False)
+    user_id = Column(Integer, ForeignKey(User.id), nullable=False)
+    published = Column(Boolean, nullable=False, default=False)
+    analysis = Column(Integer, nullable=False)
+    attribute_count = Column(Integer, default=0)
+    orgc_id = Column(Integer, ForeignKey(Organisation.id), nullable=False, index=True)
+    timestamp = Column(Integer, nullable=False, default=0)
+    distribution = Column(Integer, nullable=False, default=0)
+    sharing_group_id = Column(Integer, nullable=False, index=True, default=0)
+    proposal_email_lock = Column(Boolean, nullable=False, default=False)
+    locked = Column(Boolean, nullable=False, default=False)
+    threat_level_id = Column(Integer, nullable=False)
+    publish_timestamp = Column(Integer, nullable=False, default=0)
+    sighting_timestamp = Column(Integer, nullable=False, default=0)
+    disable_correlation = Column(Boolean, nullable=False, default=False)
+    extends_uuid = Column(String(40), default="", index=True)
+    protected = Column(Boolean)
+
+    attributes = relationship("Attribute", back_populates="event")
+
+
+class EventReport(Base):
+    __tablename__ = "event_reports"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(40), unique=True, nullable=False, default=uuid)
+    event_id = Column(Integer, ForeignKey(Event.id), nullable=False, index=True)
+    name = Column(String(255), nullable=False, index=True)
+    content = Column(Text)
+    distribution = Column(Integer, nullable=False, default=0)
+    sharing_group_id = Column(Integer)
+    timestamp = Column(Integer, nullable=False)
+    deleted = Column(Boolean, nullable=False, default=False)
+
+
+class EventTag(Base):
+    __tablename__ = "event_tags"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    event_id = Column(Integer, ForeignKey(Event.id, ondelete="CASCADE"), nullable=False, index=True)
+    tag_id = Column(Integer, ForeignKey(Tag.id, ondelete="CASCADE"), nullable=False, index=True)
+    local = Column(Boolean, nullable=False, default=False)

mmisp/db/models/feed.py

@@ -0,0 +1,32 @@
+from sqlalchemy import Boolean, Column, Integer, String, Text
+
+from mmisp.db.database import Base
+
+
+class Feed(Base):
+    __tablename__ = "feeds"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), nullable=False)
+    provider = Column(String(255), nullable=False)
+    url = Column(String(255), nullable=False)
+    rules = Column(Text, default=None)
+    enabled = Column(Boolean, default=False)
+    distribution = Column(Integer, nullable=False, default=0)
+    sharing_group_id = Column(Integer, nullable=False, default=0, index=True)
+    tag_id = Column(Integer, nullable=False, default=0)
+    default = Column(Boolean, default=False)
+    source_format = Column(String(255), default="misp")
+    fixed_event = Column(Boolean, nullable=False, default=False)
+    delta_merge = Column(Boolean, nullable=False, default=False)
+    event_id = Column(Integer, nullable=False, default=0)
+    publish = Column(Boolean, nullable=False, default=False)
+    override_ids = Column(Boolean, nullable=False, default=False)
+    settings = Column(Text)
+    input_source = Column(String(255), nullable=False, default="network", index=True)
+    delete_local_file = Column(Boolean, default=False)
+    lookup_visible = Column(Boolean, default=False)
+    headers = Column(Text)
+    caching_enabled = Column(Boolean, nullable=False, default=False)
+    force_to_ids = Column(Boolean, nullable=False, default=False)
+    orgc_id = Column(Integer, nullable=False, default=0, index=True)

mmisp/db/models/galaxy.py

@@ -0,0 +1,22 @@
+from sqlalchemy import Boolean, Column, Integer, String, Text
+
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+
+
+class Galaxy(Base):
+    __tablename__ = "galaxies"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(255), nullable=False, unique=True, default=uuid)
+    name = Column(String(255), nullable=False, default="", index=True)
+    type = Column(String(255), nullable=False, index=True)
+    description = Column(Text, nullable=False)
+    version = Column(String(255), nullable=False)
+    icon = Column(String(255), nullable=False, default="")
+    namespace = Column(String(255), nullable=False, default="misp", index=True)
+    kill_chain_order = Column(String(255))
+    """must be serialized"""
+    enabled = Column(Boolean, nullable=False, default=True)
+    local_only = Column(Boolean, default=False)

mmisp/db/models/galaxy_cluster.py

@@ -0,0 +1,52 @@
+from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, Text
+
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+from .galaxy import Galaxy
+
+
+class GalaxyCluster(Base):
+    __tablename__ = "galaxy_clusters"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(255), unique=True, default=uuid, index=True)
+    collection_uuid = Column(String(255), nullable=False, index=True, default="0")
+    type = Column(String(255), nullable=False, index=True)
+    value = Column(Text, nullable=False)
+    tag_name = Column(String(255), nullable=False, default="", index=True)
+    description = Column(Text, nullable=False)
+    galaxy_id = Column(Integer, ForeignKey(Galaxy.id, ondelete="CASCADE"), nullable=False, index=True)
+    source = Column(String(255), nullable=False, default="")
+    authors = Column(Text, nullable=False)
+    version = Column(Integer, default=0, index=True)
+    distribution = Column(Integer, nullable=False, default=0)
+    sharing_group_id = Column(Integer, index=True, default=0)
+    org_id = Column(Integer, nullable=False, index=True, default=0)
+    orgc_id = Column(Integer, nullable=False, index=True, default=0)
+    default = Column(Boolean, nullable=False, default=False, index=True)
+    locked = Column(Boolean, nullable=False, default=False)
+    extends_uuid = Column(String(40), index=True)
+    extends_version = Column(Integer, default=0, index=True)
+    published = Column(Boolean, nullable=False, default=False)
+    deleted = Column(Boolean, nullable=False, default=False)
+
+
+class GalaxyElement(Base):
+    __tablename__ = "galaxy_elements"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    galaxy_cluster_id = Column(Integer, ForeignKey(GalaxyCluster.id, ondelete="CASCADE"), nullable=False, index=True)
+    key = Column(String(255), nullable=False, default="", index=True)
+    value = Column(Text, nullable=False)
+
+
+class GalaxyReference(Base):
+    __tablename__ = "galaxy_reference"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    galaxy_cluster_id = Column(Integer, ForeignKey(GalaxyCluster.id, ondelete="CASCADE"), nullable=False, index=True)
+    referenced_galaxy_cluster_id = Column(Integer, nullable=False, index=True)
+    referenced_galaxy_cluster_uuid = Column(String(255), nullable=False, index=True)
+    referenced_galaxy_cluster_type = Column(Text, nullable=False)
+    referenced_galaxy_cluster_value = Column(Text, nullable=False)

mmisp/db/models/identity_provider.py

@@ -0,0 +1,22 @@
+from datetime import datetime
+
+from sqlalchemy import Boolean, Column, DateTime, Integer, String
+
+from ..database import Base
+
+
+class OIDCIdentityProvider(Base):
+    __tablename__ = "oidc_identity_providers"
+
+    id = Column(Integer, primary_key=True)
+    name = Column(String(255), nullable=False)
+    org_id = Column(Integer, nullable=False)
+    active = Column(Boolean, default=True)
+    base_url = Column(String(255), nullable=False)
+    client_id = Column(String(255), nullable=False)
+    client_secret = Column(String(255), nullable=False)
+    scope = Column(String(255))
+    """Possibility to add more scopes to be requested from the idp in addition to the default scopes,
+    currently not used."""
+    created = Column(DateTime, default=datetime.utcnow)
+    modified = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

mmisp/db/models/noticelist.py

@@ -0,0 +1,22 @@
+from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, Text
+
+from mmisp.db.database import Base
+
+
+class Noticelist(Base):
+    __tablename__ = "noticelists"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), nullable=False, index=True)
+    expanded_name = Column(String(255), nullable=False)
+    ref = Column(String(255))  # data serialized as json
+    geographical_area = Column(String(255))  # data serialized as json
+    version = Column(Integer, nullable=False, default=1)
+    enabled = Column(Boolean, nullable=False, default=False)
+
+
+class NoticelistEntry(Base):
+    __tablename__ = "noticelist_entries"
+    id = Column(Integer, primary_key=True, nullable=False)
+    noticelist_id = Column(Integer, ForeignKey(Noticelist.id, ondelete="CASCADE"), nullable=False)
+    data = Column(Text, nullable=False)  # data serialized as json

mmisp/db/models/object.py

@@ -0,0 +1,39 @@
+from sqlalchemy import Boolean, Column, ForeignKey, Integer, String
+
+from mmisp.db.database import Base
+from mmisp.util.uuid import uuid
+
+
+class Object(Base):
+    __tablename__ = "objects"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(255), unique=True, default=uuid, index=True)
+    name = Column(String(255), index=True)
+    meta_category = Column("meta-category", String(255), index=True, key="meta_category")
+    description = Column(String(255))
+    template_uuid = Column(String(255), index=True, default=None)
+    template_version = Column(Integer, index=True, nullable=False)
+    event_id = Column(Integer, ForeignKey("events.id"), index=True, nullable=False)
+    timestamp = Column(Integer, index=True, nullable=False, default=0)
+    distribution = Column(Integer, index=True, nullable=False, default=0)
+    sharing_group_id = Column(Integer, ForeignKey("sharing_groups.id"), index=True)
+    comment = Column(String(255), nullable=False)
+    deleted = Column(Boolean, nullable=False, default=False)
+    first_seen = Column(Integer, index=True, default=None)
+    last_seen = Column(Integer, index=True, default=None)
+
+
+class ObjectTemplate(Base):
+    __tablename__ = "object_templates"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(255), unique=True, default=uuid, index=True)
+    name = Column(String(255), index=True)
+    user_id = Column(Integer, ForeignKey("users.id"), index=True, nullable=False)
+    org_id = Column(Integer, ForeignKey("organisations.id"), index=True, nullable=False)
+    description = Column(String(255))
+    version = Column(Integer, nullable=False)
+    requirements = Column(String(255))
+    fixed = Column(Boolean, nullable=False, default=False)
+    active = Column(Boolean, nullable=False, default=False)

mmisp/db/models/organisation.py

@@ -0,0 +1,26 @@
+from datetime import datetime
+
+from sqlalchemy import Boolean, Column, DateTime, Integer, String, Text
+
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+
+
+class Organisation(Base):
+    __tablename__ = "organisations"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), nullable=False, unique=True)
+    date_created = Column(DateTime, default=datetime.utcnow, nullable=False)
+    date_modified = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)
+    description = Column(Text)
+    type = Column(String(255))
+    nationality = Column(String(255))
+    sector = Column(String(255))
+    created_by = Column(Integer, nullable=False, default=0)
+    uuid = Column(String(255), unique=True, default=uuid)
+    contacts = Column(Text)
+    local = Column(Boolean, nullable=False, default=False)
+    restricted_to_domain = Column(Text)
+    landingpage = Column(Text)

mmisp/db/models/role.py

@@ -0,0 +1,42 @@
+from sqlalchemy import Boolean, Column, DateTime, Integer, String
+
+from ..database import Base
+
+
+class Role(Base):
+    __tablename__ = "roles"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), nullable=False)
+    created = Column(DateTime, default=None)
+    modified = Column(DateTime, default=None)
+    perm_add = Column(Boolean, default=False)
+    perm_modify = Column(Boolean, default=False)
+    perm_modify_org = Column(Boolean, default=False)
+    perm_publish = Column(Boolean, default=False)
+    perm_delegate = Column(Boolean, nullable=False, default=False)
+    perm_sync = Column(Boolean, default=False)
+    perm_admin = Column(Boolean, default=False)
+    perm_audit = Column(Boolean, default=False)
+    perm_full = Column(Boolean, default=False)
+    perm_auth = Column(Boolean, nullable=False, default=False)
+    perm_site_admin = Column(Boolean, nullable=False, default=False)
+    perm_regexp_access = Column(Boolean, nullable=False, default=False)
+    perm_tagger = Column(Boolean, nullable=False, default=False)
+    perm_template = Column(Boolean, nullable=False, default=False)
+    perm_sharing_group = Column(Boolean, nullable=False, default=False)
+    perm_tag_editor = Column(Boolean, nullable=False, default=False)
+    perm_sighting = Column(Boolean, nullable=False, default=False)
+    perm_object_template = Column(Boolean, nullable=False, default=False)
+    default_role = Column(Boolean, nullable=False, default=False)
+    memory_limit = Column(String(255), default="")
+    max_execution_time = Column(String(255), default="")
+    restricted_to_site_admin = Column(Boolean, nullable=False, default=False)
+    perm_publish_zmq = Column(Boolean, nullable=False, default=False)
+    perm_publish_kafka = Column(Boolean, nullable=False, default=False)
+    perm_decaying = Column(Boolean, nullable=False, default=False)
+    enforce_rate_limit = Column(Boolean, nullable=False, default=False)
+    rate_limit_count = Column(Integer, nullable=False, default=0)
+    perm_galaxy_editor = Column(Boolean, default=False)
+    perm_warninglist = Column(Boolean, default=False)
+    perm_view_feed_correlations = Column(Boolean, default=False)

mmisp/db/models/server.py

@@ -0,0 +1,33 @@
+from sqlalchemy import VARCHAR, Boolean, Column, Integer, Text
+
+from ..database import Base
+
+
+class Server(Base):
+    __tablename__ = "servers"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(VARCHAR(255), nullable=False)
+    url = Column(VARCHAR(255), nullable=False)
+    authkey = Column(VARCHAR(40), nullable=False)
+    org_id = Column(Integer, nullable=False, index=True)
+    push = Column(Boolean, nullable=False)
+    pull = Column(Boolean, nullable=False)
+    push_sightings = Column(Boolean, nullable=False, default=False)
+    push_galaxy_clusters = Column(Boolean, default=False)
+    pull_galaxy_clusters = Column(Boolean, default=False)
+    lastpulledid = Column(Integer)
+    lastpushedid = Column(Integer)
+    organization = Column(VARCHAR(10), default=None)
+    remote_org_id = Column(Integer, nullable=False, index=True)
+    publish_without_email = Column(Boolean, nullable=False, default=False)
+    unpublish_event = Column(Boolean, nullable=False, default=False)
+    self_signed = Column(Boolean, nullable=False)
+    pull_rules = Column(Text, nullable=False)
+    push_rules = Column(Text, nullable=False)
+    cert_file = Column(VARCHAR(255))
+    client_cert_file = Column(VARCHAR(255))
+    internal = Column(Boolean, nullable=False, default=False)
+    skip_proxy = Column(Boolean, nullable=False, default=False)
+    caching_enabled = Column(Boolean, nullable=False, default=False)
+    priority = Column(Integer, nullable=False, default=0, index=True)

mmisp/db/models/sharing_group.py

@@ -0,0 +1,43 @@
+from datetime import datetime
+
+from sqlalchemy import Boolean, Column, DateTime, Integer, String, Text
+
+from mmisp.util.uuid import uuid
+
+from ..database import Base
+
+
+class SharingGroup(Base):
+    __tablename__ = "sharing_groups"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), nullable=False, unique=True)
+    releasability = Column(Text, nullable=False)
+    description = Column(Text, nullable=False, default="")
+    uuid = Column(String(40), unique=True, default=uuid, nullable=False)
+    organisation_uuid = Column(String(40), nullable=False)
+    org_id = Column(Integer, nullable=False, index=True)
+    sync_user_id = Column(Integer, nullable=False, default=0, index=True)
+    active = Column(Boolean, nullable=False, default=False)
+    created = Column(DateTime, default=datetime.utcnow, nullable=False)
+    modified = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)
+    local = Column(Boolean, nullable=False, default=True)
+    roaming = Column(Boolean, default=False, nullable=False)
+
+
+class SharingGroupOrg(Base):
+    __tablename__ = "sharing_group_orgs"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    sharing_group_id = Column(Integer, index=True, nullable=False)
+    org_id = Column(Integer, index=True, nullable=False)
+    extend = Column(Boolean, default=False, nullable=False)
+
+
+class SharingGroupServer(Base):
+    __tablename__ = "sharing_group_servers"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    sharing_group_id = Column(Integer, index=True, nullable=False)
+    server_id = Column(Integer, index=True, nullable=False)
+    all_orgs = Column(Boolean, index=True, nullable=False, default=False)

mmisp/db/models/sighting.py

@@ -0,0 +1,21 @@
+from sqlalchemy import BigInteger, Column, ForeignKey, Integer, String
+
+from mmisp.db.database import Base
+from mmisp.util.uuid import uuid
+
+from .attribute import Attribute
+from .event import Event
+from .organisation import Organisation
+
+
+class Sighting(Base):
+    __tablename__ = "sightings"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    uuid = Column(String(40), unique=True, default=uuid)
+    attribute_id = Column(Integer, ForeignKey(Attribute.id), index=True, nullable=False)
+    event_id = Column(Integer, ForeignKey(Event.id), index=True, nullable=False)
+    org_id = Column(Integer, ForeignKey(Organisation.id), index=True, nullable=False)
+    date_sighting = Column(BigInteger, nullable=False)
+    source = Column(String(255), index=True, default="")
+    type = Column(Integer, index=True, default=0)

mmisp/db/models/tag.py

@@ -0,0 +1,19 @@
+from sqlalchemy import Boolean, Column, Integer, String
+
+from mmisp.db.database import Base
+
+
+class Tag(Base):
+    __tablename__ = "tags"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), unique=True, nullable=False)
+    colour = Column(String(7), nullable=False)
+    exportable = Column(Boolean, nullable=False)
+    org_id = Column(Integer, nullable=False, default=0, index=True)
+    user_id = Column(Integer, nullable=False, default=0, index=True)
+    hide_tag = Column(Boolean, nullable=False, default=False)
+    numerical_value = Column(Integer, index=True)
+    is_galaxy = Column(Boolean, default=False)
+    is_custom_galaxy = Column(Boolean, default=False)
+    local_only = Column(Boolean, default=False)

mmisp/db/models/taxonomy.py

@@ -0,0 +1,41 @@
+from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, Text
+
+from ..database import Base
+
+
+class Taxonomy(Base):
+    __tablename__ = "taxonomies"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    namespace = Column(String(255), nullable=False)
+    description = Column(Text, nullable=False)
+    version = Column(Integer, nullable=False)
+    enabled = Column(Boolean, nullable=False, default=False)
+    exclusive = Column(Boolean, default=False)
+    required = Column(Boolean, nullable=False, default=False)
+    highlighted = Column(Boolean)
+
+
+class TaxonomyPredicate(Base):
+    __tablename__ = "taxonomy_predicates"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    taxonomy_id = Column(Integer, ForeignKey(Taxonomy.id), nullable=False, index=True)
+    value = Column(Text, nullable=False)
+    expanded = Column(Text)
+    colour = Column(String(7))
+    description = Column(Text)
+    exclusive = Column(Boolean, default=False)
+    numerical_value = Column(Integer, index=True)
+
+
+class TaxonomyEntry(Base):
+    __tablename__ = "taxonomy_entries"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    taxonomy_predicate_id = Column(Integer, ForeignKey(TaxonomyPredicate.id), nullable=False, index=True)
+    value = Column(Text, nullable=False)
+    expanded = Column(Text)
+    colour = Column(String(7))
+    description = Column(Text)
+    numerical_value = Column(Integer, index=True)

mmisp/db/models/user.py

@@ -0,0 +1,44 @@
+from time import time
+
+from sqlalchemy import BigInteger, Boolean, Column, DateTime, ForeignKey, Integer, String, Text
+
+from ..database import Base
+from .organisation import Organisation
+
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    password = Column(String(255), nullable=False)
+    org_id = Column(Integer, ForeignKey(Organisation.id), nullable=False, index=True)
+    server_id = Column(Integer, nullable=False, default=0, index=True)
+    email = Column(String(255), nullable=False, unique=True)
+    autoalert = Column(Boolean, default=False, nullable=False)
+    authkey = Column(String(40), nullable=True, default=None)
+    invited_by = Column(Integer, default=0, nullable=False)
+    gpgkey = Column(Text)
+    certif_public = Column(Text)
+    nids_sid = Column(Integer, default=0, nullable=False)
+    termsaccepted = Column(Boolean, default=False, nullable=False)
+    newsread = Column(Integer, default=0)
+    role_id = Column(Integer, nullable=False, default=0)
+    change_pw = Column(Integer, default=0, nullable=False)
+    contactalert = Column(Boolean, default=False, nullable=False)
+    disabled = Column(Boolean, default=False, nullable=False)
+    expiration = Column(DateTime, default=None)
+    current_login = Column(Integer, default=0)
+    last_login = Column(Integer, default=0)
+    force_logout = Column(Boolean, default=False, nullable=False)
+    date_created = Column(Integer, default=time)
+    date_modified = Column(Integer, default=time, onupdate=time)
+    sub = Column(String(255), unique=True)
+    external_auth_required = Column(Boolean, nullable=False, default=False)
+    external_auth_key = Column(Text)
+    last_api_access = Column(Integer, default=0)
+    notification_daily = Column(Boolean, nullable=False, default=False)
+    notification_weekly = Column(Boolean, nullable=False, default=False)
+    notification_monthly = Column(Boolean, nullable=False, default=False)
+    totp = Column(String(255))
+    hotp_counter = Column(Integer)
+    last_pw_change = Column(BigInteger)