%PDF- %PDF-
Mini Shell

Mini Shell

Direktori : /opt/imunify360/venv/lib/python3.8/site-packages/defence360agent/migrations/
Upload File :
Create Path :
Current File : //opt/imunify360/venv/lib/python3.8/site-packages/defence360agent/migrations/075_ips_as_int.py

"""
This program is free software: you can redistribute it and/or modify it under 
the terms of the GNU General Public License as published by 
the Free Software Foundation, either version 3 of the License, 
or (at your option) any later version.


This program is distributed in the hope that it will be useful, 
but WITHOUT ANY WARRANTY; without even the implied warranty of 
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
See the GNU General Public License for more details.


You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.

Copyright © 2019 Cloud Linux Software Inc.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import logging

import peewee as pw
import ipaddress


logger = logging.getLogger(__name__)


def migrate(migrator, database, fake=False, **kwargs):
    """Write your migrations here."""
    IPListNew = migrator.orm['iplist_new']
    IPList = migrator.orm['iplist']

    def iplist_select():
        # FIXME: remove this function after migrating to peewee 3
        try:
            yield from IPList.select(IPList).dicts().iterator()
        except RuntimeError:
            return

    try:
        from im360.utils.net import pack_ip_network
    except ImportError:
        pass
    else:
        with database.atomic():
            for ip_obj in iplist_select():
                try:
                    ip = ipaddress.ip_network(ip_obj['ip'])
                except ValueError:
                    # malformed ip
                    continue

                net, mask, version = pack_ip_network(ip)

                ip_obj.update({
                    'network_address': net,
                    'netmask': mask,
                    'version': version
                })
                try:
                    IPListNew.insert(ip_obj).execute()
                except pw.IntegrityError as e:
                    logger.warning("Error inserting IP: %s", e)

    migrator.sql('DROP TABLE iplist')
    migrator.sql("ALTER TABLE iplist_new RENAME TO iplist")
    migrator.sql('CREATE INDEX "iplist_listname" ON "iplist" ("listname")')
    migrator.sql('CREATE INDEX "iplist_expiration" ON "iplist" ("expiration")')
    migrator.sql('CREATE INDEX "iplist_ip" ON "iplist" ("ip")')


def rollback(migrator, database, fake=False, **kwargs):
    """Write your rollback migrations here."""
    pass

Zerion Mini Shell 1.0