import os from dotenv import load_dotenv from telegram import Bot, InlineKeyboardButton, InlineKeyboardMarkup from telegram.constants import ReactionEmoji from src.jobspy.jobs import JobPost from src.jobspy.scrapers.utils import create_logger load_dotenv() logger = create_logger("TelegramBot") class TelegramBot: def __init__(self): self._api_token = os.getenv("TELEGRAM_API_TOKEN") self.chatId = os.getenv("TELEGRAM_CHAT_ID") self.bot = Bot(token=self._api_token) async def send_job(self, job: JobPost): """ Send JobPost details to Telegram chat. """ message = f"Job ID: {job.id}\n" \ f"Job Title: {job.title}\n" \ f"Company: {job.company_name}\n" \ f"Location: {job.location.display_location()}\n" \ f"Link: {job.job_url}\n" keyboard = [ [ InlineKeyboardButton(ReactionEmoji.FIRE, callback_data=ReactionEmoji.FIRE.name), InlineKeyboardButton(ReactionEmoji.PILE_OF_POO, callback_data=ReactionEmoji.PILE_OF_POO.name) ], ] reply_markup = InlineKeyboardMarkup(keyboard) try: await self.bot.sendMessage(chat_id=self.chatId, text=message, reply_markup=reply_markup) logger.info(f"Sent job to Telegram: {job.id}") except Exception as e: logger.error(f"Failed to send job to Telegram: {job.id}") logger.error(f"Error: {e}") async def send_test_message(self): """ Send Test Message to Telegram chat. """ message = "Test Test Testing" try: keyboard = [ [ InlineKeyboardButton(ReactionEmoji.FIRE, callback_data=ReactionEmoji.FIRE.name), InlineKeyboardButton(ReactionEmoji.PILE_OF_POO, callback_data=ReactionEmoji.PILE_OF_POO.name) ], ] reply_markup = InlineKeyboardMarkup(keyboard) await self.bot.sendMessage(chat_id=self.chatId, text=message, reply_markup=reply_markup) logger.info("Sent test message to Telegram") except Exception as e: logger.error("Failed o send test message to Telegram") logger.error(f"Error: {e}") async def set_message_reaction(self, message_id: int, emoji_reaction: str): """ Send Test Message to Telegram chat. """ try: await self.bot.set_message_reaction(chat_id=self.chatId, message_id=message_id, reaction=ReactionEmoji[emoji_reaction]) logger.info("Sent test message to Telegram") except Exception as e: logger.error("Failed to send test message to Telegram") logger.error(f"Error: {e}")