1
Fork 0
mirror of https://github.com/Steffo99/greed.git synced 2024-11-30 17:44:18 +00:00
greed/worker.py

1549 lines
79 KiB
Python
Raw Normal View History

import os
import sys
2018-02-19 12:47:43 +00:00
import datetime
2020-09-07 15:13:27 +00:00
import logging
2017-12-14 07:53:16 +00:00
import queue as queuem
2018-01-03 13:52:05 +00:00
import re
2020-09-07 15:13:27 +00:00
import threading
import traceback
2020-09-07 15:13:27 +00:00
import uuid
2018-02-01 09:06:40 +00:00
from html import escape
2020-09-07 15:13:27 +00:00
from typing import *
2018-04-19 07:37:42 +00:00
import requests
import sqlalchemy
2020-09-07 15:13:27 +00:00
import telegram
import database as db
import localization
import nuconfig
log = logging.getLogger(__name__)
2017-12-13 10:20:53 +00:00
2017-12-12 18:56:38 +00:00
class StopSignal:
"""A data class that should be sent to the worker when the conversation has to be stopped abnormally."""
2020-09-07 15:13:27 +00:00
def __init__(self, reason: str = ""):
2017-12-12 18:56:38 +00:00
self.reason = reason
2018-02-08 09:18:53 +00:00
class CancelSignal:
"""An empty class that is added to the queue whenever the user presses a cancel inline button."""
pass
class Worker(threading.Thread):
2017-12-14 08:40:03 +00:00
"""A worker for a single conversation. A new one is created every time the /start command is sent."""
def __init__(self,
bot,
chat: telegram.Chat,
telegram_user: telegram.User,
cfg: nuconfig.NuConfig,
engine,
2020-09-07 15:13:27 +00:00
*args,
**kwargs):
2017-12-14 08:40:03 +00:00
# Initialize the thread
2018-05-30 09:46:18 +00:00
super().__init__(name=f"Worker {chat.id}", *args, **kwargs)
# Store the bot, chat info and config inside the class
self.bot = bot
2018-04-05 07:57:59 +00:00
self.chat: telegram.Chat = chat
self.telegram_user: telegram.User = telegram_user
self.cfg = cfg
2020-09-09 14:55:46 +00:00
self.loc = None
2017-12-21 09:42:23 +00:00
# Open a new database session
log.debug(f"Opening new database session for {self.name}")
self.session = sqlalchemy.orm.sessionmaker(bind=engine)()
2017-12-21 09:42:23 +00:00
# Get the user db data from the users and admin tables
self.user: Optional[db.User] = None
self.admin: Optional[db.Admin] = None
# The sending pipe is stored in the Worker class, allowing the forwarding of messages to the chat process
2017-12-14 07:53:16 +00:00
self.queue = queuem.Queue()
# The current active invoice payload; reject all invoices with a different payload
self.invoice_payload = None
# The price class of this worker.
self.Price = self.price_factory()
def __repr__(self):
return f"<{self.__class__.__qualname__} {self.chat.id}>"
# noinspection PyMethodParameters
def price_factory(worker):
class Price:
"""The base class for the prices in greed.
Its int value is in minimum units, while its float and str values are in decimal format."""
def __init__(self, value: Union[int, float, str, "Price"]):
if isinstance(value, int):
# Keep the value as it is
self.value = int(value)
elif isinstance(value, float):
# Convert the value to minimum units
self.value = int(value * (10 ** worker.cfg["Payments"]["currency_exp"]))
elif isinstance(value, str):
# Remove decimal points, then cast to int
self.value = int(float(value.replace(",", ".")) * (10 ** worker.cfg["Payments"]["currency_exp"]))
elif isinstance(value, Price):
# Copy self
self.value = value.value
def __repr__(self):
return f"<{self.__class__.__qualname__} of value {self.value}>"
def __str__(self):
return worker.loc.get(
"currency_format_string",
symbol=worker.cfg["Payments"]["currency_symbol"],
value="{0:.2f}".format(self.value / (10 ** worker.cfg["Payments"]["currency_exp"]))
)
def __int__(self):
return self.value
def __float__(self):
return self.value / (10 ** worker.cfg["Payments"]["currency_exp"])
def __ge__(self, other):
return self.value >= Price(other).value
def __le__(self, other):
return self.value <= Price(other).value
def __eq__(self, other):
return self.value == Price(other).value
def __gt__(self, other):
return self.value > Price(other).value
def __lt__(self, other):
return self.value < Price(other).value
def __add__(self, other):
return Price(self.value + Price(other).value)
def __sub__(self, other):
return Price(self.value - Price(other).value)
def __mul__(self, other):
return Price(int(self.value * other))
def __floordiv__(self, other):
return Price(int(self.value // other))
def __radd__(self, other):
return self.__add__(other)
def __rsub__(self, other):
return Price(Price(other).value - self.value)
def __rmul__(self, other):
return self.__mul__(other)
def __iadd__(self, other):
self.value += Price(other).value
return self
def __isub__(self, other):
self.value -= Price(other).value
return self
def __imul__(self, other):
self.value *= other
self.value = int(self.value)
return self
def __ifloordiv__(self, other):
self.value //= other
return self
return Price
2017-12-14 08:40:03 +00:00
def run(self):
"""The conversation code."""
log.debug("Starting conversation")
2018-01-10 10:29:02 +00:00
# Get the user db data from the users and admin tables
self.user = self.session.query(db.User).filter(db.User.user_id == self.chat.id).one_or_none()
self.admin = self.session.query(db.Admin).filter(db.Admin.user_id == self.chat.id).one_or_none()
2017-12-21 09:42:23 +00:00
# If the user isn't registered, create a new record and add it to the db
if self.user is None:
2018-04-26 07:59:47 +00:00
# Check if there are other registered users: if there aren't any, the first user will be owner of the bot
2018-05-02 14:32:16 +00:00
will_be_owner = (self.session.query(db.Admin).first() is None)
2017-12-21 09:42:23 +00:00
# Create the new record
self.user = db.User(w=self)
2017-12-21 09:42:23 +00:00
# Add the new record to the db
self.session.add(self.user)
2018-04-26 07:59:47 +00:00
# If the will be owner flag is set
if will_be_owner:
# Become owner
2021-05-07 12:00:15 +00:00
self.admin = db.Admin(user=self.user,
2018-04-26 07:59:47 +00:00
edit_products=True,
receive_orders=True,
create_transactions=True,
display_on_help=True,
is_owner=True,
live_mode=False)
# Add the admin to the transaction
self.session.add(self.admin)
2017-12-21 09:42:23 +00:00
# Commit the transaction
self.session.commit()
log.info(f"Created new user: {self.user}")
if will_be_owner:
log.warning(f"User was auto-promoted to Admin as no other admins existed: {self.user}")
2020-05-03 10:40:11 +00:00
# Create the localization object
self.__create_localization()
# Capture exceptions that occour during the conversation
2020-04-23 15:47:19 +00:00
# noinspection PyBroadException
try:
# Welcome the user to the bot
if self.cfg["Appearance"]["display_welcome_message"] == "yes":
self.bot.send_message(self.chat.id, self.loc.get("conversation_after_start"))
# If the user is not an admin, send him to the user menu
if self.admin is None:
self.__user_menu()
# If the user is an admin, send him to the admin menu
else:
# Clear the live orders flag
self.admin.live_mode = False
# Commit the change
self.session.commit()
# Open the admin menu
self.__admin_menu()
except Exception as e:
# Try to notify the user of the exception
2020-04-23 15:47:19 +00:00
# noinspection PyBroadException
try:
self.bot.send_message(self.chat.id, self.loc.get("fatal_conversation_exception"))
except Exception as ne:
log.error(f"Failed to notify the user of a conversation exception: {ne}")
log.error(f"Exception in {self}: {e}")
traceback.print_exception(*sys.exc_info())
2020-07-06 21:54:13 +00:00
def is_ready(self):
# Change this if more parameters are added!
return self.loc is not None
def stop(self, reason: str = ""):
2017-12-12 18:56:38 +00:00
"""Gracefully stop the worker process"""
2017-12-14 07:53:16 +00:00
# Send a stop message to the thread
self.queue.put(StopSignal(reason))
# Wait for the thread to stop
2017-12-14 08:40:03 +00:00
self.join()
def update_user(self) -> db.User:
"""Update the user data."""
log.debug("Fetching updated user data from the database")
self.user = self.session.query(db.User).filter(db.User.user_id == self.chat.id).one_or_none()
return self.user
2018-04-05 08:34:14 +00:00
# noinspection PyUnboundLocalVariable
2017-12-21 09:42:23 +00:00
def __receive_next_update(self) -> telegram.Update:
2017-12-14 08:40:03 +00:00
"""Get the next update from the queue.
If no update is found, block the process until one is received.
If a stop signal is sent, try to gracefully stop the thread."""
# Pop data from the queue
try:
data = self.queue.get(timeout=self.cfg["Telegram"]["conversation_timeout"])
2017-12-14 08:40:03 +00:00
except queuem.Empty:
# If the conversation times out, gracefully stop the thread
2018-04-16 19:20:02 +00:00
self.__graceful_stop(StopSignal("timeout"))
2017-12-14 08:40:03 +00:00
# Check if the data is a stop signal instance
if isinstance(data, StopSignal):
# Gracefully stop the process
2018-04-16 19:20:02 +00:00
self.__graceful_stop(data)
2017-12-14 08:40:03 +00:00
# Return the received update
return data
2017-12-13 10:20:53 +00:00
2018-04-05 08:34:14 +00:00
def __wait_for_specific_message(self,
items: List[str],
cancellable: bool = False) -> Union[str, CancelSignal]:
2017-12-21 09:42:23 +00:00
"""Continue getting updates until until one of the strings contained in the list is received as a message."""
log.debug("Waiting for a specific message...")
2017-12-21 09:42:23 +00:00
while True:
# Get the next update
update = self.__receive_next_update()
2020-04-08 12:51:44 +00:00
# If a CancelSignal is received...
if isinstance(update, CancelSignal):
2020-04-08 12:51:44 +00:00
# And the wait is cancellable...
if cancellable:
# Return the CancelSignal
return update
else:
# Ignore the signal
continue
2017-12-21 09:42:23 +00:00
# Ensure the update contains a message
if update.message is None:
continue
# Ensure the message contains text
if update.message.text is None:
continue
# Check if the message is contained in the list
if update.message.text not in items:
continue
# Return the message text
return update.message.text
def __wait_for_regex(self, regex: str, cancellable: bool = False) -> Union[str, CancelSignal]:
2018-01-03 13:52:05 +00:00
"""Continue getting updates until the regex finds a match in a message, then return the first capture group."""
log.debug("Waiting for a regex...")
2018-01-03 13:52:05 +00:00
while True:
# Get the next update
update = self.__receive_next_update()
2020-04-08 12:51:44 +00:00
# If a CancelSignal is received...
if isinstance(update, CancelSignal):
# And the wait is cancellable...
if cancellable:
# Return the CancelSignal
return update
else:
# Ignore the signal
continue
2018-01-03 13:52:05 +00:00
# Ensure the update contains a message
if update.message is None:
continue
# Ensure the message contains text
if update.message.text is None:
continue
# Try to match the regex with the received message
match = re.search(regex, update.message.text, re.DOTALL)
2018-01-03 13:52:05 +00:00
# Ensure there is a match
if match is None:
continue
# Return the first capture group
return match.group(1)
2018-04-05 08:34:14 +00:00
def __wait_for_precheckoutquery(self,
cancellable: bool = False) -> Union[telegram.PreCheckoutQuery, CancelSignal]:
2018-01-10 10:29:02 +00:00
"""Continue getting updates until a precheckoutquery is received.
The payload is checked by the core before forwarding the message."""
log.debug("Waiting for a PreCheckoutQuery...")
while True:
# Get the next update
update = self.__receive_next_update()
2020-04-08 12:51:44 +00:00
# If a CancelSignal is received...
if isinstance(update, CancelSignal):
# And the wait is cancellable...
if cancellable:
# Return the CancelSignal
return update
else:
# Ignore the signal
continue
# Ensure the update contains a precheckoutquery
if update.pre_checkout_query is None:
continue
# Return the precheckoutquery
return update.pre_checkout_query
2020-07-06 22:07:20 +00:00
def __wait_for_successfulpayment(self,
cancellable: bool = False) -> Union[telegram.SuccessfulPayment, CancelSignal]:
2018-01-10 10:29:02 +00:00
"""Continue getting updates until a successfulpayment is received."""
log.debug("Waiting for a SuccessfulPayment...")
2018-01-10 10:29:02 +00:00
while True:
# Get the next update
update = self.__receive_next_update()
2020-07-06 22:07:20 +00:00
# If a CancelSignal is received...
if isinstance(update, CancelSignal):
# And the wait is cancellable...
if cancellable:
# Return the CancelSignal
return update
else:
# Ignore the signal
continue
2018-01-10 10:29:02 +00:00
# Ensure the update contains a message
if update.message is None:
continue
# Ensure the message is a successfulpayment
if update.message.successful_payment is None:
continue
# Return the successfulpayment
return update.message.successful_payment
def __wait_for_photo(self, cancellable: bool = False) -> Union[List[telegram.PhotoSize], CancelSignal]:
2018-02-15 09:37:51 +00:00
"""Continue getting updates until a photo is received, then return it."""
log.debug("Waiting for a photo...")
2018-02-01 09:06:40 +00:00
while True:
# Get the next update
update = self.__receive_next_update()
2020-04-08 12:51:44 +00:00
# If a CancelSignal is received...
if isinstance(update, CancelSignal):
# And the wait is cancellable...
if cancellable:
# Return the CancelSignal
return update
else:
# Ignore the signal
continue
2018-02-01 09:06:40 +00:00
# Ensure the update contains a message
if update.message is None:
continue
# Ensure the message contains a photo
if update.message.photo is None:
continue
# Return the photo array
return update.message.photo
2020-04-08 12:51:44 +00:00
def __wait_for_inlinekeyboard_callback(self, cancellable: bool = False) \
-> Union[telegram.CallbackQuery, CancelSignal]:
2018-02-15 09:37:51 +00:00
"""Continue getting updates until an inline keyboard callback is received, then return it."""
log.debug("Waiting for a CallbackQuery...")
2018-02-15 09:37:51 +00:00
while True:
# Get the next update
update = self.__receive_next_update()
2020-04-08 12:51:44 +00:00
# If a CancelSignal is received...
if isinstance(update, CancelSignal):
# And the wait is cancellable...
if cancellable:
# Return the CancelSignal
return update
else:
# Ignore the signal
continue
2018-02-15 09:37:51 +00:00
# Ensure the update is a CallbackQuery
if update.callback_query is None:
continue
2018-04-01 16:10:14 +00:00
# Answer the callbackquery
self.bot.answer_callback_query(update.callback_query.id)
2018-02-15 09:37:51 +00:00
# Return the callbackquery
return update.callback_query
def __user_select(self) -> Union[db.User, CancelSignal]:
2018-04-30 14:14:02 +00:00
"""Select an user from the ones in the database."""
log.debug("Waiting for a user selection...")
2018-04-30 14:14:02 +00:00
# Find all the users in the database
users = self.session.query(db.User).order_by(db.User.user_id).all()
# Create a list containing all the keyboard button strings
keyboard_buttons = [[self.loc.get("menu_cancel")]]
2018-04-30 14:14:02 +00:00
# Add to the list all the users
for user in users:
keyboard_buttons.append([user.identifiable_str()])
# Create the keyboard
keyboard = telegram.ReplyKeyboardMarkup(keyboard_buttons, one_time_keyboard=True)
# Keep asking until a result is returned
while True:
# Send the keyboard
self.bot.send_message(self.chat.id, self.loc.get("conversation_admin_select_user"), reply_markup=keyboard)
2018-04-30 14:14:02 +00:00
# Wait for a reply
reply = self.__wait_for_regex("user_([0-9]+)", cancellable=True)
# Propagate CancelSignals
if isinstance(reply, CancelSignal):
return reply
2018-04-30 14:14:02 +00:00
# Find the user in the database
user = self.session.query(db.User).filter_by(user_id=int(reply)).one_or_none()
# Ensure the user exists
if not user:
self.bot.send_message(self.chat.id, self.loc.get("error_user_does_not_exist"))
2018-04-30 14:14:02 +00:00
continue
return user
2017-12-21 09:42:23 +00:00
def __user_menu(self):
"""Function called from the run method when the user is not an administrator.
Normal bot actions should be placed here."""
log.debug("Displaying __user_menu")
2017-12-22 08:56:03 +00:00
# Loop used to returning to the menu after executing a command
while True:
# Create a keyboard with the user main menu
keyboard = [[telegram.KeyboardButton(self.loc.get("menu_order"))],
[telegram.KeyboardButton(self.loc.get("menu_order_status"))],
[telegram.KeyboardButton(self.loc.get("menu_add_credit"))],
2020-05-03 10:40:11 +00:00
[telegram.KeyboardButton(self.loc.get("menu_language"))],
[telegram.KeyboardButton(self.loc.get("menu_help")),
telegram.KeyboardButton(self.loc.get("menu_bot_info"))]]
2017-12-22 08:56:03 +00:00
# Send the previously created keyboard to the user (ensuring it can be clicked only 1 time)
2018-04-05 07:30:32 +00:00
self.bot.send_message(self.chat.id,
2020-05-03 10:40:11 +00:00
self.loc.get("conversation_open_user_menu",
credit=self.Price(self.user.credit)),
2018-04-19 07:55:36 +00:00
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
2017-12-22 08:56:03 +00:00
# Wait for a reply from the user
2020-05-03 10:40:11 +00:00
selection = self.__wait_for_specific_message([
self.loc.get("menu_order"),
self.loc.get("menu_order_status"),
self.loc.get("menu_add_credit"),
self.loc.get("menu_language"),
self.loc.get("menu_help"),
self.loc.get("menu_bot_info"),
])
# After the user reply, update the user data
2018-05-15 15:00:24 +00:00
self.update_user()
2017-12-22 08:56:03 +00:00
# If the user has selected the Order option...
if selection == self.loc.get("menu_order"):
2017-12-22 08:56:03 +00:00
# Open the order menu
self.__order_menu()
# If the user has selected the Order Status option...
elif selection == self.loc.get("menu_order_status"):
2017-12-22 08:56:03 +00:00
# Display the order(s) status
self.__order_status()
# If the user has selected the Add Credit option...
elif selection == self.loc.get("menu_add_credit"):
2017-12-22 08:56:03 +00:00
# Display the add credit menu
self.__add_credit_menu()
2020-05-03 10:40:11 +00:00
# If the user has selected the Language option...
elif selection == self.loc.get("menu_language"):
# Display the language menu
self.__language_menu()
2017-12-22 08:56:03 +00:00
# If the user has selected the Bot Info option...
elif selection == self.loc.get("menu_bot_info"):
2017-12-22 08:56:03 +00:00
# Display information about the bot
self.__bot_info()
2018-04-16 10:09:51 +00:00
# If the user has selected the Help option...
elif selection == self.loc.get("menu_help"):
2018-04-16 10:09:51 +00:00
# Go to the Help menu
self.__help_menu()
2017-12-22 08:56:03 +00:00
def __order_menu(self):
2018-02-05 11:49:21 +00:00
"""User menu to order products from the shop."""
log.debug("Displaying __order_menu")
2018-02-15 09:37:51 +00:00
# Get the products list from the db
2018-03-06 16:39:02 +00:00
products = self.session.query(db.Product).filter_by(deleted=False).all()
2018-02-15 09:37:51 +00:00
# Create a dict to be used as 'cart'
# The key is the message id of the product list
cart: Dict[List[db.Product, int]] = {}
2018-02-15 09:37:51 +00:00
# Initialize the products list
for product in products:
# If the product is not for sale, don't display it
if product.price is None:
continue
# Send the message without the keyboard to get the message id
message = product.send_as_message(w=self, chat_id=self.chat.id)
2018-02-15 09:37:51 +00:00
# Add the product to the cart
cart[message['message_id']] = [product, 0]
2018-02-15 09:37:51 +00:00
# Create the inline keyboard to add the product to the cart
2020-05-03 10:40:11 +00:00
inline_keyboard = telegram.InlineKeyboardMarkup(
[[telegram.InlineKeyboardButton(self.loc.get("menu_add_to_cart"), callback_data="cart_add")]]
)
2018-02-15 09:37:51 +00:00
# Edit the sent message and add the inline keyboard
if product.image is None:
2018-04-05 08:43:27 +00:00
self.bot.edit_message_text(chat_id=self.chat.id,
message_id=message['message_id'],
text=product.text(w=self),
2018-04-05 08:43:27 +00:00
reply_markup=inline_keyboard)
2018-02-15 09:37:51 +00:00
else:
2018-04-05 08:43:27 +00:00
self.bot.edit_message_caption(chat_id=self.chat.id,
message_id=message['message_id'],
caption=product.text(w=self),
2018-04-05 08:43:27 +00:00
reply_markup=inline_keyboard)
2018-02-15 09:37:51 +00:00
# Create the keyboard with the cancel button
inline_keyboard = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_cancel"),
2018-04-05 08:43:27 +00:00
callback_data="cart_cancel")]])
2018-02-15 09:37:51 +00:00
# Send a message containing the button to cancel or pay
2020-05-03 10:40:11 +00:00
final_msg = self.bot.send_message(self.chat.id,
self.loc.get("conversation_cart_actions"),
reply_markup=inline_keyboard)
2018-02-15 09:37:51 +00:00
# Wait for user input
while True:
callback = self.__wait_for_inlinekeyboard_callback()
# React to the user input
# If the cancel button has been pressed...
if callback.data == "cart_cancel":
# Stop waiting for user input and go back to the previous menu
return
# If a Add to Cart button has been pressed...
elif callback.data == "cart_add":
# Get the selected product, ensuring it exists
p = cart.get(callback.message.message_id)
if p is None:
continue
product = p[0]
2018-02-15 09:37:51 +00:00
# Add 1 copy to the cart
cart[callback.message.message_id][1] += 1
# Create the product inline keyboard
2018-04-05 08:43:27 +00:00
product_inline_keyboard = telegram.InlineKeyboardMarkup(
[
2020-05-03 10:40:11 +00:00
[telegram.InlineKeyboardButton(self.loc.get("menu_add_to_cart"),
callback_data="cart_add"),
telegram.InlineKeyboardButton(self.loc.get("menu_remove_from_cart"),
callback_data="cart_remove")]
2018-04-05 08:43:27 +00:00
])
2018-02-15 09:37:51 +00:00
# Create the final inline keyboard
2018-04-05 08:43:27 +00:00
final_inline_keyboard = telegram.InlineKeyboardMarkup(
[
[telegram.InlineKeyboardButton(self.loc.get("menu_cancel"), callback_data="cart_cancel")],
[telegram.InlineKeyboardButton(self.loc.get("menu_done"), callback_data="cart_done")]
2018-04-05 08:43:27 +00:00
])
2018-02-15 09:37:51 +00:00
# Edit both the product and the final message
if product.image is None:
2018-04-05 08:43:27 +00:00
self.bot.edit_message_text(chat_id=self.chat.id,
message_id=callback.message.message_id,
text=product.text(w=self,
2020-05-03 10:40:11 +00:00
cart_qty=cart[callback.message.message_id][1]),
2018-04-05 08:43:27 +00:00
reply_markup=product_inline_keyboard)
2018-02-15 09:37:51 +00:00
else:
2018-04-05 08:43:27 +00:00
self.bot.edit_message_caption(chat_id=self.chat.id,
message_id=callback.message.message_id,
caption=product.text(w=self,
2020-05-03 10:40:11 +00:00
cart_qty=cart[callback.message.message_id][1]),
2018-04-05 08:43:27 +00:00
reply_markup=product_inline_keyboard)
2020-04-04 11:06:57 +00:00
self.bot.edit_message_text(
chat_id=self.chat.id,
message_id=final_msg.message_id,
2020-05-03 10:40:11 +00:00
text=self.loc.get("conversation_confirm_cart",
product_list=self.__get_cart_summary(cart),
total_cost=str(self.__get_cart_value(cart))),
2020-04-04 11:06:57 +00:00
reply_markup=final_inline_keyboard)
2018-02-15 09:37:51 +00:00
# If the Remove from cart button has been pressed...
elif callback.data == "cart_remove":
# Get the selected product, ensuring it exists
p = cart.get(callback.message.message_id)
if p is None:
continue
product = p[0]
2018-02-15 09:37:51 +00:00
# Remove 1 copy from the cart
if cart[callback.message.message_id][1] > 0:
cart[callback.message.message_id][1] -= 1
else:
continue
# Create the product inline keyboard
product_inline_list = [[telegram.InlineKeyboardButton(self.loc.get("menu_add_to_cart"),
2018-04-05 08:43:27 +00:00
callback_data="cart_add")]]
2018-02-15 09:37:51 +00:00
if cart[callback.message.message_id][1] > 0:
product_inline_list[0].append(telegram.InlineKeyboardButton(self.loc.get("menu_remove_from_cart"),
2018-04-17 06:17:16 +00:00
callback_data="cart_remove"))
2018-02-15 09:37:51 +00:00
product_inline_keyboard = telegram.InlineKeyboardMarkup(product_inline_list)
# Create the final inline keyboard
2020-05-03 10:40:11 +00:00
final_inline_list = [[telegram.InlineKeyboardButton(self.loc.get("menu_cancel"),
callback_data="cart_cancel")]]
2018-02-15 09:37:51 +00:00
for product_id in cart:
if cart[product_id][1] > 0:
final_inline_list.append([telegram.InlineKeyboardButton(self.loc.get("menu_done"),
2018-04-05 08:43:27 +00:00
callback_data="cart_done")])
2018-02-15 09:37:51 +00:00
break
final_inline_keyboard = telegram.InlineKeyboardMarkup(final_inline_list)
2018-02-19 12:47:43 +00:00
# Edit the product message
2018-02-15 09:37:51 +00:00
if product.image is None:
self.bot.edit_message_text(chat_id=self.chat.id, message_id=callback.message.message_id,
text=product.text(w=self,
2020-05-03 10:40:11 +00:00
cart_qty=cart[callback.message.message_id][1]),
2018-04-19 07:55:36 +00:00
reply_markup=product_inline_keyboard)
2018-02-15 09:37:51 +00:00
else:
2018-04-05 08:43:27 +00:00
self.bot.edit_message_caption(chat_id=self.chat.id,
message_id=callback.message.message_id,
caption=product.text(w=self,
2020-05-03 10:40:11 +00:00
cart_qty=cart[callback.message.message_id][1]),
2018-04-05 08:43:27 +00:00
reply_markup=product_inline_keyboard)
2020-04-04 11:06:57 +00:00
self.bot.edit_message_text(
chat_id=self.chat.id,
message_id=final_msg.message_id,
2020-05-03 10:40:11 +00:00
text=self.loc.get("conversation_confirm_cart",
product_list=self.__get_cart_summary(cart),
total_cost=str(self.__get_cart_value(cart))),
2020-04-04 11:06:57 +00:00
reply_markup=final_inline_keyboard)
2018-02-19 12:47:43 +00:00
# If the done button has been pressed...
elif callback.data == "cart_done":
# End the loop
break
# Create an inline keyboard with a single skip button
cancel = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_skip"),
2018-04-05 08:43:27 +00:00
callback_data="cmd_cancel")]])
2018-02-19 12:47:43 +00:00
# Ask if the user wants to add notes to the order
self.bot.send_message(self.chat.id, self.loc.get("ask_order_notes"), reply_markup=cancel)
2018-02-19 12:47:43 +00:00
# Wait for user input
notes = self.__wait_for_regex(r"(.*)", cancellable=True)
# Create a new Order
order = db.Order(user=self.user,
creation_date=datetime.datetime.now(),
notes=notes if not isinstance(notes, CancelSignal) else "")
# Add the record to the session and get an ID
self.session.add(order)
# For each product added to the cart, create a new OrderItem
2018-02-19 12:47:43 +00:00
for product in cart:
# Create {quantity} new OrderItems
for i in range(0, cart[product][1]):
2018-05-15 15:31:30 +00:00
order_item = db.OrderItem(product=cart[product][0],
2021-05-05 02:46:58 +00:00
order=order)
2018-05-15 15:31:30 +00:00
self.session.add(order_item)
2018-02-19 12:47:43 +00:00
# Ensure the user has enough credit to make the purchase
credit_required = self.__get_cart_value(cart) - self.user.credit
# Notify user in case of insufficient credit
if credit_required > 0:
self.bot.send_message(self.chat.id, self.loc.get("error_not_enough_credit"))
# Suggest payment for missing credit value if configuration allows refill
if self.cfg["Payments"]["CreditCard"]["credit_card_token"] != "" \
and self.cfg["Appearance"]["refill_on_checkout"] \
and self.Price(self.cfg["Payments"]["CreditCard"]["min_amount"]) <= \
2020-09-07 15:13:27 +00:00
credit_required <= \
self.Price(self.cfg["Payments"]["CreditCard"]["max_amount"]):
self.__make_payment(self.Price(credit_required))
# If afer requested payment credit is still insufficient (either payment failure or cancel)
if self.user.credit < self.__get_cart_value(cart):
2018-02-19 12:47:43 +00:00
# Rollback all the changes
self.session.rollback()
else:
# User has credit and valid order, perform transaction now
self.__order_transaction(order=order, value=-int(self.__get_cart_value(cart)))
2020-05-03 10:40:11 +00:00
def __get_cart_value(self, cart):
# Calculate total items value in cart
value = self.Price(0)
for product in cart:
value += cart[product][0].price * cart[product][1]
return value
2020-05-03 10:40:11 +00:00
def __get_cart_summary(self, cart):
# Create the cart summary
product_list = ""
for product_id in cart:
if cart[product_id][1] > 0:
product_list += cart[product_id][0].text(w=self,
2020-05-03 10:40:11 +00:00
style="short",
cart_qty=cart[product_id][1]) + "\n"
return product_list
def __order_transaction(self, order, value):
2018-02-19 12:47:43 +00:00
# Create a new transaction and add it to the session
transaction = db.Transaction(user=self.user,
2018-03-12 09:19:01 +00:00
value=value,
2021-05-06 22:43:22 +00:00
order=order)
2018-02-19 12:47:43 +00:00
self.session.add(transaction)
# Commit all the changes
self.session.commit()
# Update the user's credit
self.user.recalculate_credit()
2018-02-19 12:47:43 +00:00
# Commit all the changes
self.session.commit()
# Notify admins about new transation
self.__order_notify_admins(order=order)
def __order_notify_admins(self, order):
2018-02-19 12:47:43 +00:00
# Notify the user of the order result
self.bot.send_message(self.chat.id, self.loc.get("success_order_created", order=order.text(w=self,
2020-05-03 10:40:11 +00:00
user=True)))
2018-04-01 16:10:14 +00:00
# Notify the admins (in Live Orders mode) of the new order
admins = self.session.query(db.Admin).filter_by(live_mode=True).all()
# Create the order keyboard
order_keyboard = telegram.InlineKeyboardMarkup(
2018-04-05 08:43:27 +00:00
[
[telegram.InlineKeyboardButton(self.loc.get("menu_complete"), callback_data="order_complete")],
[telegram.InlineKeyboardButton(self.loc.get("menu_refund"), callback_data="order_refund")]
2018-04-05 08:43:27 +00:00
])
2018-03-07 09:48:30 +00:00
# Notify them of the new placed order
for admin in admins:
2018-04-01 16:10:14 +00:00
self.bot.send_message(admin.user_id,
2020-05-03 10:40:11 +00:00
self.loc.get('notification_order_placed',
order=order.text(w=self)),
2018-04-01 16:10:14 +00:00
reply_markup=order_keyboard)
2018-02-19 12:47:43 +00:00
2017-12-22 08:56:03 +00:00
def __order_status(self):
2018-03-19 11:24:05 +00:00
"""Display the status of the sent orders."""
log.debug("Displaying __order_status")
2018-04-05 07:30:32 +00:00
# Find the latest orders
orders = self.session.query(db.Order) \
.filter(db.Order.user == self.user) \
.order_by(db.Order.creation_date.desc()) \
.limit(20) \
2018-04-05 08:43:27 +00:00
.all()
2018-04-05 07:30:32 +00:00
# Ensure there is at least one order to display
if len(orders) == 0:
self.bot.send_message(self.chat.id, self.loc.get("error_no_orders"))
2018-04-05 07:30:32 +00:00
# Display the order status to the user
for order in orders:
self.bot.send_message(self.chat.id, order.text(w=self, user=True))
2018-04-05 07:30:32 +00:00
# TODO: maybe add a page displayer instead of showing the latest 5 orders
2017-12-22 08:56:03 +00:00
def __add_credit_menu(self):
2017-12-26 17:15:30 +00:00
"""Add more credit to the account."""
log.debug("Displaying __add_credit_menu")
2017-12-26 17:15:30 +00:00
# Create a payment methods keyboard
keyboard = list()
# Add the supported payment methods to the keyboard
# Cash
if self.cfg["Payments"]["Cash"]["enable_pay_with_cash"]:
keyboard.append([telegram.KeyboardButton(self.loc.get("menu_cash"))])
2017-12-26 17:15:30 +00:00
# Telegram Payments
if self.cfg["Payments"]["CreditCard"]["credit_card_token"] != "":
keyboard.append([telegram.KeyboardButton(self.loc.get("menu_credit_card"))])
2017-12-26 17:15:30 +00:00
# Keyboard: go back to the previous menu
keyboard.append([telegram.KeyboardButton(self.loc.get("menu_cancel"))])
2017-12-26 17:15:30 +00:00
# Send the keyboard to the user
self.bot.send_message(self.chat.id, self.loc.get("conversation_payment_method"),
2017-12-26 17:15:30 +00:00
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
# Wait for a reply from the user
2020-09-07 15:13:27 +00:00
selection = self.__wait_for_specific_message(
[self.loc.get("menu_cash"), self.loc.get("menu_credit_card"), self.loc.get("menu_cancel")],
cancellable=True)
2017-12-26 17:15:30 +00:00
# If the user has selected the Cash option...
if selection == self.loc.get("menu_cash") and self.cfg["Payments"]["Cash"]["enable_pay_with_cash"]:
2017-12-26 17:15:30 +00:00
# Go to the pay with cash function
self.bot.send_message(self.chat.id,
self.loc.get("payment_cash", user_cash_id=self.user.identifiable_str()))
2017-12-26 17:15:30 +00:00
# If the user has selected the Credit Card option...
elif selection == self.loc.get("menu_credit_card") and self.cfg["Payments"]["CreditCard"]["credit_card_token"]:
2017-12-26 17:15:30 +00:00
# Go to the pay with credit card function
self.__add_credit_cc()
# If the user has selected the Cancel option...
2020-04-03 23:05:59 +00:00
elif isinstance(selection, CancelSignal):
2017-12-26 17:15:30 +00:00
# Send him back to the previous menu
return
def __add_credit_cc(self):
2018-01-29 12:26:49 +00:00
"""Add money to the wallet through a credit card payment."""
log.debug("Displaying __add_credit_cc")
# Create a keyboard to be sent later
presets = self.cfg["Payments"]["CreditCard"]["payment_presets"]
keyboard = [[telegram.KeyboardButton(str(self.Price(preset)))] for preset in presets]
keyboard.append([telegram.KeyboardButton(self.loc.get("menu_cancel"))])
2018-02-08 09:18:53 +00:00
# Boolean variable to check if the user has cancelled the action
cancelled = False
2018-01-03 13:52:05 +00:00
# Loop used to continue asking if there's an error during the input
2018-02-08 09:18:53 +00:00
while not cancelled:
2018-01-03 13:52:05 +00:00
# Send the message and the keyboard
self.bot.send_message(self.chat.id, self.loc.get("payment_cc_amount"),
2018-01-03 13:52:05 +00:00
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
# Wait until a valid amount is sent
2020-09-07 15:13:27 +00:00
selection = self.__wait_for_regex(r"([0-9]+(?:[.,][0-9]+)?|" + self.loc.get("menu_cancel") + r")",
cancellable=True)
2018-02-08 09:18:53 +00:00
# If the user cancelled the action
2020-04-03 23:05:59 +00:00
if isinstance(selection, CancelSignal):
2018-02-08 09:18:53 +00:00
# Exit the loop
cancelled = True
continue
# Convert the amount to an integer
value = self.Price(selection)
2018-01-03 13:52:05 +00:00
# Ensure the amount is within the range
if value > self.Price(self.cfg["Payments"]["CreditCard"]["max_amount"]):
2018-04-05 08:43:27 +00:00
self.bot.send_message(self.chat.id,
2020-09-07 15:13:27 +00:00
self.loc.get("error_payment_amount_over_max",
2021-06-01 14:19:22 +00:00
max_amount=self.Price(self.cfg["CreditCard"]["max_amount"])))
2018-01-03 13:52:05 +00:00
continue
elif value < self.Price(self.cfg["Payments"]["CreditCard"]["min_amount"]):
2018-04-05 08:43:27 +00:00
self.bot.send_message(self.chat.id,
2020-09-07 15:13:27 +00:00
self.loc.get("error_payment_amount_under_min",
2021-06-01 14:19:22 +00:00
min_amount=self.Price(self.cfg["CreditCard"]["min_amount"])))
2018-01-03 13:52:05 +00:00
continue
2018-01-03 14:23:08 +00:00
break
2018-02-08 09:18:53 +00:00
# If the user cancelled the action...
else:
# Exit the function
return
2020-04-04 17:53:10 +00:00
# Issue the payment invoice
2020-04-04 17:18:35 +00:00
self.__make_payment(amount=value)
def __make_payment(self, amount):
2018-01-10 10:29:02 +00:00
# Set the invoice active invoice payload
self.invoice_payload = str(uuid.uuid4())
2018-01-15 09:16:04 +00:00
# Create the price array
prices = [telegram.LabeledPrice(label=self.loc.get("payment_invoice_label"), amount=int(amount))]
2018-01-15 09:16:04 +00:00
# If the user has to pay a fee when using the credit card, add it to the prices list
2020-04-06 23:44:23 +00:00
fee = int(self.__get_total_fee(amount))
if fee > 0:
prices.append(telegram.LabeledPrice(label=self.loc.get("payment_invoice_fee_label"),
2020-04-06 23:44:23 +00:00
amount=fee))
2018-02-08 09:18:53 +00:00
# Create the invoice keyboard
inline_keyboard = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_pay"),
pay=True)],
[telegram.InlineKeyboardButton(self.loc.get("menu_cancel"),
2018-04-05 08:43:27 +00:00
callback_data="cmd_cancel")]])
# The amount is valid, send the invoice
self.bot.send_invoice(self.chat.id,
title=self.loc.get("payment_invoice_title"),
description=self.loc.get("payment_invoice_description", amount=str(amount)),
2018-01-10 10:29:02 +00:00
payload=self.invoice_payload,
provider_token=self.cfg["Payments"]["CreditCard"]["credit_card_token"],
2018-04-18 07:23:44 +00:00
start_parameter="tempdeeplink",
currency=self.cfg["Payments"]["currency"],
2018-01-15 09:16:04 +00:00
prices=prices,
need_name=self.cfg["Payments"]["CreditCard"]["name_required"],
need_email=self.cfg["Payments"]["CreditCard"]["email_required"],
need_phone_number=self.cfg["Payments"]["CreditCard"]["phone_required"],
reply_markup=inline_keyboard,
max_tip_amount=self.cfg["Payments"]["CreditCard"]["max_tip_amount"],
suggested_tip_amounts=self.cfg["Payments"]["CreditCard"]["tip_presets"],
)
2020-04-06 23:44:23 +00:00
# Wait for the precheckout query
2018-02-08 09:18:53 +00:00
precheckoutquery = self.__wait_for_precheckoutquery(cancellable=True)
# Check if the user has cancelled the invoice
if isinstance(precheckoutquery, CancelSignal):
# Exit the function
return
2018-01-10 10:29:02 +00:00
# Accept the checkout
self.bot.answer_pre_checkout_query(precheckoutquery.id, ok=True)
# Wait for the payment
2020-07-06 22:07:20 +00:00
successfulpayment = self.__wait_for_successfulpayment(cancellable=False)
2018-01-10 10:29:02 +00:00
# Create a new database transaction
transaction = db.Transaction(user=self.user,
value=int(amount),
2018-01-10 10:29:02 +00:00
provider="Credit Card",
telegram_charge_id=successfulpayment.telegram_payment_charge_id,
provider_charge_id=successfulpayment.provider_payment_charge_id)
2020-04-04 17:18:35 +00:00
2018-01-10 10:29:02 +00:00
if successfulpayment.order_info is not None:
transaction.payment_name = successfulpayment.order_info.name
transaction.payment_email = successfulpayment.order_info.email
transaction.payment_phone = successfulpayment.order_info.phone_number
# Update the user's credit
self.user.recalculate_credit()
# Commit all the changes
2018-01-10 10:29:02 +00:00
self.session.commit()
2017-12-21 09:42:23 +00:00
def __get_total_fee(self, amount):
2020-04-06 23:44:23 +00:00
# Calculate a fee for the required amount
fee_percentage = self.cfg["Payments"]["CreditCard"]["fee_percentage"] / 100
fee_fixed = self.cfg["Payments"]["CreditCard"]["fee_fixed"]
2020-04-04 17:18:35 +00:00
total_fee = amount * fee_percentage + fee_fixed
if total_fee > 0:
return total_fee
# Set the fee to 0 to ensure no accidental discounts are applied
return 0
2017-12-22 08:56:03 +00:00
def __bot_info(self):
"""Send information about the bot."""
log.debug("Displaying __bot_info")
self.bot.send_message(self.chat.id, self.loc.get("bot_info"))
2017-12-21 09:42:23 +00:00
def __admin_menu(self):
"""Function called from the run method when the user is an administrator.
Administrative bot actions should be placed here."""
log.debug("Displaying __admin_menu")
2018-01-29 12:26:49 +00:00
# Loop used to return to the menu after executing a command
while True:
2018-03-22 09:33:50 +00:00
# Create a keyboard with the admin main menu based on the admin permissions specified in the db
keyboard = []
2018-03-22 09:33:50 +00:00
if self.admin.edit_products:
keyboard.append([self.loc.get("menu_products")])
2018-03-22 09:33:50 +00:00
if self.admin.receive_orders:
keyboard.append([self.loc.get("menu_orders")])
if self.admin.create_transactions:
if self.cfg["Payments"]["Cash"]["enable_create_transaction"]:
keyboard.append([self.loc.get("menu_edit_credit")])
keyboard.append([self.loc.get("menu_transactions"), self.loc.get("menu_csv")])
2018-05-15 15:00:24 +00:00
if self.admin.is_owner:
keyboard.append([self.loc.get("menu_edit_admins")])
keyboard.append([self.loc.get("menu_user_mode")])
2018-01-29 12:26:49 +00:00
# Send the previously created keyboard to the user (ensuring it can be clicked only 1 time)
self.bot.send_message(self.chat.id, self.loc.get("conversation_open_admin_menu"),
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
2018-01-29 12:26:49 +00:00
# Wait for a reply from the user
selection = self.__wait_for_specific_message([self.loc.get("menu_products"),
self.loc.get("menu_orders"),
self.loc.get("menu_user_mode"),
self.loc.get("menu_edit_credit"),
self.loc.get("menu_transactions"),
self.loc.get("menu_csv"),
self.loc.get("menu_edit_admins")])
# If the user has selected the Products option and has the privileges to perform the action...
if selection == self.loc.get("menu_products") and self.admin.edit_products:
2018-01-29 12:26:49 +00:00
# Open the products menu
self.__products_menu()
# If the user has selected the Orders option and has the privileges to perform the action...
elif selection == self.loc.get("menu_orders") and self.admin.receive_orders:
2018-01-29 12:26:49 +00:00
# Open the orders menu
self.__orders_menu()
# If the user has selected the Transactions option and has the privileges to perform the action...
elif selection == self.loc.get("menu_edit_credit") and self.admin.create_transactions:
# Open the edit credit menu
self.__create_transaction()
# If the user has selected the User mode option and has the privileges to perform the action...
elif selection == self.loc.get("menu_user_mode"):
# Tell the user how to go back to admin menu
self.bot.send_message(self.chat.id, self.loc.get("conversation_switch_to_user_mode"))
2018-01-29 12:26:49 +00:00
# Start the bot in user mode
self.__user_menu()
# If the user has selected the Add Admin option and has the privileges to perform the action...
elif selection == self.loc.get("menu_edit_admins") and self.admin.is_owner:
2018-05-15 15:00:24 +00:00
# Open the edit admin menu
self.__add_admin()
# If the user has selected the Transactions option and has the privileges to perform the action...
elif selection == self.loc.get("menu_transactions") and self.admin.create_transactions:
2018-04-18 19:28:39 +00:00
# Open the transaction pages
self.__transaction_pages()
# If the user has selected the .csv option and has the privileges to perform the action...
elif selection == self.loc.get("menu_csv") and self.admin.create_transactions:
2018-04-19 07:37:42 +00:00
# Generate the .csv file
self.__transactions_file()
2018-01-29 12:26:49 +00:00
def __products_menu(self):
"""Display the admin menu to select a product to edit."""
log.debug("Displaying __products_menu")
2018-01-29 12:26:49 +00:00
# Get the products list from the db
2018-03-06 16:39:02 +00:00
products = self.session.query(db.Product).filter_by(deleted=False).all()
2018-01-29 12:26:49 +00:00
# Create a list of product names
product_names = [product.name for product in products]
2018-03-06 16:39:02 +00:00
# Insert at the start of the list the add product option, the remove product option and the Cancel option
product_names.insert(0, self.loc.get("menu_cancel"))
product_names.insert(1, self.loc.get("menu_add_product"))
product_names.insert(2, self.loc.get("menu_delete_product"))
2018-01-29 12:26:49 +00:00
# Create a keyboard using the product names
keyboard = [[telegram.KeyboardButton(product_name)] for product_name in product_names]
# Send the previously created keyboard to the user (ensuring it can be clicked only 1 time)
self.bot.send_message(self.chat.id, self.loc.get("conversation_admin_select_product"),
2018-01-29 12:26:49 +00:00
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
# Wait for a reply from the user
2020-04-23 15:47:19 +00:00
selection = self.__wait_for_specific_message(product_names, cancellable=True)
2018-02-08 09:18:53 +00:00
# If the user has selected the Cancel option...
2020-04-03 23:05:59 +00:00
if isinstance(selection, CancelSignal):
2018-02-08 09:18:53 +00:00
# Exit the menu
return
2018-01-29 12:26:49 +00:00
# If the user has selected the Add Product option...
elif selection == self.loc.get("menu_add_product"):
2018-01-29 12:26:49 +00:00
# Open the add product menu
2018-02-01 09:06:40 +00:00
self.__edit_product_menu()
2018-03-06 16:39:02 +00:00
# If the user has selected the Remove Product option...
elif selection == self.loc.get("menu_delete_product"):
2018-03-06 16:39:02 +00:00
# Open the delete product menu
self.__delete_product_menu()
2018-01-29 12:26:49 +00:00
# If the user has selected a product
else:
2018-02-01 09:06:40 +00:00
# Find the selected product
2018-03-06 16:39:02 +00:00
product = self.session.query(db.Product).filter_by(name=selection, deleted=False).one()
2018-01-29 12:26:49 +00:00
# Open the edit menu for that specific product
2018-02-01 09:06:40 +00:00
self.__edit_product_menu(product=product)
2018-01-29 12:26:49 +00:00
def __edit_product_menu(self, product: Optional[db.Product] = None):
2018-02-01 09:06:40 +00:00
"""Add a product to the database or edit an existing one."""
log.debug("Displaying __edit_product_menu")
2018-02-08 09:18:53 +00:00
# Create an inline keyboard with a single skip button
cancel = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_skip"),
2018-04-05 08:43:27 +00:00
callback_data="cmd_cancel")]])
2018-01-29 12:26:49 +00:00
# Ask for the product name until a valid product name is specified
while True:
# Ask the question to the user
self.bot.send_message(self.chat.id, self.loc.get("ask_product_name"))
2018-02-01 09:06:40 +00:00
# Display the current name if you're editing an existing product
if product:
self.bot.send_message(self.chat.id, self.loc.get("edit_current_value", value=escape(product.name)),
2018-04-05 08:43:27 +00:00
reply_markup=cancel)
2018-01-29 12:26:49 +00:00
# Wait for an answer
2018-02-08 09:18:53 +00:00
name = self.__wait_for_regex(r"(.*)", cancellable=bool(product))
2018-01-29 12:26:49 +00:00
# Ensure a product with that name doesn't already exist
if (product and isinstance(name, CancelSignal)) or \
2018-04-05 08:43:27 +00:00
self.session.query(db.Product).filter_by(name=name, deleted=False).one_or_none() in [None, product]:
2018-01-29 12:26:49 +00:00
# Exit the loop
break
self.bot.send_message(self.chat.id, self.loc.get("error_duplicate_name"))
2018-01-29 12:26:49 +00:00
# Ask for the product description
self.bot.send_message(self.chat.id, self.loc.get("ask_product_description"))
2018-02-01 09:06:40 +00:00
# Display the current description if you're editing an existing product
if product:
2018-04-09 10:24:55 +00:00
self.bot.send_message(self.chat.id,
self.loc.get("edit_current_value", value=escape(product.description)),
2018-04-05 08:43:27 +00:00
reply_markup=cancel)
2018-01-29 12:26:49 +00:00
# Wait for an answer
2018-02-08 09:18:53 +00:00
description = self.__wait_for_regex(r"(.*)", cancellable=bool(product))
2018-01-29 12:26:49 +00:00
# Ask for the product price
2018-04-09 10:24:55 +00:00
self.bot.send_message(self.chat.id,
self.loc.get("ask_product_price"))
2018-02-01 09:06:40 +00:00
# Display the current name if you're editing an existing product
if product:
if product.price is not None:
value_text = str(self.Price(product.price))
else:
value_text = self.loc.get("text_not_for_sale")
self.bot.send_message(
self.chat.id,
self.loc.get("edit_current_value", value=value_text),
reply_markup=cancel
)
2018-02-01 09:06:40 +00:00
# Wait for an answer
price = self.__wait_for_regex(r"([0-9]+(?:[.,][0-9]{1,2})?|[Xx])",
2018-04-05 08:43:27 +00:00
cancellable=True)
2018-02-01 09:06:40 +00:00
# If the price is skipped
2018-02-08 09:18:53 +00:00
if isinstance(price, CancelSignal):
pass
elif price.lower() == "x":
2018-02-01 09:06:40 +00:00
price = None
else:
price = self.Price(price)
2021-03-10 15:51:21 +00:00
if not isinstance(price, CancelSignal) and price is not None:
price = int(price)
2018-02-01 09:06:40 +00:00
# Ask for the product image
self.bot.send_message(self.chat.id, self.loc.get("ask_product_image"), reply_markup=cancel)
2018-01-29 12:26:49 +00:00
# Wait for an answer
2018-02-08 09:18:53 +00:00
photo_list = self.__wait_for_photo(cancellable=True)
2018-02-01 09:06:40 +00:00
# If a new product is being added...
if not product:
# Create the db record for the product
2018-04-09 10:24:55 +00:00
# noinspection PyTypeChecker
2018-02-01 09:06:40 +00:00
product = db.Product(name=name,
description=description,
2021-03-10 15:51:21 +00:00
price=price,
2018-03-06 16:39:02 +00:00
deleted=False)
2018-02-01 09:06:40 +00:00
# Add the record to the database
self.session.add(product)
# If a product is being edited...
else:
# Edit the record with the new values
2018-02-08 09:18:53 +00:00
product.name = name if not isinstance(name, CancelSignal) else product.name
product.description = description if not isinstance(description, CancelSignal) else product.description
2021-03-10 15:51:21 +00:00
product.price = price if not isinstance(price, CancelSignal) else product.price
2018-02-08 09:18:53 +00:00
# If a photo has been sent...
2018-06-10 14:36:13 +00:00
if isinstance(photo_list, list):
2018-02-08 09:18:53 +00:00
# Find the largest photo id
largest_photo = photo_list[0]
for photo in photo_list[1:]:
if photo.width > largest_photo.width:
largest_photo = photo
# Get the file object associated with the photo
photo_file = self.bot.get_file(largest_photo.file_id)
# Notify the user that the bot is downloading the image and might be inactive for a while
self.bot.send_message(self.chat.id, self.loc.get("downloading_image"))
2018-02-08 09:18:53 +00:00
self.bot.send_chat_action(self.chat.id, action="upload_photo")
# Set the image for that product
product.set_image(photo_file)
# Commit the session changes
2018-01-29 12:26:49 +00:00
self.session.commit()
# Notify the user
self.bot.send_message(self.chat.id, self.loc.get("success_product_edited"))
2018-01-29 12:26:49 +00:00
2018-03-06 16:39:02 +00:00
def __delete_product_menu(self):
log.debug("Displaying __delete_product_menu")
2018-03-06 16:39:02 +00:00
# Get the products list from the db
products = self.session.query(db.Product).filter_by(deleted=False).all()
# Create a list of product names
product_names = [product.name for product in products]
# Insert at the start of the list the Cancel button
product_names.insert(0, self.loc.get("menu_cancel"))
2018-03-06 16:39:02 +00:00
# Create a keyboard using the product names
keyboard = [[telegram.KeyboardButton(product_name)] for product_name in product_names]
# Send the previously created keyboard to the user (ensuring it can be clicked only 1 time)
self.bot.send_message(self.chat.id, self.loc.get("conversation_admin_select_product_to_delete"),
2018-03-06 16:39:02 +00:00
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
# Wait for a reply from the user
2020-04-23 15:47:19 +00:00
selection = self.__wait_for_specific_message(product_names, cancellable=True)
2020-04-03 23:05:59 +00:00
if isinstance(selection, CancelSignal):
2018-03-06 16:39:02 +00:00
# Exit the menu
return
else:
# Find the selected product
product = self.session.query(db.Product).filter_by(name=selection, deleted=False).one()
# "Delete" the product by setting the deleted flag to true
product.deleted = True
self.session.commit()
# Notify the user
self.bot.send_message(self.chat.id, self.loc.get("success_product_deleted"))
2018-03-06 16:39:02 +00:00
2018-01-29 12:26:49 +00:00
def __orders_menu(self):
2018-03-19 11:24:05 +00:00
"""Display a live flow of orders."""
log.debug("Displaying __orders_menu")
# Create a cancel and a stop keyboard
stop_keyboard = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_stop"),
2018-04-05 08:43:27 +00:00
callback_data="cmd_cancel")]])
cancel_keyboard = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_cancel"),
2018-04-05 08:43:27 +00:00
callback_data="cmd_cancel")]])
2018-03-19 11:24:05 +00:00
# Send a small intro message on the Live Orders mode
# Remove the keyboard with the first message... (#39)
self.bot.send_message(self.chat.id,
self.loc.get("conversation_live_orders_start"),
reply_markup=telegram.ReplyKeyboardRemove())
# ...and display a small inline keyboard with the following one
self.bot.send_message(self.chat.id,
self.loc.get("conversation_live_orders_stop"),
reply_markup=stop_keyboard)
2018-03-19 11:24:05 +00:00
# Create the order keyboard
order_keyboard = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_complete"),
2018-04-05 08:43:27 +00:00
callback_data="order_complete")],
[telegram.InlineKeyboardButton(self.loc.get("menu_refund"),
2018-04-05 08:43:27 +00:00
callback_data="order_refund")]])
2018-03-19 11:24:05 +00:00
# Display the past pending orders
orders = self.session.query(db.Order) \
.filter_by(delivery_date=None, refund_date=None) \
.join(db.Transaction) \
.join(db.User) \
2018-04-05 08:43:27 +00:00
.all()
2018-03-19 11:24:05 +00:00
# Create a message for every one of them
for order in orders:
# Send the created message
self.bot.send_message(self.chat.id, order.text(w=self),
2018-04-05 08:43:27 +00:00
reply_markup=order_keyboard)
2018-04-01 16:10:14 +00:00
# Set the Live mode flag to True
self.admin.live_mode = True
# Commit the change to the database
self.session.commit()
while True:
# Wait for any message to stop the listening mode
update = self.__wait_for_inlinekeyboard_callback(cancellable=True)
# If the user pressed the stop button, exit listening mode
if isinstance(update, CancelSignal):
# Stop the listening mode
2018-04-01 16:10:14 +00:00
self.admin.live_mode = False
break
# Find the order
order_id = re.search(self.loc.get("order_number").replace("{id}", "([0-9]+)"), update.message.text).group(1)
2021-05-06 22:43:22 +00:00
order = self.session.query(db.Order).get(order_id)
2018-04-01 16:10:14 +00:00
# Check if the order hasn't been already cleared
if order.delivery_date is not None or order.refund_date is not None:
# Notify the admin and skip that order
self.bot.edit_message_text(self.chat.id, self.loc.get("error_order_already_cleared"))
2018-03-22 09:33:50 +00:00
break
# If the user pressed the complete order button, complete the order
2018-04-01 16:10:14 +00:00
if update.data == "order_complete":
# Mark the order as complete
2018-04-01 16:10:14 +00:00
order.delivery_date = datetime.datetime.now()
# Commit the transaction
self.session.commit()
2018-04-01 16:10:14 +00:00
# Update order message
self.bot.edit_message_text(order.text(w=self), chat_id=self.chat.id,
2018-04-01 16:10:14 +00:00
message_id=update.message.message_id)
# Notify the user of the completition
2018-04-05 08:43:27 +00:00
self.bot.send_message(order.user_id,
2020-09-07 15:13:27 +00:00
self.loc.get("notification_order_completed",
order=order.text(w=self, user=True)))
# If the user pressed the refund order button, refund the order...
elif update.data == "order_refund":
# Ask for a refund reason
reason_msg = self.bot.send_message(self.chat.id, self.loc.get("ask_refund_reason"),
2018-04-19 07:55:36 +00:00
reply_markup=cancel_keyboard)
# Wait for a reply
reply = self.__wait_for_regex("(.*)", cancellable=True)
# If the user pressed the cancel button, cancel the refund
if isinstance(reply, CancelSignal):
2018-04-16 10:28:00 +00:00
# Delete the message asking for the refund reason
self.bot.delete_message(self.chat.id, reason_msg.message_id)
continue
# Mark the order as refunded
order.refund_date = datetime.datetime.now()
# Save the refund reason
order.refund_reason = reply
# Refund the credit, reverting the old transaction
order.transaction.refunded = True
# Update the user's credit
order.user.recalculate_credit()
# Commit the changes
self.session.commit()
2018-04-01 16:10:14 +00:00
# Update the order message
self.bot.edit_message_text(order.text(w=self),
2018-04-05 08:43:27 +00:00
chat_id=self.chat.id,
message_id=update.message.message_id)
# Notify the user of the refund
2018-04-05 08:43:27 +00:00
self.bot.send_message(order.user_id,
self.loc.get("notification_order_refunded", order=order.text(w=self,
2020-05-03 10:40:11 +00:00
user=True)))
2018-04-16 10:28:00 +00:00
# Notify the admin of the refund
self.bot.send_message(self.chat.id, self.loc.get("success_order_refunded", order_id=order.order_id))
2017-12-21 09:42:23 +00:00
def __create_transaction(self):
"""Edit manually the credit of an user."""
log.debug("Displaying __create_transaction")
2018-04-30 14:14:02 +00:00
# Make the admin select an user
user = self.__user_select()
# Allow the cancellation of the operation
if isinstance(user, CancelSignal):
return
# Create an inline keyboard with a single cancel button
cancel = telegram.InlineKeyboardMarkup([[telegram.InlineKeyboardButton(self.loc.get("menu_cancel"),
callback_data="cmd_cancel")]])
# Request from the user the amount of money to be credited manually
self.bot.send_message(self.chat.id, self.loc.get("ask_credit"), reply_markup=cancel)
# Wait for an answer
2021-05-01 00:03:38 +00:00
reply = self.__wait_for_regex(r"(-? ?[0-9]+(?:[.,][0-9]{1,2})?)", cancellable=True)
# Allow the cancellation of the operation
if isinstance(reply, CancelSignal):
return
# Convert the reply to a price object
price = self.Price(reply)
# Ask the user for notes
self.bot.send_message(self.chat.id, self.loc.get("ask_transaction_notes"), reply_markup=cancel)
# Wait for an answer
reply = self.__wait_for_regex(r"(.*)", cancellable=True)
# Allow the cancellation of the operation
if isinstance(reply, CancelSignal):
return
# Create a new transaction
transaction = db.Transaction(user=user,
value=int(price),
provider="Manual",
notes=reply)
self.session.add(transaction)
# Change the user credit
user.recalculate_credit()
# Commit the changes
self.session.commit()
# Notify the user of the credit/debit
self.bot.send_message(user.user_id,
self.loc.get("notification_transaction_created",
transaction=transaction.text(w=self)))
# Notify the admin of the success
self.bot.send_message(self.chat.id, self.loc.get("success_transaction_created",
transaction=transaction.text(w=self)))
2018-04-16 10:09:51 +00:00
def __help_menu(self):
"""Help menu. Allows the user to ask for assistance, get a guide or see some info about the bot."""
log.debug("Displaying __help_menu")
2018-04-16 10:09:51 +00:00
# Create a keyboard with the user help menu
keyboard = [[telegram.KeyboardButton(self.loc.get("menu_guide"))],
[telegram.KeyboardButton(self.loc.get("menu_contact_shopkeeper"))],
[telegram.KeyboardButton(self.loc.get("menu_cancel"))]]
2018-04-16 10:09:51 +00:00
# Send the previously created keyboard to the user (ensuring it can be clicked only 1 time)
self.bot.send_message(self.chat.id,
self.loc.get("conversation_open_help_menu"),
2018-04-19 07:55:36 +00:00
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
2018-04-16 10:09:51 +00:00
# Wait for a reply from the user
2020-06-18 15:17:44 +00:00
selection = self.__wait_for_specific_message([
self.loc.get("menu_guide"),
self.loc.get("menu_contact_shopkeeper")
], cancellable=True)
2018-04-16 10:09:51 +00:00
# If the user has selected the Guide option...
if selection == self.loc.get("menu_guide"):
2018-04-16 10:09:51 +00:00
# Send them the bot guide
self.bot.send_message(self.chat.id, self.loc.get("help_msg"))
2018-04-16 10:09:51 +00:00
# If the user has selected the Order Status option...
elif selection == self.loc.get("menu_contact_shopkeeper"):
2018-04-16 10:09:51 +00:00
# Find the list of available shopkeepers
shopkeepers = self.session.query(db.Admin).filter_by(display_on_help=True).join(db.User).all()
# Create the string
shopkeepers_string = "\n".join([admin.user.mention() for admin in shopkeepers])
# Send the message to the user
self.bot.send_message(self.chat.id, self.loc.get("contact_shopkeeper", shopkeepers=shopkeepers_string))
2018-04-16 10:18:42 +00:00
# If the user has selected the Cancel option the function will return immediately
2018-04-16 10:09:51 +00:00
2018-04-18 19:28:39 +00:00
def __transaction_pages(self):
"""Display the latest transactions, in pages."""
log.debug("Displaying __transaction_pages")
2018-04-18 19:28:39 +00:00
# Page number
page = 0
# Create and send a placeholder message to be populated
message = self.bot.send_message(self.chat.id, self.loc.get("loading_transactions"))
2018-04-18 19:28:39 +00:00
# Loop used to move between pages
while True:
# Retrieve the 10 transactions in that page
transactions = self.session.query(db.Transaction) \
.order_by(db.Transaction.transaction_id.desc()) \
.limit(10) \
.offset(10 * page) \
2018-04-18 19:28:39 +00:00
.all()
# Create a list to be converted in inline keyboard markup
inline_keyboard_list = [[]]
# Don't add a previous page button if this is the first page
if page != 0:
# Add a previous page button
inline_keyboard_list[0].append(
telegram.InlineKeyboardButton(self.loc.get("menu_previous"), callback_data="cmd_previous")
2018-04-18 19:28:39 +00:00
)
# Don't add a next page button if this is the last page
if len(transactions) == 10:
# Add a next page button
inline_keyboard_list[0].append(
telegram.InlineKeyboardButton(self.loc.get("menu_next"), callback_data="cmd_next")
2018-04-18 19:28:39 +00:00
)
# Add a Done button
2020-09-07 15:13:27 +00:00
inline_keyboard_list.append(
[telegram.InlineKeyboardButton(self.loc.get("menu_done"), callback_data="cmd_done")])
2018-04-18 19:28:39 +00:00
# Create the inline keyboard markup
inline_keyboard = telegram.InlineKeyboardMarkup(inline_keyboard_list)
# Create the message text
transactions_string = "\n".join([transaction.text(w=self) for transaction in transactions])
text = self.loc.get("transactions_page", page=page + 1, transactions=transactions_string)
2018-04-18 19:28:39 +00:00
# Update the previously sent message
self.bot.edit_message_text(chat_id=self.chat.id, message_id=message.message_id, text=text,
2018-04-19 07:55:36 +00:00
reply_markup=inline_keyboard)
2018-04-18 19:28:39 +00:00
# Wait for user input
selection = self.__wait_for_inlinekeyboard_callback()
# If Previous was selected...
if selection.data == "cmd_previous" and page != 0:
# Go back one page
page -= 1
# If Next was selected...
elif selection.data == "cmd_next" and len(transactions) == 10:
# Go to the next page
page += 1
# If Done was selected...
elif selection.data == "cmd_done":
# Break the loop
break
2018-04-19 07:37:42 +00:00
def __transactions_file(self):
"""Generate a .csv file containing the list of all transactions."""
log.debug("Generating __transaction_file")
2018-04-19 07:37:42 +00:00
# Retrieve all the transactions
2021-05-06 22:43:22 +00:00
transactions = self.session.query(db.Transaction).order_by(db.Transaction.transaction_id).all()
2018-04-19 07:37:42 +00:00
# Write on the previously created file
with open(f"transactions_{self.chat.id}.csv", "w") as file:
# Write an header line
2018-04-19 07:39:11 +00:00
file.write(f"UserID;"
f"TransactionValue;"
f"TransactionNotes;"
f"Provider;"
f"ChargeID;"
f"SpecifiedName;"
f"SpecifiedPhone;"
f"SpecifiedEmail;"
2018-04-19 07:37:42 +00:00
f"Refunded?\n")
2018-04-19 07:39:11 +00:00
# For each transaction; write a new line on file
2018-04-19 07:37:42 +00:00
for transaction in transactions:
2018-04-19 07:39:11 +00:00
file.write(f"{transaction.user_id if transaction.user_id is not None else ''};"
f"{transaction.value if transaction.value is not None else ''};"
f"{transaction.notes if transaction.notes is not None else ''};"
f"{transaction.provider if transaction.provider is not None else ''};"
f"{transaction.provider_charge_id if transaction.provider_charge_id is not None else ''};"
f"{transaction.payment_name if transaction.payment_name is not None else ''};"
f"{transaction.payment_phone if transaction.payment_phone is not None else ''};"
f"{transaction.payment_email if transaction.payment_email is not None else ''};"
2018-04-19 07:37:42 +00:00
f"{transaction.refunded if transaction.refunded is not None else ''}\n")
# Describe the file to the user
self.bot.send_message(self.chat.id, self.loc.get("csv_caption"))
2018-04-19 07:37:42 +00:00
# Reopen the file for reading
with open(f"transactions_{self.chat.id}.csv") as file:
# Send the file via a manual request to Telegram
requests.post(f"https://api.telegram.org/bot{self.cfg['Telegram']['token']}/sendDocument",
2018-04-19 07:37:42 +00:00
files={"document": file},
params={"chat_id": self.chat.id,
"parse_mode": "HTML"})
# Delete the created file
os.remove(f"transactions_{self.chat.id}.csv")
2018-05-04 13:46:07 +00:00
def __add_admin(self):
"""Add an administrator to the bot."""
log.debug("Displaying __add_admin")
2018-05-04 13:46:07 +00:00
# Let the admin select an administrator to promote
user = self.__user_select()
# Allow the cancellation of the operation
if isinstance(user, CancelSignal):
return
2018-05-04 13:46:07 +00:00
# Check if the user is already an administrator
2021-05-06 22:43:22 +00:00
admin = self.session.query(db.Admin).filter_by(user=user).one_or_none()
2018-05-04 13:46:07 +00:00
if admin is None:
# Create the keyboard to be sent
keyboard = telegram.ReplyKeyboardMarkup([[self.loc.get("emoji_yes"), self.loc.get("emoji_no")]],
one_time_keyboard=True)
2018-05-04 13:46:07 +00:00
# Ask for confirmation
self.bot.send_message(self.chat.id, self.loc.get("conversation_confirm_admin_promotion"),
reply_markup=keyboard)
2018-05-04 13:46:07 +00:00
# Wait for an answer
selection = self.__wait_for_specific_message([self.loc.get("emoji_yes"), self.loc.get("emoji_no")])
2018-05-04 13:46:07 +00:00
# Proceed only if the answer is yes
if selection == self.loc.get("emoji_no"):
2018-05-04 13:46:07 +00:00
return
# Create a new admin
admin = db.Admin(user=user,
edit_products=False,
receive_orders=False,
create_transactions=False,
is_owner=False,
display_on_help=False)
2018-05-15 15:31:30 +00:00
self.session.add(admin)
2018-05-04 13:46:07 +00:00
# Send the empty admin message and record the id
message = self.bot.send_message(self.chat.id, self.loc.get("admin_properties", name=str(admin.user)))
2018-05-04 13:46:07 +00:00
# Start accepting edits
while True:
# Create the inline keyboard with the admin status
inline_keyboard = telegram.InlineKeyboardMarkup([
[telegram.InlineKeyboardButton(
f"{self.loc.boolmoji(admin.edit_products)} {self.loc.get('prop_edit_products')}",
callback_data="toggle_edit_products"
)],
[telegram.InlineKeyboardButton(
f"{self.loc.boolmoji(admin.receive_orders)} {self.loc.get('prop_receive_orders')}",
callback_data="toggle_receive_orders"
)],
2018-05-04 13:46:07 +00:00
[telegram.InlineKeyboardButton(
2020-05-03 10:40:11 +00:00
f"{self.loc.boolmoji(admin.create_transactions)} {self.loc.get('prop_create_transactions')}",
callback_data="toggle_create_transactions"
)],
2018-05-04 13:46:07 +00:00
[telegram.InlineKeyboardButton(
2020-05-03 10:40:11 +00:00
f"{self.loc.boolmoji(admin.display_on_help)} {self.loc.get('prop_display_on_help')}",
callback_data="toggle_display_on_help"
)],
[telegram.InlineKeyboardButton(
self.loc.get('menu_done'),
callback_data="cmd_done"
)]
2018-05-04 13:46:07 +00:00
])
# Update the inline keyboard
2018-05-15 15:00:24 +00:00
self.bot.edit_message_reply_markup(message_id=message.message_id,
chat_id=self.chat.id,
reply_markup=inline_keyboard)
2018-05-04 13:46:07 +00:00
# Wait for an user answer
callback = self.__wait_for_inlinekeyboard_callback()
# Toggle the correct property
if callback.data == "toggle_edit_products":
admin.edit_products = not admin.edit_products
2018-05-04 13:46:07 +00:00
elif callback.data == "toggle_receive_orders":
admin.receive_orders = not admin.receive_orders
elif callback.data == "toggle_create_transactions":
admin.create_transactions = not admin.create_transactions
elif callback.data == "toggle_display_on_help":
admin.display_on_help = not admin.display_on_help
elif callback.data == "cmd_done":
break
self.session.commit()
2020-05-03 10:40:11 +00:00
def __language_menu(self):
"""Select a language."""
log.debug("Displaying __language_menu")
keyboard = []
options: Dict[str, str] = {}
# https://en.wikipedia.org/wiki/List_of_language_names
if "it" in self.cfg["Language"]["enabled_languages"]:
2020-05-03 10:40:11 +00:00
lang = "🇮🇹 Italiano"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "it"
if "en" in self.cfg["Language"]["enabled_languages"]:
2020-05-03 10:40:11 +00:00
lang = "🇬🇧 English"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "en"
if "ru" in self.cfg["Language"]["enabled_languages"]:
2020-05-03 10:40:11 +00:00
lang = "🇷🇺 Русский"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "ru"
if "uk" in self.cfg["Language"]["enabled_languages"]:
2020-05-03 10:40:11 +00:00
lang = "🇺🇦 Українська"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "uk"
if "zh_cn" in self.cfg["Language"]["enabled_languages"]:
2020-08-22 14:40:47 +00:00
lang = "🇨🇳 简体中文"
2020-08-22 12:40:27 +00:00
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "zh_cn"
if "he" in self.cfg["Language"]["enabled_languages"]:
lang = "🇮🇱 עברית"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "he"
if "es_mx" in self.cfg["Language"]["enabled_languages"]:
lang = "🇲🇽 Español"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "es_mx"
if "pt_br" in self.cfg["Language"]["enabled_languages"]:
lang = "🇧🇷 Português"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "pt_br"
if "hi" in self.cfg["Language"]["enabled_languages"]:
lang = "🇮🇳 हिन्दी"
keyboard.append([telegram.KeyboardButton(lang)])
options[lang] = "hi"
2020-05-03 10:40:11 +00:00
# Send the previously created keyboard to the user (ensuring it can be clicked only 1 time)
self.bot.send_message(self.chat.id,
self.loc.get("conversation_language_select"),
reply_markup=telegram.ReplyKeyboardMarkup(keyboard, one_time_keyboard=True))
# Wait for an answer
response = self.__wait_for_specific_message(list(options.keys()))
# Set the language to the corresponding value
self.user.language = options[response]
# Commit the edit to the database
self.session.commit()
# Recreate the localization object
self.__create_localization()
def __create_localization(self):
# Check if the user's language is enabled; if it isn't, change it to the default
if self.user.language not in self.cfg["Language"]["enabled_languages"]:
2020-05-03 10:40:11 +00:00
log.debug(f"User's language '{self.user.language}' is not enabled, changing it to the default")
self.user.language = self.cfg["Language"]["default_language"]
2020-05-03 10:40:11 +00:00
self.session.commit()
# Create a new Localization object
self.loc = localization.Localization(
language=self.user.language,
fallback=self.cfg["Language"]["fallback_language"],
2020-05-03 10:40:11 +00:00
replacements={
"user_string": str(self.user),
"user_mention": self.user.mention(),
"user_full_name": self.user.full_name,
"user_first_name": self.user.first_name,
"today": datetime.datetime.now().strftime("%a %d %b %Y"),
}
)
2018-04-16 19:20:02 +00:00
def __graceful_stop(self, stop_trigger: StopSignal):
2017-12-14 08:40:03 +00:00
"""Handle the graceful stop of the thread."""
log.debug("Gracefully stopping the conversation")
2018-04-16 19:20:02 +00:00
# If the session has expired...
if stop_trigger.reason == "timeout":
# Notify the user that the session has expired and remove the keyboard
self.bot.send_message(self.chat.id, self.loc.get('conversation_expired'),
2018-04-18 19:28:39 +00:00
reply_markup=telegram.ReplyKeyboardRemove())
2018-04-16 19:20:02 +00:00
# If a restart has been requested...
# Do nothing.
2017-12-21 09:42:23 +00:00
# Close the database session
self.session.close()
2017-12-14 08:40:03 +00:00
# End the process
sys.exit(0)