From 16817ea954e247a5317abd113fdce2e06514fee0 Mon Sep 17 00:00:00 2001
From: Lyreplus <lyreplus@gmail.com>
Date: Wed, 13 Nov 2024 16:30:47 +0100
Subject: [PATCH] add ore command

---
 bot.py           | 102 +++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |   2 +-
 2 files changed, 103 insertions(+), 1 deletion(-)

diff --git a/bot.py b/bot.py
index 1043544..95c0325 100644
--- a/bot.py
+++ b/bot.py
@@ -3,11 +3,17 @@
 from telegram import Update
 from telegram.ext import Application, CommandHandler, ContextTypes
 from collections import defaultdict
+import requests
+import json
 
 DATABASE_URL = os.getenv("DATABASE_URL")
 BOT_TOKEN = os.getenv("BOT_TOKEN")
 
 
+# constant for the deep linking
+NOME_COGNOME = "nome.cognome"
+
+
 def create_table():
     conn = psycopg2.connect(DATABASE_URL)
     cursor = conn.cursor()
@@ -25,6 +31,46 @@ def create_table():
     conn.close()
 
 
+def create_table_ore():
+    conn = psycopg2.connect(DATABASE_URL)
+    cursor = conn.cursor()
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS ore (
+            id SERIAL PRIMARY KEY,
+            user_id BIGINT,
+            user_mail TEXT,
+            timestamp TIMESTAMPTZ DEFAULT NOW()
+        )
+    """)
+    conn.commit()
+    cursor.close()
+    conn.close()
+
+
+def add_user_to_db(user_id, user_mail):
+    conn = psycopg2.connect(DATABASE_URL)
+    cursor = conn.cursor()
+    cursor.execute("""
+        INSERT INTO ore (user_id, user_mail)
+        VALUES (%s, %s)
+    """, (user_id, user_mail))
+    conn.commit()
+    cursor.close()
+    conn.close()
+
+
+def get_mail_from_id_db(user_id):
+    conn = psycopg2.connect(DATABASE_URL)
+    cursor = conn.cursor()
+    cursor.execute("""
+        SELECT user_mail FROM ore WHERE user_id = %s
+    """, (user_id,))
+    rows = cursor.fetchall()
+    cursor.close()
+    conn.close()
+    return rows
+
+
 def add_topic_to_db(topic_id, author, topic_text):
     conn = psycopg2.connect(DATABASE_URL)
     cursor = conn.cursor()
@@ -101,13 +147,69 @@ async def get_chat_id_topic(update: Update, context: ContextTypes.DEFAULT_TYPE):
         await update.message.reply_text(f"Id della chat: {chat_id}, id del topic: {topic_id}")
 
 
+async def is_in_db(user_id):
+    conn = psycopg2.connect(DATABASE_URL)
+    cursor = conn.cursor()
+    cursor.execute("""
+        SELECT user_mail FROM ore WHERE user_id = %s
+    """, (user_id,))
+    mail = cursor.fetchall()
+    cursor.close()
+    conn.close()
+    if mail:
+        return True
+    else:
+        return False
+
+
+async def deep_linked_level_1(update: Update, context: ContextTypes.DEFAULT_TYPE):
+
+    user_mail = ""
+    user_id = update.message.chat_id
+    if context.args:
+        user_mail = context.args[0]
+    else:
+        await update.message.reply_text("Attenzione: non sei autenticat*. Contatta lo staff IT")
+        return
+    add_user_to_db(user_id, user_mail)
+    await update.message.reply_text("Sei autenticat*!")
+
+
+async def ore(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = update.message.from_user.id
+    if is_in_db(user_id) is not True:
+        url = "https://api.eagletrt.it/api/v2/tecsLinkOre"
+        text = f"Clicca su <a href='{url}'>questo link</a> per le ore"
+        await update.message.reply_text(text)
+        return
+    email = get_mail_from_id_db(user_id) 
+    url = f"https://api.eagletrt.it/api/v2/oreLab?username={email}"
+
+    response = requests.get(url)
+
+    if response.status_code == 200:
+        data = response.json()
+        ore = data.get("ore")
+        if ore is None:
+            await update.message.reply_text("Errore, contatta lo staff IT.\n Codice errore: in ore function: ore is None")                
+            return
+        reply_ore = f"Mi risulta che finora tu abbia trascorso {ore} ore nel laboratorio di E-Agle TRT questo mese"
+        await update.message.reply_text(reply_ore)
+    else:
+        status_code = response.status_code
+        await update.message.reply_text(f"Errore, contatta lo staff IT.\n Codice errore: in ore function, response status code is {status_code}")
+
+
 if __name__ == "__main__":
     def main():
         create_table()
+        create_table_ore()
 
         application = Application.builder().token(BOT_TOKEN).build()
+        application.add_handler(CommandHandler("start", deep_linked_level_1, filters.Regex(NOME_COGNOME)))
         application.add_handler(CommandHandler("odg", handle_odg))
         application.add_handler(CommandHandler("chatid", get_chat_id_topic))
+        application.add_handler(CommandHandler("ore", get_ore))
         application.run_polling()
 
     main()
diff --git a/requirements.txt b/requirements.txt
index 3920e19..2731496 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 python-telegram-bot
 psycopg2
-
+requests