Ejemplo n.º 1
0
def show_basket(vk, cur, con, user_id):  # отправка корзины
    basket_li = cur.execute(
        f'''SELECT basket FROM Users WHERE user_id = {user_id}''').fetchall(
        )[0][0].split(',')
    basket = Basket()
    basket = None
    basket = Basket()
    for dish in basket_li:
        if dish:
            dish = cur.execute(
                f"SELECT * FROM Menu WHERE name = '{dish}'").fetchall()[0]
            basket.append(Dish(dish))
    text = str(basket)
    if text:
        with open(path_to('data/keyboards', 'basket.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(user_id=user_id,
                             message=text,
                             random_id=random.randint(0, 2**64),
                             keyboard=kb.read())
        update_last_act(user_id, cur, con, 'show_basket')
    else:
        with open(path_to('data/keyboards', 'empty_basket.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(
                user_id=user_id,
                message='Ваша корзина пуста. Наполните ее и возвращайтесь!',
                random_id=random.randint(0, 2**64),
                keyboard=kb.read())
        update_last_act(user_id, cur, con, 'show_empty_basket')
Ejemplo n.º 2
0
def logging(*args, file=None, session_time=None):
    if file:
        with open(path_to('data/logs', file), 'w',
                  encoding='UTF-8') as logfile:
            print(args, file=logfile)
    else:
        with open(path_to(
                'data/logs',
                f'log_{session_time.strftime("%d.%m.%Y_%H-%M-%S-%f")}.txt'),
                  'wt',
                  encoding='UTF-8') as logfile:
            print(args, file=logfile)
Ejemplo n.º 3
0
def edit_basket(vk, cur, con, user_id, action='ask', elem=None):
    basket_li = cur.execute(
        f'''SELECT basket FROM Users WHERE user_id = {user_id}''').fetchall(
        )[0][0].split(',')
    basket = Basket()
    update_basket(user_id, cur, con, vk, dish=0)
    if action == 'ask':
        with open(path_to('data/keyboards', 'edit_basket.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(user_id=user_id,
                             message='Какую пиццу удалим? Ответь цифрой!',
                             random_id=random.randint(0, 2**64),
                             keyboard=kb.read())
        update_last_act(user_id, cur, con, 'ask_for_del')
    else:
        try:
            if int(elem) in range(len(basket)):
                basket.delete(int(elem) + 1)
                update_basket(user_id, cur, con, vk, new=basket.names())
                basket = None
                basket = Basket()
                with open(path_to('data/keyboards', 'edit_basket.json'),
                          'r',
                          encoding='UTF-8') as kb:
                    vk.messages.send(
                        user_id=user_id,
                        message=
                        'Выполнено! Если хочешь удалить ещё, пиши цифру!\n\n' +
                        str(basket),
                        random_id=random.randint(0, 2**64),
                        keyboard=kb.read())
                update_last_act(user_id, cur, con, 'edit_basket')
            else:
                with open(path_to('data/keyboards', 'edit_basket.json'),
                          'r',
                          encoding='UTF-8') as kb:
                    vk.messages.send(
                        user_id=user_id,
                        message='Такой пиццы в корзине нет! Ответь цифрой!',
                        random_id=random.randint(0, 2**64),
                        keyboard=kb.read())
        except ValueError:
            with open(path_to('data/keyboards', 'edit_basket.json'),
                      'r',
                      encoding='UTF-8') as kb:
                vk.messages.send(
                    user_id=user_id,
                    message='Такой пиццы в корзине нет! Ответь цифрой!',
                    random_id=random.randint(0, 2**64),
                    keyboard=kb.read())
Ejemplo n.º 4
0
def get_address(
        vk,
        event,
        user_id,
        cur,
        con,
        necessarily=True):  # получение адреса из сообщения пользователя
    address = None
    try:
        lat = event.obj.message['geo']['coordinates']['latitude']
        long = event.obj.message['geo']['coordinates']['longitude']
        address = (lat, long)
    except KeyError:
        pass
    if address:
        return conf_address(vk, address, user_id, cur, con)
    else:
        if necessarily:
            update_last_act(user_id, cur, con, 'ask_address')
            with open(path_to('data/keyboards', 'location.json'),
                      'r',
                      encoding='UTF-8') as kb:
                vk.messages.send(user_id=user_id,
                                 message=f"Попробуйте еще раз!",
                                 random_id=random.randint(0, 2**64),
                                 keyboard=kb.read())
        return None
Ejemplo n.º 5
0
def registration(vk, cur, con, user_id):
    with open(path_to('data/keyboards', 'registration.json'),
              'r',
              encoding='UTF-8') as kb:
        vk.messages.send(user_id=user_id,
                         message="Выберите способ оплаты:",
                         random_id=random.randint(0, 2**64),
                         keyboard=kb.read())
        update_last_act(user_id, cur, con, 'registration')
Ejemplo n.º 6
0
def show_current_dish(vk, cur, con, user_id):  # отправка текущего блюда
    max_num = cur.execute('SELECT num FROM Menu').fetchall()[-1][0]
    current = cur.execute(
        f'SELECT current_dish FROM Users WHERE user_id = {user_id}').fetchall(
        )[0][0]
    dish_li = cur.execute(
        f'SELECT * FROM Menu WHERE num = {int(current) + 1}').fetchall()[0]
    dish = Dish(dish_li)
    last_act = cur.execute(
        f'SELECT last_act FROM Users WHERE user_id = {user_id}').fetchall(
        )[0][0]
    if last_act != 'show_last_dish':
        kb = {
            "inline":
            True,
            "buttons": [[{
                "action": {
                    "type": "text",
                    "payload": "{\"button\": \"1\"}",
                    "label": f"{dish.name}"
                },
                "color": "positive"
            }, {
                "action": {
                    "type": "text",
                    "payload": "{\"button\": \"2\"}",
                    "label": "Следующее блюдо!"
                },
                "color": "primary"
            }, {
                "action": {
                    "type": "text",
                    "payload": "{\"button\": \"2\"}",
                    "label": "Предыдущее блюдо!"
                },
                "color": "secondary"
            }]]
        }
        kb_json = json.dumps(kb)
        vk.messages.send(user_id=user_id,
                         message=str(dish),
                         random_id=random.randint(0, 2**64),
                         keyboard=kb_json,
                         attachment=f'photo{upload_image(vk, dish.url_pic)}')
        if dish.num == max_num:
            update_last_act(user_id, cur, con, 'show_last_dish')
        else:
            update_last_act(user_id, cur, con, 'show_dish')
    else:
        with open(path_to('data/keyboards', 'last_page.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(
                user_id=user_id,
                message="Меню кончилось. Посмотрим еще раз или идем в корзину?",
                random_id=random.randint(0, 2**64),
                keyboard=kb.read())
Ejemplo n.º 7
0
def send_path(path: int):
    with open(path_to('front', f'path_{path}.geojson'),
              encoding='utf-8') as file:
        file = file.read()
        print(type(file))
        file = json.loads(file)
        file = parse(file)
        print(file)
        return file
Ejemplo n.º 8
0
def load_users():
    """
    Attempt to load users from the data file.
    If one doesn't exist, just return a new mapping with a single admin user.
    """
    try:
        with open(path_to(config["USERS_FILE"]), "rb") as f:
            return pickle.load(f)
    except FileNotFoundError:
        return {"admin": {"pword": config["DEFAULT_ADMIN_PASS"],
                          "is_op": True}}
Ejemplo n.º 9
0
def conf_address(vk, coords, user_id, cur, con):  # подтверждение адреса
    address = get_address_from_coords(coords)
    if address.split()[0] != 'Ошибка':
        with open(path_to('data/keyboards', 'confirmation.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(user_id=user_id,
                             message=f"Ваш адрес — {address}?",
                             random_id=random.randint(0, 2**64),
                             keyboard=kb.read())
            update_last_act(user_id, cur, con, 'ask_conf_address')
            return address
    else:
        with open(path_to('data/keyboards', 'location.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(
                user_id=user_id,
                message=f"Введённый адрес недействителен. Попробуйте еще раз!",
                random_id=random.randint(0, 2**64),
                keyboard=kb.read())
Ejemplo n.º 10
0
def main():
    global SESSION_TIME
    #  подключение к ВКонтакте
    try:
        vk_session = vk_api.VkApi(
            token=
            '41e1b47bb6d5d9b90ba89da796c0a85c33ebe0c6b75219b2d3f7e2c8492293f9d54a641647d56388c0f7a'
        )
        longpoll = VkBotLongPoll(vk_session, '195073403')
    except BaseException as e:
        return e
    #  подключение к БД
    try:
        con = sqlite3.connect(path_to('data/databases', 'data.db'))
        cur = con.cursor()
    except BaseException as e:
        return e
    for event in longpoll.listen():
        if event.type == VkBotEventType.GROUP_JOIN:
            vk = vk_session.get_api()
            #  логирование
            logging(f'{event.obj.user_id} вступил в группу!',
                    session_time=SESSION_TIME)
            with open(path_to('data/keyboards', 'follow.json'),
                      'r',
                      encoding='UTF-8') as kb:
                vk.messages.send(
                    user_id=event.obj.user_id,
                    message=
                    "Привет, хочешь заказать вкуснейшую пиццу к себе домой?",
                    random_id=random.randint(0, 2**64),
                    keyboard=kb.read())
            #  добавление пользователя в БД
            cur.execute(
                f'''INSERT INTO Users(user_id, status, address, last_act, menu_page) 
                            VALUES({event.obj.user_id}, 'user', 'not_stated', 'new_user', 0)'''
            )
        if event.type == VkBotEventType.MESSAGE_NEW:
            user_id = event.obj.message['from_id']
            text = event.obj.message['text']
            #  логирование
            try:
                logging('Уведомление от',
                        dt.datetime.now(),
                        session_time=SESSION_TIME)
                logging('Новое сообщение от',
                        user_id,
                        'с текстом',
                        text,
                        session_time=SESSION_TIME)
            except BaseException as e:
                print(e)
            try:
                result = cur.execute(
                    f'''SELECT user_id, status, address, last_act FROM Users
                                         WHERE user_id = {user_id}'''
                ).fetchall()
                if not result:
                    raise WrongID
            except WrongID:
                cur.execute(
                    f'''INSERT INTO Users(user_id, status, address, last_act, menu_page, current_dish, pay_way) 
                                VALUES({user_id}, 'user', 'not_stated', 'new_user', 0, 0, 'not_stated')'''
                )
                con.commit()
                result = cur.execute(
                    f'''SELECT user_id, status, address, last_act FROM Users
                                                        WHERE user_id = {user_id}'''
                ).fetchall()
            user_id, status, address, last_act = result[0]
            vk = vk_session.get_api()
            if text == '/stop':  # сброс пользователя в БД
                update_carousel_page(user_id, cur, con, 0)
                update_current_dish(user_id, cur, con, 0)
                update_basket(user_id, cur, con, vk, dish=0)
                update_last_act(user_id, cur, con, 'new_user')
                vk.messages.send(user_id=user_id,
                                 message="Ваш заказ сброшен",
                                 random_id=random.randint(0, 2**64))
            else:
                if status == 'user':
                    if address == 'not_stated':
                        if last_act == 'new_user':
                            with open(path_to('data/keyboards',
                                              'location.json'),
                                      'r',
                                      encoding='UTF-8') as kb:
                                vk.messages.send(
                                    user_id=user_id,
                                    message=
                                    "На какой адрес нужно заказать пиццу?\n" +
                                    "В любой момент напиши /stop, чтобы сбросить свой заказ",
                                    random_id=random.randint(0, 2**64),
                                    keyboard=kb.read())
                            update_last_act(user_id, cur, con, 'ask_address')
                    if last_act == 'ask_address':
                        update_address(
                            user_id, cur, con,
                            get_address(vk, event, user_id, cur, con))
                    if address != 'not_stated' and last_act == 'new_user':
                        conf_address(vk, address, user_id, cur, con)
                    if last_act == 'ask_conf_address':
                        if 'не' in text.lower():
                            update_last_act(user_id, cur, con, 'new_user')
                            get_address(vk, event, user_id, cur, con)
                        else:
                            if address != 'not_stated':
                                with open(path_to('data/keyboards',
                                                  'confirmation.json'),
                                          'r',
                                          encoding='UTF-8') as kb:
                                    vk.messages.send(
                                        user_id=user_id,
                                        message=f"Отлично, переходим к меню?",
                                        random_id=random.randint(0, 2**64),
                                        keyboard=kb.read())
                                    update_last_act(user_id, cur, con,
                                                    'ask_choosing')
                            else:
                                update_last_act(user_id, cur, con, 'new_user')
                    if last_act == 'ask_choosing':
                        if 'не' in text.lower():
                            with open(path_to('data/keyboards',
                                              'confirmation.json'),
                                      'r',
                                      encoding='UTF-8') as kb:
                                vk.messages.send(user_id=user_id,
                                                 message=f"Подумай ещё раз!",
                                                 random_id=random.randint(
                                                     0, 2**64),
                                                 keyboard=kb.read())
                        else:
                            if 'carousel' in event.client_info.keys(
                            ):  # проверяет, поддерживает ли устройство пользователя карусели
                                show_current_carousel(vk, cur, con, user_id)
                            else:
                                show_current_dish(vk, cur, con, user_id)
                    if last_act == 'show_carousel':
                        if text == 'Следующая страница!':
                            update_carousel_page(user_id, cur, con, 1)
                        elif text == 'Предыдущая страница!':
                            update_carousel_page(user_id, cur, con, -1)
                        else:
                            result = cur.execute(
                                f"SELECT * FROM Menu WHERE name = '{text.rstrip()}'"
                            ).fetchall()
                            if result:
                                update_basket(
                                    user_id,
                                    cur,
                                    con,
                                    vk,
                                    result[0],
                                )
                            else:
                                vk.messages.send(
                                    user_id=user_id,
                                    message=
                                    "Такого блюда в меню нет, попробуй ещё раз!",
                                    random_id=random.randint(0, 2**64))
                        show_current_carousel(vk, cur, con, user_id)
                    if last_act == 'show_last_carousel':
                        if text == 'Назад к меню':
                            update_carousel_page(user_id, cur, con, 0)
                            show_current_carousel(vk, cur, con, user_id)
                        elif text == 'Идём в корзину!':
                            update_last_act(user_id, cur, con, 'go_to_basket')
                            show_basket(vk, cur, con, user_id)
                        else:
                            vk.messages.send(
                                user_id=user_id,
                                message="Я тебя не понял, попробуй ещё раз!",
                                random_id=random.randint(0, 2**64))
                            show_current_carousel(vk, cur, con, user_id)
                    if last_act == 'show_dish':
                        if text == 'Следующее блюдо!':
                            update_current_dish(user_id, cur, con, 1)
                            show_current_dish(vk, cur, con, user_id)
                        elif text == 'Предыдущее блюдо!':
                            update_current_dish(user_id, cur, con, -1)
                            show_current_dish(vk, cur, con, user_id)
                        else:
                            result = cur.execute(
                                f"SELECT * FROM Menu WHERE name = '{text}'"
                            ).fetchall()
                            if result:
                                update_basket(
                                    user_id,
                                    cur,
                                    con,
                                    vk,
                                    result[0],
                                )
                            else:
                                vk.messages.send(
                                    user_id=user_id,
                                    message=
                                    "Такого блюда в меню нет, попробуй ещё раз!",
                                    random_id=random.randint(0, 2**64))
                            show_current_dish(vk, cur, con, user_id)
                    if last_act == 'show_last_dish':
                        if text == 'Назад к меню':
                            update_current_dish(user_id, cur, con, 0)
                            show_current_dish(vk, cur, con, user_id)
                        elif text == 'Идём в корзину!':
                            update_last_act(user_id, cur, con, 'go_to_basket')
                            show_basket(vk, cur, con, user_id)
                        else:
                            vk.messages.send(
                                user_id=user_id,
                                message="Я тебя не понял, попробуй ещё раз!",
                                random_id=random.randint(0, 2**64))
                            show_current_dish(vk, cur, con, user_id)
                    if last_act == 'show_basket':
                        if text == 'Оформляем!':
                            registration(vk, cur, con, user_id)
                        elif text == 'Давай кое-что удалим':
                            update_last_act(user_id, cur, con, 'edit_basket')
                            edit_basket(vk, cur, con, user_id)
                    if last_act == 'registration':
                        if text == 'Наличными курьеру' or text == 'Картой курьеру':
                            update_pay_way(user_id, cur, con, text)
                            finish(vk, cur, con, user_id)
                        else:
                            vk.messages.send(
                                user_id=user_id,
                                message=
                                "Что-то ты намудрил, отвечай лучше кнопками!",
                                random_id=random.randint(0, 2**64))
                            registration(vk, cur, con, user_id)
                    if last_act == 'edit_basket':
                        if text == 'Назад к оформлению':
                            registration(vk, cur, con, user_id)
                        else:
                            edit_basket(vk,
                                        cur,
                                        con,
                                        user_id,
                                        action='del',
                                        elem=text)
                    if last_act == 'ask_for_del':
                        edit_basket(vk,
                                    cur,
                                    con,
                                    user_id,
                                    action='del',
                                    elem=text)
                    if text == 'Назад к оформлению':
                        registration(vk, cur, con, user_id)
                    if last_act == 'show_empty_basket':
                        if text == 'К меню!':
                            if 'carousel' in event.client_info.keys():
                                show_current_carousel(vk, cur, con, user_id)
                            else:
                                show_current_dish(vk, cur, con, user_id)
Ejemplo n.º 11
0
def show_current_carousel(vk, cur, con,
                          user_id):  # отправка карусели пользователю
    max_num = cur.execute('SELECT num FROM Menu').fetchall()[-1][0]
    page_num = cur.execute(
        f'SELECT menu_page FROM Users WHERE user_id = {user_id}').fetchall(
        )[0][0]
    if page_num * 10 - 10 < max_num:
        page_content = cur.execute(f'''SELECT * FROM Menu 
                                        WHERE num BETWEEN ({page_num} * 10 + 1) AND ({page_num} * 10 + 10)'''
                                   ).fetchall()
        carousel_dict = {'type': 'carousel', 'elements': []}

        for dish in page_content:  # формирование карусели
            element_dict = {
                'title':
                dish[1],
                'description':
                dish[3],
                'photo_id':
                upload_image(vk, dish[4]),
                'action': {
                    'type': 'open_photo'
                },
                'buttons': [{
                    'action': {
                        'type': 'text',
                        'label': f'{dish[1]}'
                    },
                    'color': 'positive'
                }, {
                    'action': {
                        'type': 'text',
                        'label': 'Следующая страница!'
                    },
                    'color': 'primary'
                }, {
                    'action': {
                        'type': 'text',
                        'label': 'Предыдущая страница!'
                    },
                    'color': 'secondary'
                }]
            }
            carousel_dict['elements'].append(element_dict)
        # преобразование словаря в json-файл
        carousel_json = json.dumps(carousel_dict)
        vk.messages.send(
            user_id=user_id,
            message=f"Страница {page_num + 1}." +
            " Чтобы добавить пиццу в свою корзину, нажми на кнопку с её названием",
            random_id=random.randint(0, 2**64),
            template=carousel_json)
        update_last_act(user_id, cur, con, 'show_carousel')
    else:
        with open(path_to('data/keyboards', 'last_page.json'),
                  'r',
                  encoding='UTF-8') as kb:
            vk.messages.send(
                user_id=user_id,
                message=
                f"Меню кончилось. Посмотрим еще раз или идем в корзину?",
                random_id=random.randint(0, 2**64),
                keyboard=kb.read())
        update_last_act(user_id, cur, con, 'show_last_carousel')
Ejemplo n.º 12
0
def upload_image(vk, name):  # загрузка фото
    photo = vk_api.VkUpload(vk).photo_messages(
        path_to('data/dish_photos', name))
    return f'{photo[0]["owner_id"]}_{photo[0]["id"]}'
Ejemplo n.º 13
0
setup(name='TwistChat',
      version='0.1.1',
      description='Chatroom server written using Twisted',
      license='The Unlicense',
      author='Bede Kelly',
      author_email='*****@*****.**',
      maintainer='Bede Kelly',
      maintainer_email='*****@*****.**',
      entry_points={'console_scripts': "twistchat = twistchat.server:main"},
      keywords=['irc', 'chatroom', 'server', 'webchat'],
      classifiers=[
          'Development Status :: 4 - Beta',
          'Environment :: Console',
          'Operating System :: POSIX',
          'Programming Language :: Python',
          'Topic :: Communications',
          'Topic :: Communications :: Chat :: Internet Relay Chat',
          ],
      data_files=[(path_to("~/.twistchat"), ("etc/twistchat.yml.example",))],
      install_requires=["twisted", "pyyaml"],
      url='https://bede.club/tag/chatroom',
      packages=['twistchat'],
     )

print("*"*80)
print("\n\nGreat, you've got TwistChat installed! Your config file should go\n"
      "in ~/.twistchat/twistchat.yml. An example config file is provided for\n"
      "you at ~/.twistchat/twistchat.yml.example.")
print("*"*80)
Ejemplo n.º 14
0
def save_users(users):
    """
    Save the user-password mapping to the local folder.
    """
    with open(path_to(config["USERS_FILE"]), "wb") as users_file:
        pickle.dump(users, users_file)
Ejemplo n.º 15
0
def index():
    with open(path_to('front', 'index.html'), encoding='utf-8') as html:
        with open(path_to('front', 'styles.css'), encoding='utf-8') as css:
            return Template(html.read()).render({'style': css.read()})
Ejemplo n.º 16
0
setup(
    name='TwistChat',
    version='0.1.1',
    description='Chatroom server written using Twisted',
    license='The Unlicense',
    author='Bede Kelly',
    author_email='*****@*****.**',
    maintainer='Bede Kelly',
    maintainer_email='*****@*****.**',
    entry_points={'console_scripts': "twistchat = twistchat.server:main"},
    keywords=['irc', 'chatroom', 'server', 'webchat'],
    classifiers=[
        'Development Status :: 4 - Beta',
        'Environment :: Console',
        'Operating System :: POSIX',
        'Programming Language :: Python',
        'Topic :: Communications',
        'Topic :: Communications :: Chat :: Internet Relay Chat',
    ],
    data_files=[(path_to("~/.twistchat"), ("etc/twistchat.yml.example", ))],
    install_requires=["twisted", "pyyaml"],
    url='https://bede.club/tag/chatroom',
    packages=['twistchat'],
)

print("*" * 80)
print("\n\nGreat, you've got TwistChat installed! Your config file should go\n"
      "in ~/.twistchat/twistchat.yml. An example config file is provided for\n"
      "you at ~/.twistchat/twistchat.yml.example.")
print("*" * 80)