main.load_attachs() - Code Metrics - Inspection of "Merge branch 'release/6.4.0'" - dadyarri/ralph - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 1663f9...461fea )

by Daniil

created 2020-03-22 08:15 UTC

main.load_attachs() C

↳ Parent: Project

Complexity

Conditions

Size

Total Lines	29
Code Lines	28

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	28
dl	0
loc	29
rs	5.9999
c	0
b	0
f	0
cc	10
nop	0

How to fix Complexity

import datetime
import json
import os
import re
from enum import Enum

import requests
from vk_api.bot_longpoll import VkBotEventType

from bot import Bot
from database import Database
from keyboard import Keyboards
from scheduler import Date
from scheduler import Schedule

db = Database(os.environ["DATABASE_URL"])
bot = Bot()
kbs = Keyboards()

bot.auth()
bot.update_version()


class EventTypes(Enum):
    NEW_MESSAGE = VkBotEventType.MESSAGE_NEW


def generate_call_message():
    f = db.get_names_using_status(event["message"]["from_id"])
    students_ids = db.get_call_ids(event["message"]["from_id"])
    if students_ids:
        mentions = bot.generate_mentions(ids=students_ids, names=f)
    else:
        mentions = ""
    message = db.get_call_message(event["message"]["from_id"]) or ""
    message = f"{mentions}\n{message}"
    return message


def send_call_confirm():
    chat_id = int(str(db.get_conversation(event["message"]["from_id"]))[-1])
    message = generate_call_message()
    atch = db.get_call_attaches(event["message"]["from_id"])
    if atch is None:
        atch = ""
    if message != "\n":
        bot.send_message(
            msg=f"В {'тестовую ' if chat_id == 1 else 'основную '}"
            f"беседу будет отправлено сообщение:",
            pid=event["message"]["from_id"],
            keyboard=kbs.prompt(event["message"]["from_id"]),
        )
        bot.send_message(msg=message, pid=event["message"]["from_id"], attachment=atch)
    else:
        db.empty_call_storage(event["message"]["from_id"])
        bot.send_gui(
            pid=event["message"]["from_id"],
            text="Сообщение не может быть пустым. Отмена...",
        )


def load_attachs():
    attachments = []
    for i, v in enumerate(event["message"]["attachments"]):
        m = -1
        m_url = ""
        for ind, val in enumerate(event["message"]["attachments"][i]["photo"]["sizes"]):
            if val["height"] > m:
                m_url = val["url"]
        if ".jpeg" or ".jpg" in m_url:
            ext = ".jpg"
        elif ".png" in m_url:
            ext = ".png"
        else:
            ext = ""
        req = requests.get(m_url)
        server = bot.bot_vk.photos.getMessagesUploadServer()
        with open(f"photo{ext}", "wb") as f:
            f.write(req.content)
        file = open(f"photo{ext}", "rb")
        upload = requests.post(server["upload_url"], files={"photo": file},).json()
        save = bot.bot_vk.photos.saveMessagesPhoto(**upload)
        photo = f"photo{save[0]['owner_id']}_{save[0]['id']}"
        attachments.append(photo)
    atch = ",".join(attachments)
    state = db.get_session_state(event["message"]["from_id"])
    if state == "ask_for_mailing_message":
        db.update_mailing_attaches(event["message"]["from_id"], atch)
    elif state == "ask_for_call_message":
        db.update_call_attaches(event["message"]["from_id"], atch)


