Update bot.py

Author: Programmeris

bot.py

@@ -1,3 +1,5 @@
+"""Simple Telegram Bot which parse .xlsx file by user message pattern"""
+
 import os
 from telegram import Update
 from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes
@@ -7,38 +9,59 @@
 XLSX_FILE_PATH = os.getenv("XLSX_FILE_PATH")
 
 async def find(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """Function which parse excel file by user message and send response to Telegram Bot"""
     if update.message and update.message.text:
         workbook = load_workbook(XLSX_FILE_PATH)
         sheet = workbook.active
         headers = get_headers(sheet)
         keywords_column = headers.get('Keywords')
         links_column = headers.get('Link')
+        tencode_column = headers.get('Ten-code')
+        name_column = headers.get('Name')
+        parsed_message = update.message.text.replace("/find", "").strip()
         response_message = ''
 
-        if keywords_column and links_column:
-            for row in sheet.iter_rows():
-                parsed_message = update.message.text.replace("/find", "").strip()
-                keywords = str(row[keywords_column - 1].value).replace(" ", "").split(";")
-                keyword_match = any(element in parsed_message for element in keywords)
-                if keyword_match:
-                    response_message += row[links_column - 1].value
-                
+        if any(char.isdigit() for char in parsed_message):
+            if tencode_column:
+               for row in sheet.iter_rows():
+                    tencode = str(row[tencode_column - 1].value)
+                    if parsed_message == tencode:
+                        response_message += f"{row[name_column - 1].value} : {row[links_column - 1].value}"
+        else:
+            if keywords_column and links_column:
+                for row in sheet.iter_rows():
+                    keywords = str(row[keywords_column - 1].value).split(";")
+                    trimmed_keywords = [s.lstrip() for s in keywords]
+                    lower_keywords = [s.lower() for s in trimmed_keywords]
+                    if parsed_message.lower() in lower_keywords:
+                        response_message += f"{row[name_column - 1].value} : {row[links_column - 1].value}"
+
+        if response_message == '':
+            response_message += "Not Found"
+
         await update.message.reply_text(f"{response_message}")
 
 def get_headers(sheet: any ) -> dict:
+    """Function which return excel column ids and cells as dictionary"""
     headers = {}
     for col_idx, cell in enumerate(sheet[1], start=1):
         headers[cell.value] = col_idx
-    
+
     return headers
 
 def main():
+    """Main function"""
     try:
         app = ApplicationBuilder().token(TELEGRAM_BOT_TOKEN).build()
         app.add_handler(CommandHandler("find", find))
+        app.add_handler(CommandHandler("f", find))
         app.run_polling()
     except Exception as err:
         print(f"Unexpected {err=}, {type(err)=}")
 
 if __name__ == "__main__":
     main()
+
+
+if __name__ == "__main__":
+    main()