major refactor (untested)
This commit is contained in:
parent
a29412b4da
commit
a77a0c0393
22 changed files with 1037 additions and 24 deletions
175
telegram_bot.py
Normal file
175
telegram_bot.py
Normal file
|
|
@ -0,0 +1,175 @@
|
|||
import os
|
||||
import logging
|
||||
import threading
|
||||
import time
|
||||
import requests
|
||||
|
||||
# Configuration from environment
|
||||
TELEGRAM_BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN", "")
|
||||
TELEGRAM_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID", "")
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class TelegramBot:
|
||||
"""Handle Telegram commands for controlling the monitor"""
|
||||
|
||||
def __init__(self, monitor, bot_token=None, chat_id=None):
|
||||
self.monitor = monitor
|
||||
self.bot_token = bot_token or TELEGRAM_BOT_TOKEN
|
||||
self.chat_id = chat_id or TELEGRAM_CHAT_ID
|
||||
self.last_update_id = 0
|
||||
self.running = False
|
||||
|
||||
def start(self):
|
||||
if not self.bot_token:
|
||||
logger.warning("Telegram bot token not configured, commands disabled")
|
||||
return
|
||||
self.running = True
|
||||
thread = threading.Thread(target=self._poll_updates, daemon=True)
|
||||
thread.start()
|
||||
logger.info("Telegram command listener started")
|
||||
|
||||
def stop(self):
|
||||
self.running = False
|
||||
|
||||
def _poll_updates(self):
|
||||
while self.running:
|
||||
try:
|
||||
url = f"https://api.telegram.org/bot{self.bot_token}/getUpdates"
|
||||
params = {"offset": self.last_update_id + 1, "timeout": 30}
|
||||
response = requests.get(url, params=params, timeout=35)
|
||||
if response.ok:
|
||||
data = response.json()
|
||||
if data.get("ok") and data.get("result"):
|
||||
for update in data["result"]:
|
||||
self.last_update_id = update["update_id"]
|
||||
self._handle_update(update)
|
||||
except requests.exceptions.Timeout:
|
||||
continue
|
||||
except Exception as e:
|
||||
logger.error(f"Telegram polling error: {e}")
|
||||
time.sleep(5)
|
||||
|
||||
def _handle_update(self, update):
|
||||
message = update.get("message", {})
|
||||
text = message.get("text", "")
|
||||
chat_id = str(message.get("chat", {}).get("id", ""))
|
||||
if chat_id != self.chat_id:
|
||||
logger.debug(f"Ignoring message from unknown chat: {chat_id}")
|
||||
return
|
||||
logger.info(f"Received Telegram command: {text}")
|
||||
if text.startswith("/autopilot"):
|
||||
self._handle_autopilot_command(text)
|
||||
elif text == "/status":
|
||||
self._handle_status_command()
|
||||
elif text == "/help":
|
||||
self._handle_help_command()
|
||||
elif text == "/plot":
|
||||
self._handle_plot_command()
|
||||
elif text == "/errorrate":
|
||||
self._handle_error_rate_command()
|
||||
elif text.startswith("/"):
|
||||
self._handle_unknown_command(text)
|
||||
|
||||
def _handle_autopilot_command(self, text):
|
||||
logger.info(f"Processing autopilot command: {text}")
|
||||
parts = text.split()
|
||||
if len(parts) < 2:
|
||||
self._send_message("Usage: /autopilot on|off")
|
||||
return
|
||||
action = parts[1].lower()
|
||||
if action == "on":
|
||||
logger.info("Enabling autopilot mode")
|
||||
self.monitor.set_autopilot(True)
|
||||
self._send_message("🤖 <b>Autopilot ENABLED</b>\n\nI will automatically apply to new listings!")
|
||||
elif action == "off":
|
||||
self.monitor.set_autopilot(False)
|
||||
self._send_message("🛑 <b>Autopilot DISABLED</b>\n\nI will only notify you of new listings.")
|
||||
else:
|
||||
self._send_message("Usage: /autopilot on|off")
|
||||
|
||||
def _handle_status_command(self):
|
||||
state = self.monitor.load_state()
|
||||
autopilot = state.get("autopilot", False)
|
||||
applications = self.monitor.load_applications()
|
||||
status = "🤖 <b>Autopilot:</b> " + ("ON ✅" if autopilot else "OFF ❌")
|
||||
status += f"\n📝 <b>Applications sent:</b> {len(applications)}"
|
||||
by_company = {}
|
||||
for app in applications.values():
|
||||
company = app.get("company", "unknown")
|
||||
by_company[company] = by_company.get(company, 0) + 1
|
||||
if by_company:
|
||||
status += "\n\n<b>By company:</b>"
|
||||
for company, count in sorted(by_company.items()):
|
||||
status += f"\n • {company}: {count}"
|
||||
self._send_message(status)
|
||||
|
||||
def _handle_help_command(self):
|
||||
help_text = """🏠 <b>InBerlin Monitor Commands</b>
|
||||
|
||||
/autopilot on - Enable automatic applications
|
||||
/autopilot off - Disable automatic applications
|
||||
/status - Show current status and stats
|
||||
/plot - Show weekly listing patterns
|
||||
/help - Show this help message
|
||||
|
||||
When autopilot is ON, I will automatically apply to new listings."""
|
||||
self._send_message(help_text)
|
||||
|
||||
def _handle_unknown_command(self, text):
|
||||
cmd = text.split()[0] if text else text
|
||||
self._send_message(f"❓ Unknown command: {cmd}")
|
||||
|
||||
def _handle_error_rate_command(self):
|
||||
"""Generate and send a plot showing success vs failure ratio for autopilot applications."""
|
||||
logger.info("Generating autopilot errorrate plot...")
|
||||
try:
|
||||
plot_path, summary = self._generate_error_rate_plot()
|
||||
if plot_path:
|
||||
caption = "📉 <b>Autopilot Success vs Failure</b>\n\n" + summary
|
||||
self._send_photo(plot_path, caption)
|
||||
else:
|
||||
self._send_message("📉 Not enough application data to generate errorrate plot.")
|
||||
except Exception as e:
|
||||
logger.error(f"Error generating error rate plot: {e}")
|
||||
self._send_message("📉 Error generating error rate plot.")
|
||||
|
||||
def _send_message(self, text):
|
||||
"""Send a text message to the configured Telegram chat."""
|
||||
if not self.bot_token or not self.chat_id:
|
||||
logger.warning("Telegram bot token or chat ID not configured, cannot send message")
|
||||
return
|
||||
url = f"https://api.telegram.org/bot{self.bot_token}/sendMessage"
|
||||
payload = {"chat_id": self.chat_id, "text": text, "parse_mode": "HTML"}
|
||||
try:
|
||||
response = requests.post(url, json=payload, timeout=10)
|
||||
if not response.ok:
|
||||
logger.error(f"Failed to send Telegram message: {response.text}")
|
||||
except Exception as e:
|
||||
logger.error(f"Error while sending Telegram message: {e}")
|
||||
|
||||
def _send_photo(self, photo_path, caption):
|
||||
"""Send a photo to the configured Telegram chat."""
|
||||
if not self.bot_token or not self.chat_id:
|
||||
logger.warning("Telegram bot token or chat ID not configured, cannot send photo")
|
||||
return
|
||||
url = f"https://api.telegram.org/bot{self.bot_token}/sendPhoto"
|
||||
with open(photo_path, "rb") as photo:
|
||||
payload = {"chat_id": self.chat_id, "caption": caption, "parse_mode": "HTML"}
|
||||
files = {"photo": photo}
|
||||
try:
|
||||
response = requests.post(url, data=payload, files=files, timeout=10)
|
||||
if not response.ok:
|
||||
logger.error(f"Failed to send Telegram photo: {response.text}")
|
||||
except Exception as e:
|
||||
logger.error(f"Error while sending Telegram photo: {e}")
|
||||
|
||||
def _generate_error_rate_plot(self):
|
||||
"""Placeholder for generating an error rate plot."""
|
||||
logger.warning("_generate_error_rate_plot is not implemented.")
|
||||
return None, "Error rate plot generation not implemented."
|
||||
|
||||
def _handle_plot_command(self):
|
||||
"""Placeholder for handling the /plot command."""
|
||||
logger.warning("_handle_plot_command is not implemented.")
|
||||
self._send_message("📊 Plot command is not implemented yet.")
|
||||
Loading…
Add table
Add a link
Reference in a new issue