Added constants. Added image creation. Refactored send found article

wallamanta/wallamanta.py

@@ -5,17 +5,12 @@ import logging
 import prettytable
 import helpers
 import walladb
+import constants
 
 from worker import Worker
 from telegram import Update
 from telegram.ext import Application, CommandHandler, ContextTypes
 
-TELEGRAM_CHANNEL_ID = os.getenv("TELEGRAM_CHANNEL_ID") 
-TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN")
-LATITUDE = os.getenv("LATITUDE")
-LONGITUDE = os.getenv("LONGITUDE")
-SLEEP_TIME = os.getenv("SLEEP_TIME")
-
 # Enable logging
 logging.basicConfig(
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
@@ -33,24 +28,33 @@ def save_json_file(products):
 
 async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     if walladb.is_user_valid(helpers.get_telegram_user_id(update)):
-        message = """Añade un producto con `/add producto;precio_mínimo;precio_máximo,excluir_título(opcional, separado por comas);excluir_descripción_y_título(opciona, separado por comas);latitud(opcional);longitud(opcional),distancia(opcional)`\n
-    Ejemplo: `/add placa base itx;0;150`\n
-    Ejemplo 2: `/add cpu;10;30;;intel,core 2 duo,celeron;;;100`\n
-    Los campos opcionales que se dejen vacíos tomarán el valor configurado en el archivo `.env`\n
-    Lista los productos con `/list` o obtén la información de uno en concreto con `/list nombre del producto`\n
-    Borra un producto con `/remove nombre del producto`"""
+        message = """Añade un producto con `/add producto;precio_mínimo;precio_máximo,categoría,excluir_título(opcional, separado por comas);
+excluir_descripción_y_título(opciona, separado por comas);latitud(opcional);longitud(opcional),distancia(opcional)`\n
+Ejemplo: `/add placa base itx;0;150`\n
+Ejemplo 2: `/add cpu;10;30;;intel,core 2 duo,celeron;;;100`\n
+Los campos opcionales que se dejen vacíos tomarán el valor configurado en el archivo `.env`\n
+Lista los productos con `/list` o obtén la información de uno en concreto con `/list nombre del producto`\n
+Borra un producto con `/remove nombre del producto`\n
+`/status` muestra tu tipo de membresía y fecha de caducidad"""
     else:
         message = """Activa tu periodo de prueba de 7 días con `/test` o contacta con @jocarduck para más información."""    
     await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
 
+async def categories_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    if walladb.is_user_valid(helpers.get_telegram_user_id(update)):
+        message = ''
+        for category in constants.CATEGORIES:
+            message = f"{message}, {category}"
+        await update.message.reply_markdown_v2(f"```{helpers.telegram_escape_characters(message)}```")
+
 async def add_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     telegram_user_id = helpers.get_telegram_user_id(update)
     if walladb.is_user_valid(telegram_user_id):
         number_of_products = walladb.count_user_products(telegram_user_id)
-        message = """Tienes que pasar el número correcto de parámetros: `/add producto;precio_mínimo;precio_máximo,excluir_título(opcional, separado por comas);excluir_descripción_y_título(opciona, separado por comas);latitud(opcional);longitud(opcional),distancia(opcional)`\n
-    Ejemplo: `/add placa base itx;0;150`\n
-    Ejemplo 2: `/add cpu;10;30;;intel,core 2 duo,celeron;;;100`\n
-    Los campos opcionales que se dejen vacíos tomarán el valor configurado en el archivo `.env`"""
+        message = """Tienes que pasar el número correcto de parámetros: `/add producto;precio_mínimo;precio_máximo,categoría,excluir_título(opcional, separado por comas);excluir_descripción_y_título(opciona, separado por comas);latitud(opcional);longitud(opcional),distancia(opcional)`\n
+Ejemplo: `/add placa base itx;0;150`\n
+Ejemplo 2: `/add cpu;10;30;;intel,core 2 duo,celeron;;;100`\n
+Los campos opcionales que se dejen vacíos tomarán el valor configurado en el archivo `.env`"""
         valid = False
         if walladb.is_user_testing(telegram_user_id):
             valid = True
@@ -72,15 +76,17 @@ async def add_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> Non
                 args = args[1].split(";")
                 product['product_name'], product['min_price'], product['max_price'] = args[0:3]
                 if len(args) > 3 and args[3]:
-                    product['title_exclude'] = args[3]
+                    product['category'] = helpers.get_category_id(args[3])
                 if len(args) > 4 and args[4]:
-                    product['title_description_exclude'] = args[4]
+                    product['title_exclude'] = args[4]
                 if len(args) > 5 and args[5]:
-                    product['latitude'] = args[5]
+                    product['title_description_exclude'] = args[5]
                 if len(args) > 6 and args[6]:
-                    product['longitude'] = args[6]
+                    product['latitude'] = args[6]
                 if len(args) > 7 and args[7]:
-                    product['distance'] = args[7]
+                    product['longitude'] = args[7]
+                if len(args) > 8 and args[8]:
+                    product['distance'] = args[8]
                 
                 logging.info(f'Adding: {product}')
                 if not walladb.get_product(product):
@@ -156,12 +162,14 @@ async def remove_user_command(update: Update, context: ContextTypes.DEFAULT_TYPE
         await update.message.reply_markdown_v2(helpers.telegram_escape_characters(f"{telegram_user_id} desactivado."))
 
 async def status_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
-    # TODO: Obtener estado del usuario y fecha de caducidad de suscripción en el caso de haberla
-    message = ''    
-    await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
+    telegram_user_id = helpers.get_telegram_user_id(update)
+    if walladb.is_user_valid(telegram_user_id):
+        type = walladb.get_user_type(telegram_user_id)
+        until = walladb.get_user_until(telegram_user_id)
+        message = f"Tu cuenta es tipo: {type} y caduca el {until}."
+        await update.message.reply_markdown_v2(helpers.telegram_escape_characters(message))
 
 async def test_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
-    # TODO: Activar usuario en modo testing y responder con fecha de caducidad
     telegram_user_id = helpers.get_telegram_user_id(update)
     if not walladb.is_user_valid(telegram_user_id):
         until = helpers.get_date_ahead(7)
@@ -186,10 +194,11 @@ def main()->None:
 
     """Start the bot."""
     # Create the Application and pass it your bot's token.
-    application = Application.builder().token(TELEGRAM_TOKEN).build()
+    application = Application.builder().token(constants.TELEGRAM_TOKEN).build()
 
     # on different commands - answer in Telegram
     application.add_handler(CommandHandler("help", help_command))
+    application.add_handler(CommandHandler("categories", categories_command))
     application.add_handler(CommandHandler("add", add_command))
     application.add_handler(CommandHandler("remove", remove_command))
     application.add_handler(CommandHandler("list", list_command))