1#@!#!123s

: / opt / imunify360 / venv / lib64 / python3.11 / site-packages / im360 / model /

Filename : update_hooks.py

back

import asyncio

from defence360agent.model import instance
from defence360agent.model.simplification import run_in_executor
from im360.model.geoip_data import GeoCSVReader

from .country import Country


async def update_geodb(_, is_updated):
    """
    Load countries list (if it is empty) from the csv file on update.
    """

    def update_country_db_from_csv():
        csv = GeoCSVReader()
        with instance.db.atomic():
            Country.update_from(csv.countries())

    loop = asyncio.get_event_loop()

    # Check that table country is not empty
    countries_count = await run_in_executor(
        loop, lambda: (Country.select().count())
    )

    if (countries_count == 0) or is_updated:
        await run_in_executor(loop, update_country_db_from_csv)