import logging logger = logging.getLogger(__name__) def migrate(migrator, database, fake=False, **kwargs): MalwareHit = migrator.orm["malware_hits"] try: migrator.remove_fields(MalwareHit, "uid", "gid") except Exception as e: logger.exception(e) def rollback(migrator, database, fake=False, **kwargs): pass