mmisp-lib 0.1.0__py3-none-any.whl

mmisp/db/models/user_setting.py

@@ -0,0 +1,29 @@
+from enum import Enum
+from time import time
+
+from sqlalchemy import Column, ForeignKey, Integer, String, Text
+
+from mmisp.db.database import Base
+
+from .user import User
+
+
+class SettingName(Enum):
+    PUBLISH_ALERT_FILTER = "publish_alert_filter"
+    DASHBOARD_ACCESS = "dashboard_access"
+    DASHBOARD = "dashboard"
+    HOMEPAGE = "homepage"
+    DEFAULT_RESTSEARCH_PARAMETERS = "default_restsearch_parameters"
+    TAG_NUMERICAL_VALUE_OVERRIDE = "tag_numerical_value_override"
+    EVENT_INDEX_HIDE_COLUMNS = "event_index_hide_columns"
+    PERIODIC_NOTIFICATION_FILTERS = "periodic_notification_filters"
+
+
+class UserSetting(Base):
+    __tablename__ = "user_settings"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    setting = Column(String(255), nullable=False, index=True)
+    value = Column(Text, nullable=False)
+    user_id = Column(Integer, ForeignKey(User.id), nullable=False, index=True)
+    timestamp = Column(Integer, default=time, onupdate=time, nullable=False, index=True)

mmisp/db/models/warninglist.py

@@ -0,0 +1,33 @@
+from sqlalchemy import Boolean, Column, ForeignKey, Integer, String
+
+from mmisp.db.database import Base
+
+
+class Warninglist(Base):
+    __tablename__ = "warninglists"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String(255), nullable=False)
+    type = Column(String(255), nullable=False, default="string")
+    description = Column(String(255), nullable=False)
+    version = Column(Integer, nullable=False, default=1)
+    enabled = Column(Boolean, default=False, nullable=False)
+    default = Column(Boolean, default=True)
+    category = Column(String(255))
+
+
+class WarninglistEntry(Base):
+    __tablename__ = "warninglist_entries"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    value = Column(String(255), nullable=False)
+    warninglist_id = Column(Integer, ForeignKey(Warninglist.id, ondelete="CASCADE"), nullable=False)
+    comment = Column(String(255))
+
+
+class WarninglistType(Base):
+    __tablename__ = "warninglist_types"
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    type = Column(String(255), nullable=False)
+    warninglist_id = Column(Integer, ForeignKey(Warninglist.id, ondelete="CASCADE"), nullable=False)

mmisp/db/print_changes.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python3
+
+"""
+USAGE:
+    - Create env file with the variables
+        MYSQL_USER
+        MYSQL_PASSWORD
+        MYSQL_HOST
+        MYSQL_DBNAME
+    - Populate your environment `export $(cat ENVFILE)`
+    - Run `python -m mmisp.db.print_changes`
+"""
+
+
+import os
+import pprint
+
+from alembic.autogenerate import compare_metadata
+from alembic.migration import MigrationContext
+from sqlalchemy import MetaData, create_engine
+
+from .database import Base
+
+# import all models, so Base is populated
+from .models import (  # noqa: F401
+    attribute,
+    auth_key,
+    event,
+    feed,
+    galaxy,
+    galaxy_cluster,
+    noticelist,
+    object,
+    organisation,
+    role,
+    server,
+    sharing_group,
+    sighting,
+    tag,
+    taxonomy,
+    user,
+    user_setting,
+    warninglist,
+)
+
+myuser = os.getenv("MYSQL_USER")
+password = os.getenv("MYSQL_PASSWORD")
+host = os.getenv("MYSQL_HOST")
+db_name = os.getenv("MYSQL_DBNAME")
+
+engine = create_engine(f"mysql+mysqlconnector://{myuser}:{password}@{host}/{db_name}")
+
+
+metadata = MetaData()
+
+mc = MigrationContext.configure(engine.connect())
+
+diff = compare_metadata(mc, Base.metadata)
+pprint.pprint(diff, indent=2, width=20)