Current Directory: /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations
Viewing File: /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/082_add_manual_flag.py
import peewee as pw
def migrate(migrator, database, fake=False, **kwargs):
IPList = migrator.orm["iplist"]
migrator.add_fields(
IPList, manual=pw.BooleanField(null=False, default=True)
)
def rollback(migrator, database, fake=False, **kwargs):
IPList = migrator.orm["iplist"]
migrator.remove_fields(IPList, "manual")