Current File : //opt/imunify360/venv/lib/python3.11/site-packages/im360/model/update_hooks.py
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)
Mini Shell