Initial commit

app.py

@@ -0,0 +1,71 @@
+from flask import Flask, jsonify, request
+from drivers.nokia import NokiaDriver
+from drivers.fiberhome import FiberhomeDriver
+from config import Config
+from utils.cache import cache_response
+import time
+
+import json
+import os
+
+app = Flask(__name__)
+
+# Carregar inventário de hosts
+HOSTS_CONFIG = {}
+if os.path.exists('hosts.json'):
+    with open('hosts.json', 'r') as f:
+        HOSTS_CONFIG = json.load(f)
+
+# Mapeamento de drivers
+DRIVERS = {
+    'nokia': NokiaDriver,
+    'fiberhome': FiberhomeDriver
+}
+
+@app.route('/api/v1/olt_stats', methods=['GET'])
+@cache_response # Cacheia a resposta baseado nos argumentos da request
+def get_olt_stats():
+    host_ip = request.args.get('host')
+    driver_name_param = request.args.get('driver')
+    
+    if not host_ip:
+        return jsonify({'error': 'Missing host parameter'}), 400
+    
+    # Determinar configurações (hosts.json ou defaults)
+    host_config = HOSTS_CONFIG.get(host_ip, {})
+    
+    # Parâmetros: Preference para query param > hosts.json > default config
+    driver_name = driver_name_param or host_config.get('driver')
+    username = host_config.get('username') or Config.OLT_USERNAME
+    password = host_config.get('password') or Config.OLT_PASSWORD
+    
+    # Opções extras para o driver (ex: ssh_options, snmp_community)
+    # Remove chaves padrão para não duplicar e deixa o resto como opção
+    driver_options = {k: v for k, v in host_config.items() if k not in ['username', 'password', 'driver']}
+    
+    if not driver_name:
+         return jsonify({'error': 'Driver not specified (param or hosts.json)'}), 400
+
+    if driver_name not in DRIVERS:
+        return jsonify({'error': f'Driver {driver_name} not supported'}), 400
+    
+    try:
+        # Instancia o driver com opções extras
+        driver_class = DRIVERS[driver_name]
+        driver = driver_class(host_ip, username, password, **driver_options)
+        
+        # Coleta estatísticas completas (Cards > PONs > ONTs)
+        stats = driver.get_olt_stats()
+        
+        # Estrutura flexível, o driver define o retorno
+        return jsonify(stats)
+        
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/health', methods=['GET'])
+def health_check():
+    return jsonify({'status': 'ok', 'service': 'IPv0 OLT API', 'version': '3.1'}), 200
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5050)