Update main.py

main.py

@@ -1,45 +1,100 @@
-import os
-from dotenv import load_dotenv
-from fastapi import FastAPI, Header, HTTPException, Depends
-from telegram import Update, Bot
-from pydantic import BaseModel
-
-class TelegramUpdate(BaseModel):
-    update_id: int
-    message: dict
-
-app = FastAPI()
-
-# Load variables from .env file if present
-load_dotenv()
-
-# Read the variable from the environment (or .env file)
-bot_token = os.getenv('BOT_TOKEN')
-secret_token = os.getenv("SECRET_TOKEN")
-# webhook_url = os.getenv('CYCLIC_URL', 'http://localhost:8181') + "/webhook/"
-
-bot = Bot(token=bot_token)
-# bot.set_webhook(url=webhook_url)
-# webhook_info = bot.get_webhook_info()
-# print(webhook_info)
-
-def auth_telegram_token(x_telegram_bot_api_secret_token: str = Header(None)) -> str:
-    # return true # uncomment to disable authentication
-    if x_telegram_bot_api_secret_token != secret_token:
-        raise HTTPException(status_code=403, detail="Not authenticated")
-    return x_telegram_bot_api_secret_token
-
-@app.post("/webhook/")
-async def handle_webhook(update: TelegramUpdate, token: str = Depends(auth_telegram_token)):
-    chat_id = update.message["chat"]["id"]
-    text = update.message["text"]
-    # print("Received message:", update.message)
-
-    if text == "/start":
-        with open('hello.gif', 'rb') as photo:
-            await bot.send_photo(chat_id=chat_id, photo=photo)
-        await bot.send_message(chat_id=chat_id, text="Welcome to Cyclic Starter Python Telegram Bot!")
+
+import telebot
+from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
+
+#from commands.start_command import start
+import fastf1
+def returnBot():
+    token = "7029872178:AAE5_8OXUHBpgxPc6wkKiIWbe5CIB8Pj1Bk"
+    return telebot.TeleBot(token)
+
+bot = returnBot()
+print("Bot on.")
+
+def chiudi_messaggio(messaggio):
+    markup = InlineKeyboardMarkup()
+    markup.row_width = 1
+    markup.add(InlineKeyboardButton("Chiudi", callback_data="cb_yes"),
+                               InlineKeyboardButton("No", callback_data="cb_no"))
+    return markup
+@bot.message_handler(commands="start")
+def start(message):
+    bot.send_message(message.chat.id, "Questo bot ha lo scopo di mandare i tempi sul giro di tutte le sessioni di F1 di uno specifico pilota.\nI tempi sul giro diventeranno disponibili dopo circa 1 ora-1 ora e mezza dalla fine della sessione.\nDigita /help per avere maggiore informazioni sui comandi\nQuesto bot è stato creato da @andreanxx")
+
+@bot.message_handler(commands="help")
+def help(message):
+    lista_comandi = "LISTA DEI COMANDI:\n/sessione [anno] [circuito] [tipo_sessione] [pilota] - Questo comando ti darà i tempi sul giro di un pilota a tua scelta. Per utilizzarlo dovrai inserire l'anno della sessione, il nome del circuito, il tipo di sessione e il pilota.\nI tipi di sessione che puoi inserire: 'R' (gara), 'Q' (qualifica), 'SQ' (qualifica sprint), 'FP1' (prova lib.1), 'FP2' (prova lib.2), 'FP3' (prova lib.3)\nI nomi dei piloti dovranno essere inseriti con il loro cognome abbreviato di 3 caratteri (Es: LEC,VER,ALO,SAI,NOR). Se non te li ricordi, puoi ottenere una lista dei nomi dei piloti abbreviati con il comando /nomi_piloti.\nES UTILIZZO DEL COMANDO /sessione:\n/sessione 2024 Bahrain R LEC"
+    bot.send_message(message.chat.id, lista_comandi)
+@bot.message_handler(commands="nomi_piloti")
+def nomi_piloti(message):
+    lista_piloti = """
+ALB	Alexander Albon
+ALO	Fernando Alonso
+BEA	Oliver Bearman
+BOT	Valtteri Bottas
+DEV	Nyck de Vries
+GAS	Pierre Gasly
+HAM	Lewis Hamilton
+HUL	Nico Hülkenberg
+LAT	Nicholas Latifi
+LAW	Liam Lawson
+LEC	Charles Leclerc
+MAG	Kevin Magnussen
+NOR	Lando Norris
+OCO	Esteban Ocon
+PER	Sergio Pérez
+PIA	Oscar Piastri
+RIC	Daniel Ricciardo
+RUS	George Russell
+SAI	Carlos Sainz Jr.
+SAR	Logan Sargeant
+SCH [a]	Mick Schumacher
+STR	Lance Stroll
+TSU	Yuki Tsunoda
+VER	Max Verstappen
+ZHO	Zhou Guanyu
+VER Max Verstappen
+                        """
+    bot.reply_to(message,"LISTA PILOTI:\n"+lista_piloti)
+@bot.message_handler(commands=["sessione"])
+def get_sessione(message):
+    args = message.text.split()
+    anno = int(args[1])
+    track = args[2]
+    tipo = args[3]
+    pilota = args[4]
+    if len(pilota) != 3:
+        bot.reply_to(message, "Pilota non valido, ricorda di mandare il pilota nella formattazione a 3 caratteri, es: LEC,ALO,VER,SAI")
     else:
-        await bot.send_message(chat_id=chat_id, reply_to_message_id=update.message["message_id"], text="Yo!")
+        attendi_messaggio = bot.reply_to(message, "Carico la sessione...")
+        race = fastf1.get_session(anno,track,tipo)
+        race.load()
+        bot.delete_message(message.chat.id, attendi_messaggio.message_id)
+        laps = race.laps.pick_driver(pilota).reset_index()
+        lap_times = [str(index) + ". " +str(lap) for index,lap in enumerate(laps['LapTime'])]
+        testo = f"TEMPI DI {pilota} in {track} nella {tipo}:\n" + "\n".join(lap_times).replace("0 days 00:", "")
+        if(len(lap_times) > 0):
+            markup = InlineKeyboardMarkup(row_width=1)
+            chiudi = InlineKeyboardButton('Chiudi', callback_data='chiusura')
+            markup.add(chiudi)
+            bot.reply_to(message, testo, reply_markup = markup)
+        else:
+            bot.reply_to(message, "Errore: Sessione non disponibile!\nRicontrolla i parametri che mi hai fornito o attendi che i dati verranno caricati")
+@bot.callback_query_handler(func=lambda call:True)
+def chiusura(callback):
+    if callback.message:
+        if callback.data == 'chiusura':
+            bot.delete_message(callback.message.chat.id,callback.message.id)
+def register_handlers():
+    #bot.register_message_handler(start, commands=['start'])
+    pass
+
+
+#register_handlers()
+
+def main():
+    bot.polling(non_stop=True)
+
 
-    return {"ok": True}
+if __name__ == '__main__':
+    main()