126 lines
3.7 KiB
Python
126 lines
3.7 KiB
Python
import sqlite3
|
||
import os
|
||
import datetime
|
||
import urllib.request
|
||
import urllib.parse
|
||
import json
|
||
|
||
# --- CONFIGURAZIONE ---
|
||
BOT_TOKEN = os.environ.get('BOT_TOKEN')
|
||
CHAT_ID = os.environ.get('ALLOWED_USER_ID')
|
||
DB_PATH = "/data/speedtest.sqlite"
|
||
|
||
# SOGLIE DI ALLARME (Mbps)
|
||
WARN_DOWN = 400
|
||
WARN_UP = 100
|
||
|
||
def send_telegram_message(message):
|
||
if not message: return
|
||
|
||
url = f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage"
|
||
payload = {
|
||
"chat_id": CHAT_ID,
|
||
"text": message,
|
||
"parse_mode": "Markdown"
|
||
}
|
||
|
||
try:
|
||
data = urllib.parse.urlencode(payload).encode('utf-8')
|
||
req = urllib.request.Request(url, data=data)
|
||
with urllib.request.urlopen(req) as response:
|
||
if response.status == 200:
|
||
print("✅ Report inviato.")
|
||
except Exception as e:
|
||
print(f"❌ Errore invio Telegram: {e}")
|
||
|
||
def generate_report():
|
||
if not os.path.exists(DB_PATH):
|
||
print(f"❌ Database non trovato: {DB_PATH}")
|
||
return None
|
||
|
||
try:
|
||
conn = sqlite3.connect(DB_PATH)
|
||
cursor = conn.cursor()
|
||
# Calcola l'orario UTC di 24h fa (perché il DB è in UTC)
|
||
yesterday = datetime.datetime.now(datetime.timezone.utc) - datetime.timedelta(hours=24)
|
||
|
||
query = "SELECT download, upload, ping, created_at FROM results WHERE created_at > ? AND status = 'completed' ORDER BY created_at ASC"
|
||
cursor.execute(query, (yesterday,))
|
||
rows = cursor.fetchall()
|
||
except Exception as e:
|
||
print(f"❌ Errore DB: {str(e)}")
|
||
return None
|
||
finally:
|
||
if 'conn' in locals(): conn.close()
|
||
|
||
if not rows:
|
||
print("ℹ️ Nessun test trovato.")
|
||
return None
|
||
|
||
# Variabili per statistiche
|
||
total_down = 0
|
||
total_up = 0
|
||
count = 0
|
||
issues = 0
|
||
|
||
# Intestazione Messaggio
|
||
# Usiamo l'ora locale del sistema per l'intestazione
|
||
now_local = datetime.datetime.now()
|
||
msg = f"📊 **REPORT VELOCITÀ 24H**\n📅 {now_local.strftime('%d/%m/%Y')}\n\n"
|
||
|
||
msg += "```text\n"
|
||
msg += "ORA | DOWN | UP \n"
|
||
msg += "------+------+-----\n"
|
||
|
||
for row in rows:
|
||
# Conversione Bit -> Mbps
|
||
d_val = (int(row[0]) * 8) / 1000000
|
||
u_val = (int(row[1]) * 8) / 1000000
|
||
|
||
total_down += d_val
|
||
total_up += u_val
|
||
count += 1
|
||
|
||
marker = ""
|
||
if d_val < WARN_DOWN or u_val < WARN_UP:
|
||
issues += 1
|
||
marker = "!"
|
||
|
||
# Parsing Data e CONVERSIONE TIMEZONE
|
||
try:
|
||
d_str = row[3].split(".")[0].replace("T", " ")
|
||
# 1. Creiamo l'oggetto datetime e gli diciamo "Tu sei UTC"
|
||
dt_utc = datetime.datetime.strptime(d_str, '%Y-%m-%d %H:%M:%S').replace(tzinfo=datetime.timezone.utc)
|
||
|
||
# 2. Convertiamo nell'orario locale del sistema (definito da TZ nel docker-compose)
|
||
dt_local = dt_utc.astimezone()
|
||
|
||
time_str = dt_local.strftime('%H:%M')
|
||
except:
|
||
time_str = "--:--"
|
||
|
||
# Formattazione tabella
|
||
row_str = f"{time_str} | {d_val:>4.0f} | {u_val:>3.0f} {marker}"
|
||
msg += f"{row_str}\n"
|
||
|
||
msg += "```\n"
|
||
|
||
if count > 0:
|
||
avg_d = total_down / count
|
||
avg_u = total_up / count
|
||
|
||
icon_d = "✅" if avg_d >= WARN_DOWN else "⚠️"
|
||
icon_u = "✅" if avg_u >= WARN_UP else "⚠️"
|
||
|
||
msg += f"📈 **MEDIA:**\n⬇️ {icon_d} `{avg_d:.0f}` Mbps\n⬆️ {icon_u} `{avg_u:.0f}` Mbps"
|
||
|
||
if issues > 0:
|
||
msg += f"\n\n⚠️ **{issues}** test sotto soglia (!)"
|
||
|
||
return msg
|
||
|
||
if __name__ == "__main__":
|
||
report = generate_report()
|
||
if report:
|
||
send_telegram_message(report)
|