| Server IP : 188.40.95.74 / Your IP : 216.73.216.142 Web Server : Apache System : Linux cp01.striminghost.net 3.10.0-1160.119.1.el7.tuxcare.els13.x86_64 #1 SMP Fri Nov 22 06:29:45 UTC 2024 x86_64 User : vlasotin ( 1054) PHP Version : 5.6.40 Disable Function : NONE MySQL : ON | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/ |
Upload File : |
import peewee as pw
class BlockedPort(pw.Model):
"""
Port + protocol for blocking data
"""
port = pw.IntegerField(null=False)
proto = pw.CharField(
null=False, constraints=[pw.Check("proto in ('tcp', 'udp', 'all')")]
)
comment = pw.CharField(null=True)
class Meta:
db_table = "blocked_port"
indexes = (
# create an unique on port/proto
(("port", "proto"), True),
)
class IgnoredByPort(pw.Model):
"""
Ignored IPs for port + protocol
"""
port_proto = pw.ForeignKeyField(
BlockedPort, null=False, on_delete="CASCADE", related_name="ips"
)
ip = pw.CharField(null=False)
comment = pw.CharField(null=True)
class Meta:
db_table = "ignored_by_port_proto"
indexes = (
# create an unique on port/ip
(("port_proto", "ip"), True),
)
def migrate(migrator, database, fake=False, **kwargs):
migrator.create_model(BlockedPort)
migrator.create_model(IgnoredByPort)
IPList = migrator.orm["iplist"]
migrator.add_fields(IPList, full_access=pw.BooleanField(null=True))
def rollback(migrator, database, fake=False, **kwargs):
BlockedPort = migrator.orm["blocked_port"]
IgnoredByPort = migrator.orm["blocked_port_ip"]
IPList = migrator.orm["iplist"]
migrator.remove_model(BlockedPort)
migrator.remove_model(IgnoredByPort)
migrator.remove_fields(IPList, "full_access")