From 674cdf907ee60a564fc2da1c7dbd096b5f43ba26 Mon Sep 17 00:00:00 2001 From: dan63047 Date: Sun, 12 Apr 2020 00:50:14 +0300 Subject: [PATCH] OOP & !echo (-start.py) --- dan63047bot.py | 63 ++++++++++++++++++++++++++++++++++++++++++++------ start.py | 43 ---------------------------------- 2 files changed, 56 insertions(+), 50 deletions(-) delete mode 100644 start.py diff --git a/dan63047bot.py b/dan63047bot.py index 97de170..d8735b8 100644 --- a/dan63047bot.py +++ b/dan63047bot.py @@ -8,20 +8,38 @@ import random import json import asyncio import wikipediaapi as wiki -from config import vk, owm, vk_mda +from config import vk, owm, vk_mda, group_id from vk_api.longpoll import VkLongPoll, VkEventType +from vk_api.bot_longpoll import VkBotLongPoll, VkBotEventType -bot_logger = logging.getLogger("dan63047bot") +root_logger= logging.getLogger() +root_logger.setLevel(logging.INFO) +handler = logging.FileHandler('bot.log', 'w', 'utf-8') +handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')) +root_logger.addHandler(handler) + +longpoll = VkBotLongPoll(vk, group_id) + +class MyVkLongPoll(VkBotLongPoll): + def listen(self): + while True: + try: + for event in self.check(): + yield event + except Exception as e: + logging.warning("Беды с ВК: "+str(e)) + time.sleep(15) + continue class VkBot: def __init__(self, peer_id, user_id): - bot_logger.info("Создан объект бота!") + logging.info(f"Создан объект бота! id{peer_id}") self._USER_ID = user_id self._CHAT_ID = peer_id - self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki", "!byn"] + self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki", "!byn", "!echo"] def get_weather(self, place): logger = logging.getLogger("dan63047bot.get_weather") @@ -130,7 +148,7 @@ class VkBot: respond['text'] = "Ваш ид: "+str(self._USER_ID) elif message[0] == self._COMMANDS[2] or message[0] == self._COMMANDS[5]: - respond['text'] = "Я бот, призванный доставлять неудобства.
Команды:
!my_id - сообщит ваш id в ВК
!user_id *id* - сообщит информацию о этом пользователе
!group_id *id* - сообщит информацию о этой группе
!image - отправляет рандомную картинку из альбома
!weather *город* - отправляет текущую погоду в городе (данные из OpenWeather API)
!wiki *запрос* - отправляет информацию об этом из Wikipedia
!byn - отправляет текущий курс валют, полученный из API НБ РБ
!h, !help - справка
Дата последнего обновления: 10.04.2020 (попытка использовать async/await)
Проект бота на GitHub: https://github.com/dan63047/dan63047pythonbot" + respond['text'] = "Я бот, призванный доставлять неудобства.
Команды:
!my_id - сообщит ваш id в ВК
!user_id *id* - сообщит информацию о этом пользователе
!group_id *id* - сообщит информацию о этой группе
!image - отправляет рандомную картинку из альбома
!weather *город* - отправляет текущую погоду в городе (данные из OpenWeather API)
!wiki *запрос* - отправляет информацию об этом из Wikipedia
!byn - отправляет текущий курс валют, полученный из API НБ РБ
!echo - бот отправляет вам всё, что вы ему пишите(работа команды не проверена)
!h, !help - справка
Дата последнего обновления: 12.04.2020
Проект бота на GitHub: https://github.com/dan63047/dan63047pythonbot" elif message[0] == self._COMMANDS[3]: try: @@ -158,5 +176,36 @@ class VkBot: elif message[0] == self._COMMANDS[8]: respond['text'] = await self.exchange_rates() - message = vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': respond['text'], 'random_id': time.time(), 'attachment': respond['attachment']}) - bot_logger.info(f'Ответ бота: {respond}') + + elif message[0] == self._COMMANDS[9]: + vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': "echo on", 'random_id': time.time()}) + for event in MyVkLongPoll.listen(longpoll): + if event.type == VkBotEventType.MESSAGE_NEW: + if event.message.text == "!echo off": + respond['text'] = "echo off" + break + else: + vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': event.message.text, 'random_id': time.time()}) + + if respond['text'] or respond['attachment']: + message = vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': respond['text'], 'random_id': time.time(), 'attachment': respond['attachment']}) + logging.info(f'Ответ бота в чат id{self._CHAT_ID}: {respond}') + +async def main(): + bot = {} + for event in MyVkLongPoll.listen(longpoll): + try: + if event.type == VkBotEventType.MESSAGE_NEW: + logging.info(f'Новое сообщение в чате id{event.message.peer_id}: {event.message.text}') + if event.message.peer_id in bot: + await bot[event.message.peer_id].new_message(event.message.text) + else: + bot[event.message.peer_id] = VkBot(event.message.peer_id, event.message.from_id) + await bot[event.message.peer_id].new_message(event.message.text) + except Exception as kek: + logging.warning("Беды с ботом: "+str(kek)) + continue + +logging.info("Бот начал работу") + +asyncio.run(main()) diff --git a/start.py b/start.py deleted file mode 100644 index 8486c11..0000000 --- a/start.py +++ /dev/null @@ -1,43 +0,0 @@ -import vk_api -import time -import logging -import requests -import asyncio -from config import vk, group_id -from dan63047bot import VkBot -from vk_api.bot_longpoll import VkBotLongPoll, VkBotEventType - -root_logger= logging.getLogger() -root_logger.setLevel(logging.INFO) -handler = logging.FileHandler('bot.log', 'w', 'utf-8') -handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')) -root_logger.addHandler(handler) - -longpoll = VkBotLongPoll(vk, group_id) - -class MyVkLongPoll(VkBotLongPoll): - def listen(self): - while True: - try: - for event in self.check(): - yield event - except Exception as e: - logging.warning("Беды с ВК: "+str(e)) - continue - -async def main(): - for event in MyVkLongPoll.listen(longpoll): - try: - if event.type == VkBotEventType.MESSAGE_NEW: - logging.info(f'Новое сообщение в чате id{event.message.peer_id}: {event.message.text}') - bot = VkBot(event.message.peer_id, event.message.from_id) - await bot.new_message(event.message.text) - except Exception as kek: - logging.warning("Беды с ботом: "+str(kek)) - continue - -logging.info("Бот начал работу") - -asyncio.run(main()) - - \ No newline at end of file