for event in bot.longpoll.listen():
    event = {
        "type": event.type,
        "client_info": event.object.client_info,
        "message": event.object.message,
    }
    if (
        event["type"] == EventTypes.NEW_MESSAGE.value
        and (event["message"]["text"] or event["message"]["attachments"])
        and event["message"]["out"] == 0
        and event["message"]["from_id"] == event["message"]["peer_id"]
    ):
        try:
            payload = json.loads(event["message"]["payload"])
        except KeyError:
            payload = {"button": ""}
        text = event["message"]["text"].lower()

        # :blockstart: Запуск интерфейса
        if text in ["начать", "старт", "r"]:
            if not db.is_user_exist(event["message"]["from_id"]):
                db.create_user(event["message"]["from_id"])
            if not db.is_session_exist(event["message"]["from_id"]):
                db.create_session(event["message"]["from_id"])
            bot.send_gui(pid=event["message"]["from_id"])
        # :blockend: Запуск интерфейса

        # :blockstart: Возврат на главный экран
        elif payload["button"] == "home":
            bot.send_gui(text="Главный экран", pid=event["message"]["from_id"])
        # :blockend: Возврат на главный экран

        # :blockstart: Призыв
        elif payload["button"] == "call":
            db.update_session_state(event["message"]["from_id"], "ask_for_call_message")
            if not db.call_session_exist(event["message"]["from_id"]):
                db.create_call_session(event["message"]["from_id"])
            bot.send_message(
                msg="Отправьте сообщение к призыву (есть поддержка изображений)",
                pid=event["message"]["from_id"],
                keyboard=kbs.skip(),
            )
        elif payload["button"] == "letter":
            bot.send_message(
                msg=f"Отправка клавиатуры с фамилиями на букву \"{payload['letter']}\"",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_names_keyboard(payload["letter"]),
            )
        elif payload["button"] == "student":
            ids = db.get_call_ids(event["message"]["from_id"])
            if ids:
                students = ids.split(",")
            else:
                students = [ids]
            if str(db.get_vk_id(payload["id"])) in students:
                bot.send_message(
                    msg=f"{payload['name']} уже был выбран для призыва. Пропуск.",
                    pid=event["message"]["from_id"],
                )
            else:
                db.append_to_call_ids(
                    event["message"]["from_id"], db.get_vk_id(payload["id"])
                )
                bot.send_message(
                    msg=f"{payload['name']} добавлен к списку призыва.",
                    pid=event["message"]["from_id"],
                )
        elif payload["button"] == "back":
            bot.send_message(
                msg="Отправка клавиатуры с алфавитом.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_call_prompt(),
            )
        elif payload["button"] == "skip":
            db.update_session_state(event["message"]["from_id"], "call_configuring")
            bot.send_message(
                msg="Отправка клавиатуры с алфавитом.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_call_prompt(),
            )
        elif (
            db.get_session_state(event["message"]["from_id"]) == "ask_for_call_message"
        ):
            db.update_call_message(
                event["message"]["from_id"], event["message"]["text"]
            )
            if event["message"]["attachments"]:
                load_attachs()
            bot.send_message(
                msg="Отправка клавиатуры призыва",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_call_prompt(),
            )
            db.update_session_state(event["message"]["from_id"], "call_configuring")
        elif payload["button"] == "send_to_all":
            ids = ",".join(db.get_active_students_ids())
            db.update_call_ids(event["message"]["from_id"], ids)
            bot.send_message(
                msg="Все студенты отмечены как получатели уведомления",
                pid=event["message"]["from_id"],
            )
            send_call_confirm()
        elif payload["button"] == "save":
            send_call_confirm()
        elif (
            payload["button"] == "cancel"
            and db.get_session_state(event["message"]["from_id"]) == "call_configuring"
        ):
            db.empty_call_storage(event["message"]["from_id"])
            db.update_session_state(event["message"]["from_id"], "main")
            bot.send_gui(
                text="Выполнение команды отменено.", pid=event["message"]["from_id"]
            )
        elif (
            payload["button"] == "confirm"
            and db.get_session_state(event["message"]["from_id"]) == "call_configuring"
        ):
            bot.log.info("Отправка призыва...")
            cid = db.get_conversation(event["message"]["from_id"])
            text = generate_call_message()
            attachment = db.get_call_attaches(event["message"]["from_id"])
            if attachment is None:
                attachment = ""
            bot.send_message(pid=cid, msg=text, attachment=attachment)
            db.empty_call_storage(event["message"]["from_id"])
            db.update_session_state(event["message"]["from_id"], "main")
            bot.send_gui(text="Сообщение отправлено.", pid=event["message"]["from_id"])
        elif (
            payload["button"] == "deny"
            and db.get_session_state(event["message"]["from_id"]) == "call_configuring"
        ):
            db.update_call_message(event["message"]["from_id"], " ")
            db.update_call_ids(event["message"]["from_id"], " ")
            db.update_session_state(event["message"]["from_id"], "main")
            bot.send_gui(
                text="Выполнение команды отменено.", pid=event["message"]["from_id"]
            )
        elif payload["button"] == "chconv_call":
            conv = db.get_conversation(event["message"]["from_id"])
            chat = int(str(conv)[-1])
            if chat == 1:
                db.update_conversation(event["message"]["from_id"], 2000000002)
                chat = 2
            elif chat == 2:
                db.update_conversation(event["message"]["from_id"], 2000000001)
                chat = 1
            msg = db.get_call_message(event["message"]["from_id"])
            bot.send_message(
                msg=f"Теперь это сообщение будет отправлено в "
                f"{'тестовую' if chat == 1 else 'основную'} беседу:",
                pid=event["message"]["from_id"],
            )
        elif payload["button"] == "chnames_call":
            if db.get_names_using_status(event["message"]["from_id"]):
                status = 0
            else:
                status = 1
            db.update_names_using_status(event["message"]["from_id"], status)
            send_call_confirm()
        # :blockend: Призыв

        # :blockstart: Расписание
        elif payload["button"] == "schedule":
            bot.send_message(
                msg="Отправка клавиатуры с расписанием.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_schedule_keyboard(),
            )
        elif payload["button"] == "today":
            d = Date()
            s = Schedule(d.today)
            s.get_raw()
            if s.is_exist():
                schedule = s.generate()
                bot.send_message(msg=schedule, pid=event["message"]["from_id"])
            else:
                bot.send_message(
                    msg="Расписание отсутствует.", pid=event["message"]["from_id"]
                )
        elif payload["button"] == "tomorrow":
            d = Date()
            s = Schedule(d.tomorrow)
            s.get_raw()
            if s.is_exist():
                schedule = s.generate()
                bot.send_message(msg=schedule, pid=event["message"]["from_id"])
            else:
                bot.send_message(
                    msg="Расписание отсутствует.", pid=event["message"]["from_id"]
                )
        elif payload["button"] == "day_after_tomorrow":
            d = Date()
            s = Schedule(d.day_after_tomorrow)
            s.get_raw()
            if s.is_exist():
                schedule = s.generate()
                bot.send_message(msg=schedule, pid=event["message"]["from_id"])
            else:
                bot.send_message(
                    msg="Расписание отсутствует.", pid=event["message"]["from_id"]
                )
        elif payload["button"] == "arbitrary":
            bot.send_message(
                msg="Напишите дату в формате ДД-ММ-ГГГГ.",
                pid=event["message"]["from_id"],
                keyboard=kbs.cancel(),
            )
            db.update_session_state(
                event["message"]["from_id"], "ask_for_schedule_date"
            )
        elif (
            payload["button"] == "cancel"
            and db.get_session_state(event["message"]["from_id"])
            == "ask_for_schedule_date"
        ):
            db.update_session_state(event["message"]["from_id"], "main")
            bot.send_message(
                msg="Выполнение команды отменено.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_schedule_keyboard(),
            )
        elif (
            db.get_session_state(event["message"]["from_id"]) == "ask_for_schedule_date"
        ):
            if re.match(r"^\d\d(.|-|/)\d\d(.|-|/)20\d\d$", event["message"]["text"]):
                try:
                    d = datetime.datetime.strptime(
                        event["message"]["text"], "%d-%m-%Y"
                    ).strftime("%Y-%m-%d")
                except ValueError:
                    bot.send_message(
                        msg="Неверный формат даты. Попробуйте еще раз.",
                        pid=event["message"]["from_id"],
                    )
                else:
                    s = Schedule(d)
                    s.get_raw()
                    if s.is_exist():
                        schedule = s.generate()
                        bot.send_message(
                            msg=schedule,
                            pid=event["message"]["from_id"],
                            keyboard=kbs.generate_schedule_keyboard(),
                        )
                        db.update_session_state(event["message"]["from_id"], "main")
                    else:
                        bot.send_message(
                            msg="Расписание отсутствует.\nПопробуй указать другую "
                            "дату.",
                            pid=event["message"]["from_id"],
                        )
                        db.update_session_state(
                            event["message"]["from_id"], "ask_for_schedule_date"
                        )
            else:
                bot.send_message(
                    msg="Неверный формат даты. Попробуйте еще раз.",
                    pid=event["message"]["from_id"],
                )
        # :blockend: Расписание

        # :blockstart: Рассылки
        elif payload["button"] == "mailings":
            bot.send_message(
                msg="Отправка клавиатуры со списком рассылок.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailings_keyboard(),
            )
        elif payload["button"] == "mailing":
            if not db.mailing_session_exist(event["message"]["from_id"]):
                db.create_mailing_session(event["message"]["from_id"])
            db.update_mailing_session(event["message"]["from_id"], payload["slug"])
            bot.send_message(
                msg=f"Меню управления рассылкой \"{payload['name']}\":",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailing_mgmt(
                    is_admin=bot.is_admin(event["message"]["from_id"]),
                    slug=payload["slug"],
                    user_id=event["message"]["from_id"],
                ),
            )
        elif payload["button"] == "subscribe":
            db.query(
                f"UPDATE vk_subscriptions SET {payload['slug']}=1 WHERE "
                f"user_id={payload['user_id']}"
            )
            bot.send_message(
                msg="Вы были успешно подписаны на рассылку.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailing_mgmt(
                    is_admin=bot.is_admin(event["message"]["from_id"]),
                    slug=payload["slug"],
                    user_id=event["message"]["from_id"],
                ),
            )
        elif payload["button"] == "unsubscribe":
            db.query(
                f"UPDATE vk_subscriptions SET {payload['slug']}=0 WHERE "
                f"user_id={payload['user_id']}"
            )
            bot.send_message(
                msg="Вы были успешно отписаны от рассылки.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailing_mgmt(
                    is_admin=bot.is_admin(event["message"]["from_id"]),
                    slug=payload["slug"],
                    user_id=event["message"]["from_id"],
                ),
            )
        elif payload["button"] == "send_mailing":
            db.update_session_state(
                event["message"]["from_id"], "ask_for_mailing_message"
            )
            bot.send_message(
                msg="Отправьте текст рассылки (есть поддержка изображений)",
                pid=event["message"]["from_id"],
                keyboard=kbs.cancel(),
            )
        elif (
            payload["button"] == "cancel"
            and db.get_session_state(event["message"]["from_id"])
            == "ask_for_mailing_message"
        ):
            bot.send_message(
                msg="Выполнение команды отменено. Возвращаюсь на экран управления "
                "рассылкой.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailing_mgmt(
                    is_admin=bot.is_admin(event["message"]["from_id"]),
                    slug=db.get_mailing_session(event["message"]["from_id"]),
                    user_id=event["message"]["from_id"],
                ),
            )
            db.update_session_state(event["message"]["from_id"], "main")
        elif (
            db.get_session_state(event["message"]["from_id"])
            == "ask_for_mailing_message"
        ):
            db.update_mailing_message(
                event["message"]["from_id"], event["message"]["text"]
            )
            if event["message"]["attachments"]:
                load_attachs()
            bot.send_message(
                msg="Всем подписчикам рассылки будет отправлено сообщение с указанным вами текстом",
                pid=event["message"]["from_id"],
                keyboard=kbs.prompt(),
                forward=f"{event['message']['id']}",
            )
            db.update_session_state(event["message"]["from_id"], "prompt_mailing")
        elif (
            payload["button"] == "confirm"
            and db.get_session_state(event["message"]["from_id"]) == "prompt_mailing"
        ):
            attach = db.get_mailing_attaches(event["message"]["from_id"])
            if attach is None:
                attach = ""
            bot.send_mailing(
                slug=db.get_mailing_session(event["message"]["from_id"]),
                text=db.get_mailing_message(event["message"]["from_id"]),
                attach=attach,
            )
            bot.send_message(
                msg="Рассылка отправлена.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailings_keyboard(),
            )
        elif (
            payload["button"] == "deny"
            and db.get_session_state(event["message"]["from_id"]) == "prompt_mailing"
        ):
            db.empty_mailing_storage(event["message"]["from_id"])
            bot.send_message(
                msg="Отправка рассылки отменена.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_mailings_keyboard(),
            )
        # :blockend: Рассылки

        # :blockstart: Параметры
        elif payload["button"] == "prefs":
            bot.send_message(
                msg="Параметры",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_prefs_keyboard(),
            )

        elif payload["button"] == "chconv":
            chat = db.get_conversation(event["message"]["from_id"])
            if chat == 2000000001:
                bot.send_message(
                    msg="Тестовая беседа сейчас активна",
                    pid=event["message"]["from_id"],
                    keyboard=kbs.generate_conv_selector(chat),
                )
            elif chat == 2000000002:
                bot.send_message(
                    msg="Основная беседа сейчас активна",
                    pid=event["message"]["from_id"],
                    keyboard=kbs.generate_conv_selector(chat),
                )

        elif payload["button"] == "select_main_conv":
            chat = 2000000002
            db.update_conversation(event["message"]["from_id"], chat)
            bot.send_message(
                msg="Основная беседа активна.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_conv_selector(chat),
            )
        elif payload["button"] == "select_test_conv":
            chat = 2000000001
            db.update_conversation(event["message"]["from_id"], chat)
            bot.send_message(
                msg="Тестовая беседа активна.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_conv_selector(chat),
            )

        elif payload["button"] == "names":
            status = db.get_names_using_status(event["message"]["from_id"])
            msg = (
                f"Использование имён в призыве "
                f"{'активно' if status else 'неактивно'}."
            )
            bot.send_message(
                msg=msg,
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_names_selector(status),
            )

        elif payload["button"] == "off_using_names":
            status = 0
            db.update_names_using_status(event["message"]["from_id"], status)
            bot.send_message(
                msg="Использование имён в призыве отключено.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_names_selector(bool(status)),
            )
        elif payload["button"] == "on_using_names":
            status = 1
            db.update_names_using_status(event["message"]["from_id"], status)
            bot.send_message(
                msg="Использование имён в призыве включено.",
                pid=event["message"]["from_id"],
                keyboard=kbs.generate_names_selector(bool(status)),
            )
        # :blockend: Параметры


