| Current Path : /proc/thread-self/root/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/ |
| Current File : //proc/thread-self/root/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/147_remove_vendor_field.py |
import peewee
def migrate(migrator, database, fake=False, **kwargs):
malware_hits = migrator.orm["malware_hits"]
migrator.remove_fields(malware_hits, "vendor")
def rollback(migrator, database, fake=False, **kwargs):
malware_hits = migrator.orm["malware_hits"]
migrator.add_fields(
malware_hits, vendor=peewee.CharField(null=False, default="ai-bolit")
)