TFinance

1
from items import User
2

3

4
# Исключения для игры
5
class StockSelectedAlready(Exception):  # Акция уже выбрана
6
    pass
7

8

9
class PredictionAlreadySet(Exception):  # Прогноз на акцию уже установлен
10
    pass
11

12

13
# Создать экземпляр класса User
14
def create_user(update) -> User:
15
    return User(update.effective_user.to_dict())
16

17

18
# Установить прогноз акции
19
def generate_prediction(query, database, user: User, prediction: str):
20
    message_id = query.message.message_id  # Получаем id сообщения, для нахождения нужной сессии игры.
21
    database.add_prediction(user, database.get_selected_stock_byid(user, message_id), prediction)
22
    database.remove_selected_stock(user, message_id)  # Удаляем акцию из выбранных.
23

24

25
# Сообщить о победе пользователя.
26
def user_won(context, database, user: User, stock: str):
27
    context.bot.send_message(chat_id=user.id, text=f"Прогноз {stock} был верным. "
28
                                                   f"\nВы получили 1 очко. "
29
                                                   f"\nПосмотреть кол-во очков можно, "
30
                                                   f"использовав /stats.")
31
    database.add_point(user)
32