[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:
parent
7ce3171a0e
commit
b8ae14fbe3
1 changed files with 47 additions and 0 deletions
|
@ -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.")
|
Loading…
Reference in a new issue