95 lines
3.9 KiB
Python
95 lines
3.9 KiB
Python
import sys
|
|
import telegram
|
|
import time
|
|
import strings
|
|
import worker
|
|
import configloader
|
|
|
|
def main():
|
|
"""The core code of the program. Should be run only in the main process!"""
|
|
|
|
# Load the config from config.ini
|
|
config = configloader.load_config()
|
|
|
|
# Create a bot instance
|
|
bot = telegram.Bot(config["Telegram"]["token"])
|
|
|
|
# Test the specified token
|
|
try:
|
|
bot.get_me()
|
|
except telegram.error.Unauthorized:
|
|
print("The token you have entered in the config file is invalid.\n"
|
|
"Fix it, then restart this script.")
|
|
sys.exit(1)
|
|
|
|
# Create a dictionary linking the chat ids to the ChatWorker objects
|
|
# {"1234": <ChatWorker>}
|
|
chat_workers = {}
|
|
|
|
# Current update offset; if None it will get the last 100 unparsed messages
|
|
next_update = None
|
|
|
|
# Main loop of the program
|
|
while True:
|
|
# Get a new batch of 100 updates and mark the last 100 parsed as read
|
|
# TODO: handle possible errors
|
|
updates = bot.get_updates(offset=next_update)
|
|
# Parse all the updates
|
|
for update in updates:
|
|
# If the update is a message...
|
|
if update.message is not None:
|
|
# Ensure the message has been sent in a private chat
|
|
if update.message.chat.type != "private":
|
|
# Notify the chat
|
|
bot.send_message(update.message.chat.id, strings.error_nonprivate_chat)
|
|
# Skip the update
|
|
continue
|
|
# If the message is a start command...
|
|
if update.message.text == "/start":
|
|
# Check if a worker already exists for that chat
|
|
old_worker = chat_workers.get(update.message.chat.id)
|
|
# If it exists, gracefully stop the worker
|
|
if old_worker:
|
|
old_worker.stop()
|
|
# Initialize a new worker for the chat
|
|
new_worker = worker.ChatWorker(bot, update.message.chat)
|
|
# Start the worker
|
|
new_worker.start()
|
|
# Store the worker in the dictionary
|
|
chat_workers[update.message.chat.id] = new_worker
|
|
# Skip the update
|
|
continue
|
|
# Otherwise, forward the update to the corresponding worker
|
|
receiving_worker = chat_workers.get(update.message.chat.id)
|
|
# Ensure a worker exists for the chat and is alive
|
|
if receiving_worker is None or not receiving_worker.process.is_alive():
|
|
# Suggest that the user restarts the chat with /start
|
|
bot.send_message(update.message.chat.id, strings.error_no_worker_for_chat)
|
|
# Skip the update
|
|
continue
|
|
# Forward the update to the worker
|
|
receiving_worker.pipe.send(update)
|
|
# If the update is a inline keyboard press...
|
|
if update.inline_query is not None:
|
|
# Forward the update to the corresponding worker
|
|
receiving_worker = chat_workers.get(update.inline_query.chat.id)
|
|
# Ensure a worker exists for the chat
|
|
if receiving_worker is None:
|
|
# Suggest that the user restarts the chat with /start
|
|
bot.send_message(update.inline_query.chat.id, strings.error_no_worker_for_chat)
|
|
# Skip the update
|
|
continue
|
|
# Forward the update to the worker
|
|
receiving_worker.pipe.send(update)
|
|
# If there were any updates...
|
|
if len(updates):
|
|
# Mark them as read by increasing the update_offset
|
|
next_update = updates[-1].update_id + 1
|
|
# Temporarily prevent rate limits
|
|
# TODO: (remove this later)
|
|
time.sleep(5)
|
|
|
|
|
|
# Run the main function only in the main process
|
|
if __name__ == "__main__":
|
|
main() |