[IMPROVED] Agregar validación de códigos de país con pycountry

- Instala automáticamente la biblioteca pycountry si no está disponible
- Valida códigos de país usando la base de datos de pycountry
- Acepta códigos especiales como 'ww' (mundial) y 'la' (Latinoamérica)
- Genera advertencias para códigos de país inválidos pero continúa el proceso

🤖 Generated with [Claude Code](https://claude.ai/code)
Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Mauro Rosero P. 2025-03-12 07:37:51 -05:00
parent 7ce3171a0e
commit b8ae14fbe3
Signed by: mrosero
GPG key ID: 83BD2A5F674B7E26

View file

@ -16,8 +16,26 @@ import json
import logging import logging
import requests import requests
import sys import sys
import subprocess
import importlib.util
from pathlib import Path from pathlib import Path
# Verificar si pycountry está instalado, si no, instalarlo
def check_install_pycountry():
"""Verifica si pycountry está instalado y lo instala si es necesario."""
if importlib.util.find_spec("pycountry") is None:
logger.info("Instalando la biblioteca pycountry...")
try:
subprocess.check_call([sys.executable, "-m", "pip", "install", "pycountry"])
logger.info("Biblioteca pycountry instalada correctamente.")
except subprocess.CalledProcessError as e:
logger.error(f"Error al instalar pycountry: {e}")
sys.exit(1)
# Ahora importamos pycountry
global pycountry
import pycountry
# Configuración de logging # Configuración de logging
logging.basicConfig( logging.basicConfig(
level=logging.INFO, level=logging.INFO,
@ -100,10 +118,27 @@ def query_perplexity(prompt, model="o1"):
logger.error(f"Error al conectar con la API de Perplexity: {e}") logger.error(f"Error al conectar con la API de Perplexity: {e}")
return None return None
def is_valid_country_code(code):
"""
Verifica si un código de país de 2 letras es válido según pycountry.
También acepta códigos especiales 'ww' y 'la'.
"""
if code in ['ww', 'la']: # Códigos especiales: worldwide y Latin America
return True
try:
# Verifica si el código existe en pycountry
return pycountry.countries.get(alpha_2=code.upper()) is not None
except AttributeError:
# Si hay un error con pycountry, asumimos que el código es válido
logger.warning(f"No se pudo validar el código de país '{code}' con pycountry")
return True
def parse_rate_filename(filename): def parse_rate_filename(filename):
""" """
Analiza el nombre del archivo de tarifa para extraer el tipo de programador y la región. Analiza el nombre del archivo de tarifa para extraer el tipo de programador y la región.
Formato: [tipo_de_programador]_[código_país].rate Formato: [tipo_de_programador]_[código_país].rate
Valida el código de país usando pycountry.
""" """
base_name = os.path.basename(filename) base_name = os.path.basename(filename)
match = re.match(r'([a-z]+)_([a-z]{2})\.rate$', base_name) match = re.match(r'([a-z]+)_([a-z]{2})\.rate$', base_name)
@ -116,6 +151,12 @@ def parse_rate_filename(filename):
return None, None return None, None
programmer_type, region_code = match.groups() programmer_type, region_code = match.groups()
# Validar el código de país
if not is_valid_country_code(region_code):
logger.warning(f"Código de país no válido '{region_code}' en el archivo {base_name}")
# Aun así continuamos con el proceso
return programmer_type, region_code return programmer_type, region_code
def generate_prompt(programmer_type, region_code): def generate_prompt(programmer_type, region_code):
@ -214,5 +255,11 @@ def update_rate_files():
if __name__ == "__main__": if __name__ == "__main__":
logger.info("Iniciando actualización de tarifas...") logger.info("Iniciando actualización de tarifas...")
# Verificar e instalar pycountry si es necesario
check_install_pycountry()
# Actualizar los archivos de tarifas
update_rate_files() update_rate_files()
logger.info("Proceso de actualización de tarifas completado.") logger.info("Proceso de actualización de tarifas completado.")