1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
import datetime
import signal
import sys
import logging
from os import getenv
from telegram.ext import Updater, CommandHandler, CallbackContext
from telegram import Update
from sheets_client import authenticate_and_fetch_sheets_data
from parser import filter_only_this_weeks_rows
chats_mem_cache = set()
ALLOWED_USER_IDS = map(lambda id: int(id), getenv("TELEGRAM_USER_IDS", "").split(","))
def _signal_handler(sig, frame):
print("You pressed Ctrl+C!")
updater = frame.f_locals["updater"]
for chat_id in chats_mem_cache:
updater.bot.send_message(chat_id=chat_id, text="HommaBot-palvelin sammutettu!")
updater.stop()
sys.exit(0)
def _callback_alarm(context: CallbackContext):
chat_id: int = context.job.context
logging.info(f"Authenticating and fetching data from Sheets...")
rows = authenticate_and_fetch_sheets_data()
logging.info(f"Processing data...")
processed_data = filter_only_this_weeks_rows(rows)
logging.info(f"Sending sheet data to chat id {chat_id}...")
message = "Tällä viikolla tehtävät hommat:\n" + "\n".join(
map(lambda task: f"{task[1]} ({task[2]})", processed_data)
)
context.bot.send_message(chat_id=chat_id, text=message)
def _callback_timer(update: Update, context: CallbackContext):
chat_id = update.message.chat_id
user_id = update.message.from_user.id
if user_id not in ALLOWED_USER_IDS:
logging.info(f"User {user_id} is not allowed to start timer.")
update.message.reply_text(
"Käyttäjälläsi ei ole oikeuksia startata HommaBottia."
)
return
logging.info(f"Starting timer in chat id {chat_id}...")
update.message.reply_text(
"HommaBot startattu. Uusia hommapäivityksiä viikon välein."
)
chats_mem_cache.add(chat_id)
context.job_queue.run_repeating(
_callback_alarm,
datetime.timedelta(weeks=1),
1, # run once immediately after 1 sec
context=chat_id,
)
def _stop_timer(update: Update, context: CallbackContext):
chat_id = update.message.chat_id
user_id = update.message.from_user.id
if user_id not in ALLOWED_USER_IDS:
logging.info(f"User {user_id} is not allowed to stop timer.")
update.message.reply_text(
"Käyttäjälläsi ei ole oikeuksia pysäyttää HommaBottia."
)
return
logging.info(f"Stopping timer in chat id {chat_id}...")
update.message.reply_text("HommaBot pysäytetty!")
chats_mem_cache.remove(chat_id)
context.job_queue.stop()
def run_telegram():
tg_token = getenv("TELEGRAM_TOKEN")
logging.info("Creating Telegram bot object...")
updater = Updater(tg_token)
logging.info("Registering Telegram bot /start handler...")
updater.dispatcher.add_handler(
CommandHandler("start", _callback_timer, pass_job_queue=True)
)
logging.info("Registering Telegram bot /stop handler...")
updater.dispatcher.add_handler(
CommandHandler("stop", _stop_timer, pass_job_queue=True)
)
logging.info("Starting Telegram bot (HommaBot)...")
updater.start_polling()
logging.info("Registering SIGINT handler...")
signal.signal(signal.SIGINT, _signal_handler)
logging.info("Bot started successfully. Pausing main thread... (this is expected)")
signal.pause()
|
