mirror of
https://github.com/Steffo99/greed.git
synced 2024-11-22 05:54:18 +00:00
More work on the basic bot code
pass, ..., and NotImplementedError() are unfinished parts of code
This commit is contained in:
parent
94d4ac4ff9
commit
56d06ea1d2
3 changed files with 49 additions and 7 deletions
23
core.py
23
core.py
|
@ -4,6 +4,7 @@ import configparser
|
|||
import telegram
|
||||
import threading
|
||||
import time
|
||||
import strings
|
||||
|
||||
# Check if a configuration file exists, create one if it doesn't and get the template version number.
|
||||
with open("config/template_config.ini") as template_file:
|
||||
|
@ -52,25 +53,33 @@ except telegram.error.Unauthorized:
|
|||
"Fix it, then restart this script.")
|
||||
sys.exit(1)
|
||||
|
||||
# Create a dictionary containing the chat instances threads and the pipes from the main thread to the chat instance thread
|
||||
# {"1234": (<Thread>, <Pipe>)}
|
||||
chat_threads = {}
|
||||
# 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
|
||||
update_offset = None
|
||||
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
|
||||
updates = bot.get_updates(offset=update_offset)
|
||||
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
|
||||
# TODO: add stuff here
|
||||
# If the update is a inline keyboard press...
|
||||
if update.inline_query is not None:
|
||||
...
|
||||
pass
|
||||
# Mark the update as read by increasing the update_offset
|
||||
next_update = update.update_id + 1
|
||||
# Temporarily prevent rate limits (remove this later)
|
||||
time.sleep(5)
|
5
strings.py
Normal file
5
strings.py
Normal file
|
@ -0,0 +1,5 @@
|
|||
# Strings / localization file for greed
|
||||
# Can be edited, but DON'T REMOVE THE REPLACEMENT FIELDS (words surrounded by {curly braces})
|
||||
|
||||
# Error: message received not in a private chat
|
||||
error_nonprivate_chat = "⚠️ Questo bot funziona solo in chat private."
|
28
worker.py
Normal file
28
worker.py
Normal file
|
@ -0,0 +1,28 @@
|
|||
import multiprocessing
|
||||
import telegram
|
||||
|
||||
class ChatWorker:
|
||||
"""A worker for a single conversation. A new one should be created every time the /start command is sent."""
|
||||
|
||||
def __init__(self, bot: telegram.Bot, chat: telegram.Chat):
|
||||
# A pipe connecting the main process to the chat process is created
|
||||
in_pipe, out_pipe = multiprocessing.Pipe(duplex=False)
|
||||
# The sending pipe is stored in the ChatWorker class, allowing the forwarding of messages to the chat process
|
||||
self.pipe = in_pipe
|
||||
# A new process running the conversation handler is created, and the receiving pipe is passed to its arguments to enable the receiving of messages
|
||||
self.process = multiprocessing.Process(target=conversation_handler, args=(bot, chat, out_pipe))
|
||||
|
||||
def start(self):
|
||||
"""Start the worker process."""
|
||||
self.process.start()
|
||||
|
||||
def stop(self):
|
||||
# Gracefully stop the worker process
|
||||
# TODO: send a stop message to the process
|
||||
raise NotImplementedError()
|
||||
# Wait for the process to stop
|
||||
self.process.join()
|
||||
|
||||
|
||||
def conversation_handler(bot: telegram.Bot, chat: telegram.Chat, pipe: multiprocessing.Connection):
|
||||
raise NotImplementedError()
|
Loading…
Reference in a new issue