From 7ce3171a0e66a7eda7c8f5052c1060012354b682 Mon Sep 17 00:00:00 2001 From: "Mauro Rosero P." Date: Wed, 12 Mar 2025 07:16:25 -0500 Subject: [PATCH] =?UTF-8?q?[ADDED]=20Script=20para=20actualizaci=C3=B3n=20?= =?UTF-8?q?automatizada=20de=20tarifas=20via=20Perplexity?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Se a帽ade script rate_update.py para: - Analizar archivos .rate en la carpeta de configuraci贸n - Consultar a la API de Perplexity para obtener tarifas actualizadas - Guardar valores num茅ricos con 2 decimales en los archivos correspondientes - Preservar archivos especiales como kdevs.rate sin modificarlos - Soportar diferentes tipos de programadores y regiones 馃 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude --- bin/rate_update.py | 218 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 218 insertions(+) create mode 100755 bin/rate_update.py diff --git a/bin/rate_update.py b/bin/rate_update.py new file mode 100755 index 0000000..8c2bbb4 --- /dev/null +++ b/bin/rate_update.py @@ -0,0 +1,218 @@ +#!/usr/bin/env python3 +""" +Rate Update Script + +Este script actualiza las tarifas por hora de diferentes tipos de programadores +por regi贸n utilizando la API de Perplexity para obtener datos actualizados. + +Licencia: AGPL-3.0 +Modified: 2025-03-12 +""" + +import os +import glob +import re +import json +import logging +import requests +import sys +from pathlib import Path + +# Configuraci贸n de logging +logging.basicConfig( + level=logging.INFO, + format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', + handlers=[logging.StreamHandler()] +) +logger = logging.getLogger('rate_update') + +# Directorio base del proyecto +BASE_DIR = Path(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) +CONFIG_DIR = BASE_DIR / 'bin' / 'config' + +# Archivo de configuraci贸n para el modelo de IA +MODEL_CONFIG_FILE = CONFIG_DIR / 'rate_model.ai' + +def get_ai_model(): + """Obtener el modelo de IA configurado.""" + try: + with open(MODEL_CONFIG_FILE, 'r') as f: + return f.read().strip() + except FileNotFoundError: + logger.warning(f"Archivo de configuraci贸n {MODEL_CONFIG_FILE} no encontrado. Usando modelo predeterminado 'o1'.") + return "o1" + +def get_perplexity_api_key(): + """Obtener la clave API de Perplexity desde una variable de entorno.""" + api_key = os.environ.get('PERPLEXITY_API_KEY') + if not api_key: + logger.error("No se encontr贸 la clave API de Perplexity. Establezca la variable de entorno PERPLEXITY_API_KEY.") + sys.exit(1) + return api_key + +def query_perplexity(prompt, model="o1"): + """Realizar una consulta a la API de Perplexity.""" + api_key = get_perplexity_api_key() + + url = "https://api.perplexity.ai/chat/completions" + headers = { + "Authorization": f"Bearer {api_key}", + "Content-Type": "application/json" + } + + data = { + "model": model, + "messages": [ + { + "role": "system", + "content": "You are a helpful assistant that provides accurate salary information for developers worldwide. Respond only with numeric values in USD with 2 decimal places." + }, + { + "role": "user", + "content": prompt + } + ], + "max_tokens": 100 + } + + try: + response = requests.post(url, headers=headers, json=data) + response.raise_for_status() + result = response.json() + + # Extraer solo el valor num茅rico de la respuesta + content = result.get('choices', [{}])[0].get('message', {}).get('content', '') + # Intentar encontrar un n煤mero con 2 decimales en la respuesta + match = re.search(r'\$?(\d+\.\d{2})', content) + if match: + return float(match.group(1)) + + # Si no encuentra un formato exacto, intentar convertir toda la respuesta a float + try: + # Eliminar cualquier s铆mbolo de moneda y espacios + cleaned_content = re.sub(r'[^\d.]', '', content) + return round(float(cleaned_content), 2) + except ValueError: + logger.error(f"No se pudo extraer un valor num茅rico de la respuesta: {content}") + return None + + except requests.exceptions.RequestException as e: + logger.error(f"Error al conectar con la API de Perplexity: {e}") + return None + +def parse_rate_filename(filename): + """ + 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 + """ + base_name = os.path.basename(filename) + match = re.match(r'([a-z]+)_([a-z]{2})\.rate$', base_name) + + if not match: + # Para archivos como kdevs.rate que no siguen el patr贸n est谩ndar + if base_name.endswith('.rate'): + programmer_type = base_name.replace('.rate', '') + return programmer_type, None + return None, None + + programmer_type, region_code = match.groups() + return programmer_type, region_code + +def generate_prompt(programmer_type, region_code): + """Genera un prompt para consultar a Perplexity sobre la tarifa horaria.""" + region_map = { + 'ww': 'mundial', + 'la': 'Latinoam茅rica', + 'pa': 'Panam谩', + 'co': 'Colombia', + 'mx': 'M茅xico', + 'ar': 'Argentina', + 'br': 'Brasil', + 'cl': 'Chile', + 'pe': 'Per煤', + 'ec': 'Ecuador', + 'us': 'Estados Unidos', + 'ca': 'Canad谩', + 'uk': 'Reino Unido', + 'de': 'Alemania', + 'fr': 'Francia', + 'es': 'Espa帽a', + 'it': 'Italia' + } + + # Archivos especiales como kdevs.rate se manejan de forma diferente + if programmer_type == 'kdevs': + # Simplemente mantener el valor actual para kdevs.rate + return None + + programmer_map = { + 'bash': 'programador de scripts Bash/Shell', + 'python': 'desarrollador Python', + 'fullstack': 'desarrollador Full Stack', + 'frontend': 'desarrollador Frontend', + 'backend': 'desarrollador Backend', + 'devops': 'ingeniero DevOps', + 'mobile': 'desarrollador de aplicaciones m贸viles', + 'java': 'desarrollador Java', + 'php': 'desarrollador PHP', + 'ruby': 'desarrollador Ruby', + 'dotnet': 'desarrollador .NET', + 'data': 'cient铆fico de datos', + 'ml': 'ingeniero de Machine Learning', + 'cloud': 'arquitecto Cloud', + 'odoo': 'desarrollador Odoo' + } + + region_name = region_map.get(region_code, f'regi贸n con c贸digo {region_code}') + programmer_description = programmer_map.get(programmer_type, f'desarrollador {programmer_type}') + + prompt = f"驴Cu谩l es la tarifa por hora promedio en d贸lares estadounidenses (USD) para un {programmer_description} en {region_name} en 2025? Responde solo con el valor num茅rico con dos decimales." + + return prompt + +def update_rate_files(): + """Actualiza los archivos de tarifas con datos de Perplexity.""" + # Obtener modelo configurado + model = get_ai_model() + logger.info(f"Usando modelo de IA: {model}") + + # Buscar todos los archivos .rate + rate_files = glob.glob(str(CONFIG_DIR / '*.rate')) + logger.info(f"Encontrados {len(rate_files)} archivos de tarifas para actualizar.") + + for rate_file in rate_files: + programmer_type, region_code = parse_rate_filename(rate_file) + + if programmer_type is None: + logger.warning(f"No se pudo analizar el nombre del archivo: {rate_file}, saltando.") + continue + + # Si es un archivo especial como kdevs.rate + if region_code is None: + prompt = generate_prompt(programmer_type, None) + # Saltamos los archivos especiales que no necesitan actualizaci贸n + if prompt is None: + logger.info(f"Saltando archivo especial: {os.path.basename(rate_file)}") + continue + else: + prompt = generate_prompt(programmer_type, region_code) + + logger.info(f"Consultando tarifa para {programmer_type}" + + (f" en regi贸n {region_code}" if region_code else "")) + + # Consultar a Perplexity + rate = query_perplexity(prompt, model) + + if rate is not None: + # Guardar el resultado en el archivo + with open(rate_file, 'w') as f: + f.write(f"{rate:.2f}\n") + logger.info(f"Actualizado {os.path.basename(rate_file)} con valor: {rate:.2f}") + else: + logger.error(f"No se pudo obtener la tarifa para {programmer_type}" + + (f" en regi贸n {region_code}" if region_code else "")) + +if __name__ == "__main__": + logger.info("Iniciando actualizaci贸n de tarifas...") + update_rate_files() + logger.info("Proceso de actualizaci贸n de tarifas completado.") \ No newline at end of file