OwlCyberSecurity - MANAGER

Edit File: 102_proactive_ignore_list.py

from time import time

import peewee as pw

def migrate(migrator, database, fake=False, **kwargs):
    class ProactiveIgnoredPath(pw.Model):
        """
        Ignore list for proactive defence
        """

path = pw.TextField(null=False, primary_key=True)
        timestamp = pw.IntegerField(null=False, default=time)

class Meta:
            db_table = "proactive_ignored_path"

class ProactiveIgnoredRule(pw.Model):
        """
        Specific rules ignored
        """

path = pw.ForeignKeyField(
            ProactiveIgnoredPath,
            null=False,
            on_delete="CASCADE",
            related_name="rules",
        )
        rule_id = pw.IntegerField(null=False)
        rule_name = pw.TextField(null=False)

class Meta:
            db_table = "proactive_ignored_rule"
            indexes = ((("path", "rule_id"), True),)

migrator.create_model(ProactiveIgnoredPath)
    migrator.create_model(ProactiveIgnoredRule)

Proactive = migrator.orm["proactive"]
    migrator.add_fields(
        Proactive,
        rule_id=pw.IntegerField(null=True),
    )
    migrator.rename_field(Proactive, "reason", "rule_name")

def rollback(migrator, database, fake=False, **kwargs):
    ProactiveIgnoredPath = migrator.orm["proactive_ignored_path"]
    ProactiveIgnoredRule = migrator.orm["proactive_ignored_rule"]
    migrator.remove_model(ProactiveIgnoredRule)
    migrator.remove_model(ProactiveIgnoredPath)

Proactive = migrator.orm["proactive"]
    migrator.remove_fields(Proactive, "rule_id")
    migrator.rename_field(Proactive, "rule_name", "reason")