Как реализовать создание заявки пользователем в боте ВК Python?

Question

[Георгий Салтыков]

Суть бота такова: пользователь создает заявку, последовательно отвечая на вопросы, которые задает бот. Потом заявка отправляется на рассмотрение модератору и в итоге пользователь получает ответ( пока что это не реализовано). Чтобы это все реализовать была создана функция которая и создает данную заявку. Вопрос таков: есть ли другие способы реализовать заполнение заявки или мой код вполне себе ничего.
P.s. Не судите строго это мой первый проект в плане программирования. Хотелость бы реальных советов и адекватной критики. Заранее спасибо)

import vk_api
from vk_api import keyboard
from vk_api.longpoll import VkLongPoll, VkEventType 
from vk_api import VkUpload
from vk_api.utils import get_random_id
from vk_api.keyboard import VkKeyboard, VkKeyboardColor 

# \\\КЛАВИАТУРЫ
keyboard1 = VkKeyboard(one_time=True)
keyboard1.add_button('Создать заявку', color=VkKeyboardColor.POSITIVE)

keyboard2 = VkKeyboard(one_time=False)
keyboard2.add_button('Назад', color=VkKeyboardColor.SECONDARY)
keyboard2.add_button('Отменить', color=VkKeyboardColor.NEGATIVE)

keyboard3 = VkKeyboard(one_time=True)
keyboard3.add_button('Сохранить', color=VkKeyboardColor.POSITIVE)
keyboard3.add_line()
keyboard3.add_button('Назад', color=VkKeyboardColor.SECONDARY)
keyboard3.add_button('Отменить', color=VkKeyboardColor.NEGATIVE)

# \\\ФУНКЦИЯ ОТПРАВКИ СООБЩЕНИЯ
def write_message (sender, message):
    if i==1: 
        authorize.method('messages.send', {'user_id': sender, 'message':  message, 'random_id': get_random_id(), 'keyboard': keyboard1.get_keyboard()})
    elif i==2:
        authorize.method('messages.send', {'user_id': sender, 'message':  message, 'random_id': get_random_id(), 'keyboard': keyboard2.get_keyboard()})
    elif i==3:
        authorize.method('messages.send', {'user_id': sender, 'message':  message, 'random_id': get_random_id(), 'keyboard': keyboard3.get_keyboard()})

# \\\ФУНКЦИЯ СОЗДАНИЯ АНКЕТЫ 
def reg():
    global i
    questionnaire = []
    for event in longpoll.listen():  
        if event.type == VkEventType.MESSAGE_NEW and event.to_me:
            name_user = event.text
            sender = event.user_id
            if name_user == event.text and name_user != "Назад" and name_user != "Отменить":    
                write_message(sender, "Вы ввели имя: "+ name_user + "\nВведите Ваш возраст")
                questionnaire.append(name_user)
                for event in longpoll.listen():  
                    if event.type == VkEventType.MESSAGE_NEW and event.to_me:
                        year_user = event.text
                        sender = event.user_id
                        if year_user == event.text and year_user != "Назад" and year_user != 'Отменить': 
                            write_message(sender, "Вам "+ year_user + " лет\n С какого Вы города?")
                            questionnaire.append(year_user)
                            for event in longpoll.listen():  
                                if event.type == VkEventType.MESSAGE_NEW and event.to_me:
                                    city = event.text
                                    sender = event.user_id
                                    if city == event.text and city != "Назад" and city != "Отменить":
                                        write_message(sender, "Вы с г. "+ city + "\n" + "Заполните заявку, укажите все нужные Вам критерии")
                                        questionnaire.append(city)
                                        for event in longpoll.listen():
                                            if event.type == VkEventType.MESSAGE_NEW and event.to_me:
                                                text_user = event.text
                                                sender = event.user_id
                                                if text_user == event.text and text_user != "Назад" and text_user != "Отменить":
                                                    i=3
                                                    write_message(sender, "Ваша заявка: \n"+ name_user + " " + year_user + " " + city + "\n" + text_user)
                                                    questionnaire.append(text_user)
                                                    for event in longpoll.listen():
                                                        if event.type == VkEventType.MESSAGE_NEW and event.to_me:
                                                            end_user = event.text
                                                            sender = event.user_id
                                                            if end_user =="Сохранить" and end_user != "Назад" and end_user != "Отменить":
                                                                i=1
                                                                write_message(sender, "Вы успешно создали заявку")
                                                                return questionnaire
                                                            elif end_user =="Назад":
                                                                i=2
                                                                write_message(sender, "Измените текст заявки")
                                                                break
                                                            elif end_user =="Отменить":
                                                                i=1
                                                                write_message(sender, "Заявка отменена")
                                                                questionnaire = []
                                                                return questionnaire
                                                elif text_user == "Назад":
                                                    write_message(sender, "Измени свой город")
                                                    break
                                                elif text_user == "Отменить":
                                                    i=1
                                                    text_user = None
                                                    questionnaire = []
                                                    write_message(sender, "Заявка отменена")
                                                    return questionnaire
                                    elif city == "Назад":
                                        write_message(sender, "Измени свой возраст")
                                        break
                                    elif city == "Отменить":
                                        i=1
                                        city = None
                                        questionnaire = []
                                        write_message(sender, "Заявка отменена")
                                        return questionnaire
                        elif year_user == "Назад":
                            write_message(sender, "Введи имя заново")
                            break
                        elif year_user == "Отменить":
                            i=1
                            year_user = None
                            questionnaire = []
                            write_message(sender, "Заявка отменена")
                            return questionnaire
            elif name_user == "Назад":
                i=1
                name_user = None
                questionnaire = []
                write_message(sender, "Заявка отменена")
                return questionnaire
            elif name_user == "Отменить":
                i=1
                name_user = None
                questionnaire = []
                write_message(sender, "Заявка отменена")
                return questionnaire