1			import datetime
2			import json
3			import os
4			import re
5			from enum import Enum
6
7			import requests
8			from vk_api.bot_longpoll import VkBotEventType
9
10			from bot import Bot
11			from database import Database
12			from keyboard import Keyboards
13			from scheduler import Date
14			from scheduler import Schedule
15
16			db = Database(os.environ["DATABASE_URL"])
17			bot = Bot()
18			kbs = Keyboards()
19
20			bot.auth()
21			bot.update_version()
22
23
24			class EventTypes(Enum):
25			NEW_MESSAGE = VkBotEventType.MESSAGE_NEW
26
27
28			def generate_call_message():
29			f = db.get_names_using_status(event["message"]["from_id"])
30			students_ids = db.get_call_ids(event["message"]["from_id"])
31			if students_ids:
32			mentions = bot.generate_mentions(ids=students_ids, names=f)
33			else:
34			mentions = ""
35			message = db.get_call_message(event["message"]["from_id"]) or ""
36			message = f"{mentions}\n{message}"
37			return message
38
39
40			def send_call_confirm():
41			chat_id = int(str(db.get_conversation(event["message"]["from_id"]))[-1])
42			message = generate_call_message()
43			atch = db.get_call_attaches(event["message"]["from_id"])
44			if atch is None:
45			atch = ""
46			if message != "\n":
47			bot.send_message(
48			msg=f"В {'тестовую ' if chat_id == 1 else 'основную '}"
49			f"беседу будет отправлено сообщение:",
50			pid=event["message"]["from_id"],
51			keyboard=kbs.prompt(event["message"]["from_id"]),
52			)
53			bot.send_message(msg=message, pid=event["message"]["from_id"], attachment=atch)
54			else:
55			db.empty_call_storage(event["message"]["from_id"])
56			bot.send_gui(
57			pid=event["message"]["from_id"],
58			text="Сообщение не может быть пустым. Отмена...",
59			)
60
61
62			def load_attachs():
63			attachments = []
64			for i, v in enumerate(event["message"]["attachments"]):
65			m = -1
66			m_url = ""
67			for ind, val in enumerate(event["message"]["attachments"][i]["photo"]["sizes"]):
68			if val["height"] > m:
69			m_url = val["url"]
70			if ".jpeg" or ".jpg" in m_url:
71			ext = ".jpg"
72			elif ".png" in m_url:
73			ext = ".png"
74			else:
75			ext = ""
76			req = requests.get(m_url)
77			server = bot.bot_vk.photos.getMessagesUploadServer()
78			with open(f"photo{ext}", "wb") as f:
79			f.write(req.content)
80			file = open(f"photo{ext}", "rb")
81			upload = requests.post(server["upload_url"], files={"photo": file},).json()
82			save = bot.bot_vk.photos.saveMessagesPhoto(**upload)
83			photo = f"photo{save[0]['owner_id']}_{save[0]['id']}"
84			attachments.append(photo)
85			atch = ",".join(attachments)
86			state = db.get_session_state(event["message"]["from_id"])
87			if state == "ask_for_mailing_message":
88			db.update_mailing_attaches(event["message"]["from_id"], atch)
89			elif state == "ask_for_call_message":
90			db.update_call_attaches(event["message"]["from_id"], atch)
91
92
93			for event in bot.longpoll.listen():
94			event = {
95			"type": event.type,
96			"client_info": event.object.client_info,
97			"message": event.object.message,
98			}
99			if (
100			event["type"] == EventTypes.NEW_MESSAGE.value
101			and (event["message"]["text"] or event["message"]["attachments"])
102			and event["message"]["out"] == 0
103			and event["message"]["from_id"] == event["message"]["peer_id"]
104			):
105			try:
106			payload = json.loads(event["message"]["payload"])
107			except KeyError:
108			payload = {"button": ""}
109			text = event["message"]["text"].lower()
110
111			# :blockstart: Запуск интерфейса
112			if text in ["начать", "старт", "r"]:
113			if not db.is_user_exist(event["message"]["from_id"]):
114			db.create_user(event["message"]["from_id"])
115			if not db.is_session_exist(event["message"]["from_id"]):
116			db.create_session(event["message"]["from_id"])
117			bot.send_gui(pid=event["message"]["from_id"])
118			# :blockend: Запуск интерфейса
119
120			# :blockstart: Возврат на главный экран
121			elif payload["button"] == "home":
122			bot.send_gui(text="Главный экран", pid=event["message"]["from_id"])
123			# :blockend: Возврат на главный экран
124
125			# :blockstart: Призыв
126			elif payload["button"] == "call":
127			db.update_session_state(event["message"]["from_id"], "ask_for_call_message")
128			if not db.call_session_exist(event["message"]["from_id"]):
129			db.create_call_session(event["message"]["from_id"])
130			bot.send_message(
131			msg="Отправьте сообщение к призыву (есть поддержка изображений)",
132			pid=event["message"]["from_id"],
133			keyboard=kbs.skip(),
134			)
135			elif payload["button"] == "letter":
136			bot.send_message(
137			msg=f"Отправка клавиатуры с фамилиями на букву \"{payload['letter']}\"",
138			pid=event["message"]["from_id"],
139			keyboard=kbs.generate_names_keyboard(payload["letter"]),
140			)
141			elif payload["button"] == "student":
142			ids = db.get_call_ids(event["message"]["from_id"])
143			if ids:
144			students = ids.split(",")
145			else:
146			students = [ids]
147			if str(db.get_vk_id(payload["id"])) in students:
148			bot.send_message(
149			msg=f"{payload['name']} уже был выбран для призыва. Пропуск.",
150			pid=event["message"]["from_id"],
151			)
152			else:
153			db.append_to_call_ids(
154			event["message"]["from_id"], db.get_vk_id(payload["id"])
155			)
156			bot.send_message(
157			msg=f"{payload['name']} добавлен к списку призыва.",
158			pid=event["message"]["from_id"],
159			)
160			elif payload["button"] == "back":
161			bot.send_message(
162			msg="Отправка клавиатуры с алфавитом.",
163			pid=event["message"]["from_id"],
164			keyboard=kbs.generate_call_prompt(),
165			)
166			elif payload["button"] == "skip":
167			db.update_session_state(event["message"]["from_id"], "call_configuring")
168			bot.send_message(
169			msg="Отправка клавиатуры с алфавитом.",
170			pid=event["message"]["from_id"],
171			keyboard=kbs.generate_call_prompt(),
172			)
173			elif (
174			db.get_session_state(event["message"]["from_id"]) == "ask_for_call_message"
175			):
176			db.update_call_message(
177			event["message"]["from_id"], event["message"]["text"]
178			)
179			if event["message"]["attachments"]:
180			load_attachs()
181			bot.send_message(
182			msg="Отправка клавиатуры призыва",
183			pid=event["message"]["from_id"],
184			keyboard=kbs.generate_call_prompt(),
185			)
186			db.update_session_state(event["message"]["from_id"], "call_configuring")
187			elif payload["button"] == "send_to_all":
188			ids = ",".join(db.get_active_students_ids())
189			db.update_call_ids(event["message"]["from_id"], ids)
190			bot.send_message(
191			msg="Все студенты отмечены как получатели уведомления",
192			pid=event["message"]["from_id"],
193			)
194			send_call_confirm()
195			elif payload["button"] == "save":
196			send_call_confirm()
197			elif (
198			payload["button"] == "cancel"
199			and db.get_session_state(event["message"]["from_id"]) == "call_configuring"
200			):
201			db.empty_call_storage(event["message"]["from_id"])
202			db.update_session_state(event["message"]["from_id"], "main")
203			bot.send_gui(
204			text="Выполнение команды отменено.", pid=event["message"]["from_id"]
205			)
206			elif (
207			payload["button"] == "confirm"
208			and db.get_session_state(event["message"]["from_id"]) == "call_configuring"
209			):
210			bot.log.info("Отправка призыва...")
211			cid = db.get_conversation(event["message"]["from_id"])
212			text = generate_call_message()
213			attachment = db.get_call_attaches(event["message"]["from_id"])
214			if attachment is None:
215			attachment = ""
216			bot.send_message(pid=cid, msg=text, attachment=attachment)
217			db.empty_call_storage(event["message"]["from_id"])
218			db.update_session_state(event["message"]["from_id"], "main")
219			bot.send_gui(text="Сообщение отправлено.", pid=event["message"]["from_id"])
220			elif (
221			payload["button"] == "deny"
222			and db.get_session_state(event["message"]["from_id"]) == "call_configuring"
223			):
224			db.update_call_message(event["message"]["from_id"], " ")
225			db.update_call_ids(event["message"]["from_id"], " ")
226			db.update_session_state(event["message"]["from_id"], "main")
227			bot.send_gui(
228			text="Выполнение команды отменено.", pid=event["message"]["from_id"]
229			)
230			elif payload["button"] == "chconv_call":
231			conv = db.get_conversation(event["message"]["from_id"])
232			chat = int(str(conv)[-1])
233			if chat == 1:
234			db.update_conversation(event["message"]["from_id"], 2000000002)
235			chat = 2
236			elif chat == 2:
237			db.update_conversation(event["message"]["from_id"], 2000000001)
238			chat = 1
239			msg = db.get_call_message(event["message"]["from_id"])
240			bot.send_message(
241			msg=f"Теперь это сообщение будет отправлено в "
242			f"{'тестовую' if chat == 1 else 'основную'} беседу:",
243			pid=event["message"]["from_id"],
244			)
245			elif payload["button"] == "chnames_call":
246			if db.get_names_using_status(event["message"]["from_id"]):
247			status = 0
248			else:
249			status = 1
250			db.update_names_using_status(event["message"]["from_id"], status)
251			send_call_confirm()
252			# :blockend: Призыв
253
254			# :blockstart: Расписание
255			elif payload["button"] == "schedule":
256			bot.send_message(
257			msg="Отправка клавиатуры с расписанием.",
258			pid=event["message"]["from_id"],
259			keyboard=kbs.generate_schedule_keyboard(),
260			)
261			elif payload["button"] == "today":
262			d = Date()
263			s = Schedule(d.today)
264			s.get_raw()
265			if s.is_exist():
266			schedule = s.generate()
267			bot.send_message(msg=schedule, pid=event["message"]["from_id"])
268			else:
269			bot.send_message(
270			msg="Расписание отсутствует.", pid=event["message"]["from_id"]
271			)
272			elif payload["button"] == "tomorrow":
273			d = Date()
274			s = Schedule(d.tomorrow)
275			s.get_raw()
276			if s.is_exist():
277			schedule = s.generate()
278			bot.send_message(msg=schedule, pid=event["message"]["from_id"])
279			else:
280			bot.send_message(
281			msg="Расписание отсутствует.", pid=event["message"]["from_id"]
282			)
283			elif payload["button"] == "day_after_tomorrow":
284			d = Date()
285			s = Schedule(d.day_after_tomorrow)
286			s.get_raw()
287			if s.is_exist():
288			schedule = s.generate()
289			bot.send_message(msg=schedule, pid=event["message"]["from_id"])
290			else:
291			bot.send_message(
292			msg="Расписание отсутствует.", pid=event["message"]["from_id"]
293			)
294			elif payload["button"] == "arbitrary":
295			bot.send_message(
296			msg="Напишите дату в формате ДД-ММ-ГГГГ.",
297			pid=event["message"]["from_id"],
298			keyboard=kbs.cancel(),
299			)
300			db.update_session_state(
301			event["message"]["from_id"], "ask_for_schedule_date"
302			)
303			elif (
304			payload["button"] == "cancel"
305			and db.get_session_state(event["message"]["from_id"])
306			== "ask_for_schedule_date"
307			):
308			db.update_session_state(event["message"]["from_id"], "main")
309			bot.send_message(
310			msg="Выполнение команды отменено.",
311			pid=event["message"]["from_id"],
312			keyboard=kbs.generate_schedule_keyboard(),
313			)
314			elif (
315			db.get_session_state(event["message"]["from_id"]) == "ask_for_schedule_date"
316			):
317			if re.match(r"^\d\d(.\|-\|/)\d\d(.\|-\|/)20\d\d$", event["message"]["text"]):
318			try:
319			d = datetime.datetime.strptime(
320			event["message"]["text"], "%d-%m-%Y"
321			).strftime("%Y-%m-%d")
322			except ValueError:
323			bot.send_message(
324			msg="Неверный формат даты. Попробуйте еще раз.",
325			pid=event["message"]["from_id"],
326			)
327			else:
328			s = Schedule(d)
329			s.get_raw()
330			if s.is_exist():
331			schedule = s.generate()
332			bot.send_message(
333			msg=schedule,
334			pid=event["message"]["from_id"],
335			keyboard=kbs.generate_schedule_keyboard(),
336			)
337			db.update_session_state(event["message"]["from_id"], "main")
338			else:
339			bot.send_message(
340			msg="Расписание отсутствует.\nПопробуй указать другую "
341			"дату.",
342			pid=event["message"]["from_id"],
343			)
344			db.update_session_state(
345			event["message"]["from_id"], "ask_for_schedule_date"
346			)
347			else:
348			bot.send_message(
349			msg="Неверный формат даты. Попробуйте еще раз.",
350			pid=event["message"]["from_id"],
351			)
352			# :blockend: Расписание
353
354			# :blockstart: Рассылки
355			elif payload["button"] == "mailings":
356			bot.send_message(
357			msg="Отправка клавиатуры со списком рассылок.",
358			pid=event["message"]["from_id"],
359			keyboard=kbs.generate_mailings_keyboard(),
360			)
361			elif payload["button"] == "mailing":
362			if not db.mailing_session_exist(event["message"]["from_id"]):
363			db.create_mailing_session(event["message"]["from_id"])
364			db.update_mailing_session(event["message"]["from_id"], payload["slug"])
365			bot.send_message(
366			msg=f"Меню управления рассылкой \"{payload['name']}\":",
367			pid=event["message"]["from_id"],
368			keyboard=kbs.generate_mailing_mgmt(
369			is_admin=bot.is_admin(event["message"]["from_id"]),
370			slug=payload["slug"],
371			user_id=event["message"]["from_id"],
372			),
373			)
374			elif payload["button"] == "subscribe":
375			db.query(
376			f"UPDATE vk_subscriptions SET {payload['slug']}=1 WHERE "
377			f"user_id={payload['user_id']}"
378			)
379			bot.send_message(
380			msg="Вы были успешно подписаны на рассылку.",
381			pid=event["message"]["from_id"],
382			keyboard=kbs.generate_mailing_mgmt(
383			is_admin=bot.is_admin(event["message"]["from_id"]),
384			slug=payload["slug"],
385			user_id=event["message"]["from_id"],
386			),
387			)
388			elif payload["button"] == "unsubscribe":
389			db.query(
390			f"UPDATE vk_subscriptions SET {payload['slug']}=0 WHERE "
391			f"user_id={payload['user_id']}"
392			)
393			bot.send_message(
394			msg="Вы были успешно отписаны от рассылки.",
395			pid=event["message"]["from_id"],
396			keyboard=kbs.generate_mailing_mgmt(
397			is_admin=bot.is_admin(event["message"]["from_id"]),
398			slug=payload["slug"],
399			user_id=event["message"]["from_id"],
400			),
401			)
402			elif payload["button"] == "send_mailing":
403			db.update_session_state(
404			event["message"]["from_id"], "ask_for_mailing_message"
405			)
406			bot.send_message(
407			msg="Отправьте текст рассылки (есть поддержка изображений)",
408			pid=event["message"]["from_id"],
409			keyboard=kbs.cancel(),
410			)
411			elif (
412			payload["button"] == "cancel"
413			and db.get_session_state(event["message"]["from_id"])
414			== "ask_for_mailing_message"
415			):
416			bot.send_message(
417			msg="Выполнение команды отменено. Возвращаюсь на экран управления "
418			"рассылкой.",
419			pid=event["message"]["from_id"],
420			keyboard=kbs.generate_mailing_mgmt(
421			is_admin=bot.is_admin(event["message"]["from_id"]),
422			slug=db.get_mailing_session(event["message"]["from_id"]),
423			user_id=event["message"]["from_id"],
424			),
425			)
426			db.update_session_state(event["message"]["from_id"], "main")
427			elif (
428			db.get_session_state(event["message"]["from_id"])
429			== "ask_for_mailing_message"
430			):
431			db.update_mailing_message(
432			event["message"]["from_id"], event["message"]["text"]
433			)
434			if event["message"]["attachments"]:
435			load_attachs()
436			bot.send_message(
437			msg="Всем подписчикам рассылки будет отправлено сообщение с указанным вами текстом",
438			pid=event["message"]["from_id"],
439			keyboard=kbs.prompt(),
440			forward=f"{event['message']['id']}",
441			)
442			db.update_session_state(event["message"]["from_id"], "prompt_mailing")
443			elif (
444			payload["button"] == "confirm"
445			and db.get_session_state(event["message"]["from_id"]) == "prompt_mailing"
446			):
447			attach = db.get_mailing_attaches(event["message"]["from_id"])
448			if attach is None:
449			attach = ""
450			bot.send_mailing(
451			slug=db.get_mailing_session(event["message"]["from_id"]),
452			text=db.get_mailing_message(event["message"]["from_id"]),
453			attach=attach,
454			)
455			bot.send_message(
456			msg="Рассылка отправлена.",
457			pid=event["message"]["from_id"],
458			keyboard=kbs.generate_mailings_keyboard(),
459			)
460			elif (
461			payload["button"] == "deny"
462			and db.get_session_state(event["message"]["from_id"]) == "prompt_mailing"
463			):
464			db.empty_mailing_storage(event["message"]["from_id"])
465			bot.send_message(
466			msg="Отправка рассылки отменена.",
467			pid=event["message"]["from_id"],
468			keyboard=kbs.generate_mailings_keyboard(),
469			)
470			# :blockend: Рассылки
471
472			# :blockstart: Параметры
473			elif payload["button"] == "prefs":
474			bot.send_message(
475			msg="Параметры",
476			pid=event["message"]["from_id"],
477			keyboard=kbs.generate_prefs_keyboard(),
478			)
479
480			elif payload["button"] == "chconv":
481			chat = db.get_conversation(event["message"]["from_id"])
482			if chat == 2000000001:
483			bot.send_message(
484			msg="Тестовая беседа сейчас активна",
485			pid=event["message"]["from_id"],
486			keyboard=kbs.generate_conv_selector(chat),
487			)
488			elif chat == 2000000002:
489			bot.send_message(
490			msg="Основная беседа сейчас активна",
491			pid=event["message"]["from_id"],
492			keyboard=kbs.generate_conv_selector(chat),
493			)
494
495			elif payload["button"] == "select_main_conv":
496			chat = 2000000002
497			db.update_conversation(event["message"]["from_id"], chat)
498			bot.send_message(
499			msg="Основная беседа активна.",
500			pid=event["message"]["from_id"],
501			keyboard=kbs.generate_conv_selector(chat),
502			)
503			elif payload["button"] == "select_test_conv":
504			chat = 2000000001
505			db.update_conversation(event["message"]["from_id"], chat)
506			bot.send_message(
507			msg="Тестовая беседа активна.",
508			pid=event["message"]["from_id"],
509			keyboard=kbs.generate_conv_selector(chat),
510			)
511
512			elif payload["button"] == "names":
513			status = db.get_names_using_status(event["message"]["from_id"])
514			msg = (
515			f"Использование имён в призыве "
516			f"{'активно' if status else 'неактивно'}."
517			)
518			bot.send_message(
519			msg=msg,
520			pid=event["message"]["from_id"],
521			keyboard=kbs.generate_names_selector(status),
522			)
523
524			elif payload["button"] == "off_using_names":
525			status = 0
526			db.update_names_using_status(event["message"]["from_id"], status)
527			bot.send_message(
528			msg="Использование имён в призыве отключено.",
529			pid=event["message"]["from_id"],
530			keyboard=kbs.generate_names_selector(bool(status)),
531			)
532			elif payload["button"] == "on_using_names":
533			status = 1
534			db.update_names_using_status(event["message"]["from_id"], status)
535			bot.send_message(
536			msg="Использование имён в призыве включено.",
537			pid=event["message"]["from_id"],
538			keyboard=kbs.generate_names_selector(bool(status)),
539			)
540			# :blockend: Параметры
541

dadyarri / ralph

Push — master ( 1663f9...461fea )

main.load_attachs() C

Complexity

Size

Duplication

Importance

How to fix Complexity

Complexity

Duplication Side-by-Side

Filter issues like