# \\\АВТОРИЗАЦИЯ            
token = 'TOKEN'
authorize = vk_api.VkApi(token = token) 
longpoll = VkLongPoll(authorize)

# \\\ОСНОВНАЯ ЛОГИКА
for event in longpoll.listen():  
    if event.type == VkEventType.MESSAGE_NEW and event.to_me:
        reseived_message = event.text
        sender = event.user_id
        if reseived_message == "Создать заявку":
            i=2
            write_message(sender, "Как тебя зовут?")
            reg()
        elif reseived_message == event.text:
            i=1
            write_message(sender, "Напиши 'Создать заявку'")

Comments

[edward_freedom]

я на путоне не пишу, но зачем столько

for event in longpoll.listen():

[Георгий Салтыков]

edward_freedom, чтобы сообщение от пользователя было получено именно на данном этапе, тоесть если нам нужно его имя, то бот примет ответ и запишет в переменную для имени а не отправит ответ "Напиши 'Создать заявку'". В общем алгоритм будет двигаться по 1 ветке (как это понял я)

[WolfInChains]

Георгий Салтыков, а если боту напишет сразу 2 пользователя? Я бы создал базу данных и записывал туда состояние каждого пользователя, а именно на каком он этапе, при получении сообщения проверял бы этап и в зависимости от него уже отправлял следующий вопрос. Так и кода в разы меньше, и он будет более читаем.

Answer 1

[soremix]

Хотелость бы реальных советов и адекватной критики

Зачем столько longpoll.listen ?
Среди диалога с одним пользователем, напишите от имени другого в ЛС группы.

Глобальные переменные, тем более в многопользовательском приложении, плохо

Сначала reseived_message = event.text, потом elif reseived_message == event.text. Зачем?

Клавиатуры растянуты имхо

Нейминг переменных странный. name_user != "Назад" Почему имя проверяется на "Назад"?. Нет таких имен, либо переменная может содержать что-то другое, вместо имени.

Бесполезная проверка, условие же будет выполняться всегда.

text_user = event.text
if text_user == event.text

Используйте для работы базы данных. На сколько я помню, в vk_api нет поддежки FSM, так что руками.

Comments

[Hemul GM]

Мог просто написать "Говнокод"

Answer 2

[Hemul GM]

У вас проблемы с логикой. Всё в мгновение пойдёт по вульве как только напишет второй человек. А если будет больше, то каша будет просто у всех.

Для таких чат-ботов используется база данных. В которой хранится состояние каждого пользователя и для каждого отдельного пользователя будет по-разному происходить проверка условий в зависимости от последнего состояния (текущего шага).

Как только кто-то пишет. Идете в базу, забираете состояние текущего пользователя (текущий шаг) и реагируйте на его сообщение относительно текущего шага. И так для каждого сообщения.

Критика:
У вас ужасный, не понятный код. Куча вложенных условий. Магические строки, магические числа. Переменная city сравнивается со строкой "Назад". Как это логически вообще обосновать?
Глобальные переменные, ошибочное использование принципа работы бота: longpoll.listen() - это метод для одного глобального цикла

Comments

[Георгий Салтыков]

Попробовал использовать базу данных, примерно так получилось. Для своего удобства убрал клавиатуры и возможность отменить заполнение заявки, а так же вернуться к предыдущему вопросу

import vk_api
import sqlite3
from vk_api.longpoll import VkLongPoll, VkEventType
from vk_api.utils import get_random_id
  
token = 'TOKEN'
authorize = vk_api.VkApi(token = token) 
longpoll = VkLongPoll(authorize)

db = sqlite3.connect('action.db')
sql = db.cursor()
sql.execute("""CREATE TABLE IF NOT EXISTS users (
    userId BIGINT, 
    flag TEXT, 
    name TEXT, 
    age TEXT, 
    city TEXT, 
    text TEXT
)""")
db.commit()

userFlag = '0'

def write_message (id, message):
    authorize.method('messages.send', {'user_id': id, 'message':  message, 'random_id': get_random_id()})

for event in longpoll.listen():  
    if event.type == VkEventType.MESSAGE_NEW and event.to_me:
        reseived_message = event.text
        id = event.user_id
        sql.execute(f"SELECT userId FROM users WHERE userId = '{id}'")
        if sql.fetchone() is None:
            sql.execute("INSERT INTO users VALUES (?, ?, ?, ?, ?, ?)", (id, "newUser", "0", "0", "0", "0"))
            write_message(id, "Привет, напиши Создать заявку")
        else:
            userFlag = sql.execute(f"SELECT flag FROM users WHERE userId = '{id}'").fetchone()[0]
            if userFlag == "newUser" and reseived_message == "Создать заявку":
                sql.execute(f"UPDATE users SET flag = 'getName' WHERE userId = {id}")
                db.commit()
                write_message(id, "Напиши свое имя")
            elif userFlag == "getName":
                sql.execute(f"UPDATE users SET flag = 'getAge' WHERE userId = {id}")
                db.commit()
                write_message(id, "Укажи возраст")
            elif userFlag == "getAge":
                sql.execute(f"UPDATE users SET flag = 'getCity' WHERE userId = {id}")
                db.commit()
                write_message(id, "С какого ты города?")
            elif userFlag == "getCity":
                sql.execute(f"UPDATE users SET flag = 'getText' WHERE userId = {id}")
                db.commit()
                write_message(id, "Укажи текст заявки")
            elif userFlag == "getText":
                write_message(id, "Ваша заявка отправлена модератору")

[Hemul GM]

Георгий Салтыков, это уже больше похоже на нормальный код. Но, запросы лучше вынести в отдельные методы.
И метод самой обработки тоже стоит вынести в отдельный метод и переменную userFlag сделать локальной этого метода

[Георгий Салтыков]

Hemul GM, Тоесть создать клас и туда засунуть методы с запросами, я правильно понял?

[Hemul GM]

Георгий Салтыков, не обязательно класс

[Георгий Салтыков]

Hemul GM, Если я правильно понял, то получится примерно так + еще добавил возможность вернуться к предыдущему вопросу и возможность отмены заполнения заявки

import vk_api
import sqlite3
from vk_api.longpoll import VkLongPoll, VkEventType
from vk_api.utils import get_random_id
  
token = 'TOKEN'
authorize = vk_api.VkApi(token = token) 
longpoll = VkLongPoll(authorize)

db = sqlite3.connect('action.db')
sql = db.cursor()
sql.execute("""CREATE TABLE IF NOT EXISTS users (
    userId BIGINT, 
    flag TEXT, 
    name TEXT, 
    age TEXT, 
    city TEXT, 
    text TEXT
)""")
db.commit()

def write_message (id, message):
    authorize.method('messages.send', {'user_id': id, 'message':  message, 'random_id': get_random_id()})

def zapros(flag):
    sql.execute(f"UPDATE users SET flag = '{flag}' WHERE userId = {id}")
    db.commit()
    
def reg_user():
    userFlag = sql.execute(f"SELECT flag FROM users WHERE userId = '{id}'").fetchone()[0]
    if userFlag == "newUser" and reseived_message == "Создать заявку":
        zapros('getName')
        write_message(id, "Напиши свое имя")
    elif userFlag == "getName" and reseived_message == "Назад":
        zapros('newUser')
        write_message(id, "Вы вышли с регистрации")
    elif userFlag == "getName" and reseived_message == "Отменить":
        zapros('newUser')
        write_message(id, "Вы вышли с регистрации")
            
    elif userFlag == "getName":
        zapros('getAge')
        write_message(id, "Укажи возраст")
    elif userFlag == "getAge" and reseived_message == "Назад":
        zapros('getName')
        write_message(id, "Напиши свое имя")
    elif userFlag == "getAge" and reseived_message == "Отменить":
        zapros('newUser')
        write_message(id, "Вы вышли с регистрации")

    elif userFlag == "getAge":
        zapros('getCity')
        write_message(id, "С какого ты города?")
    elif userFlag == "getCity" and reseived_message == "Назад":
        zapros('getAge')
        write_message(id, "Укажи возраст")
    elif userFlag == "getCity" and reseived_message == "Отменить":
        zapros('newUser')
        write_message(id, "Вы вышли с регистрации")

    elif userFlag == "getCity":
        zapros('getText')
        write_message(id, "Укажи текст заявки")
    elif userFlag == "getText" and reseived_message == "Назад":
        zapros('getAge')
        write_message(id, "Укажи возраст")
    elif userFlag == "getText" and reseived_message == "Отменить":
        zapros('newUser')
        write_message(id, "Вы вышли с регистрации")

    elif userFlag == "getText":
        write_message(id, "Ваша заявка отправлена модератору")

for event in longpoll.listen():  
    if event.type == VkEventType.MESSAGE_NEW and event.to_me:
        reseived_message = event.text
        id = event.user_id
        sql.execute(f"SELECT userId FROM users WHERE userId = '{id}'")
        if sql.fetchone() is None:
            sql.execute("INSERT INTO users VALUES (?, ?, ?, ?, ?, ?)", (id, "newUser", "0", "0", "0", "0"))
            write_message(id, "Привет, напиши Создать заявку")
        else:
            reg_user()