diff --git a/dan63047bot.py b/dan63047bot.py index 440d2e5..fa3994f 100644 --- a/dan63047bot.py +++ b/dan63047bot.py @@ -12,7 +12,7 @@ from config import vk, owm, vk_mda, group_id from vk_api.longpoll import VkLongPoll, VkEventType from vk_api.bot_longpoll import VkBotLongPoll, VkBotEventType -root_logger= logging.getLogger() +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')) @@ -20,17 +20,36 @@ root_logger.addHandler(handler) longpoll = VkBotLongPoll(vk, group_id) + +def toFixed(numObj, digits=0): + return f"{numObj:.{digits}f}" + + class MyVkLongPoll(VkBotLongPoll): def listen(self): while True: - try: + try: for event in self.check(): yield event except Exception as e: - logging.warning("Беды с ВК: "+str(e)) + logging.warning("Беды с ВК: " + str(e)) time.sleep(15) continue + +def get_weather(place): + logger = logging.getLogger("dan63047bot.get_weather") + try: + weather_request = owm.weather_at_place(place) + except pyowm.exceptions.api_response_error.NotFoundError as i: + logger.warning("Ошибка OpenWeather API: " + str(i)) + return "Такого города нет, либо данных о погоде нет" + weather_answer = weather_request.get_weather() + logger.info("Результат поиска погоды через OpenWeather API: " + str(weather_answer)) + return "В городе " + place + " сейчас " + weather_answer.get_detailed_status() + ", " + str( + round(weather_answer.get_temperature('celsius')['temp'])) + "°C" + + class VkBot: def __init__(self, peer_id, user_id): @@ -40,12 +59,14 @@ class VkBot: self._CHAT_ID = peer_id self._ECHO_MODE = False - self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki", "!byn", "!echo"] + self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki", "!byn", + "!echo", "!game"] async def get_message(self, message): if self._ECHO_MODE: if message == "!echo off": - vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': "Режим эхо выключен", 'random_id': time.time()}) + vk.method('messages.send', + {'peer_id': self._CHAT_ID, 'message': "Режим эхо выключен", 'random_id': time.time()}) self._ECHO_MODE = False logging.info(f"Бот id{self._CHAT_ID} вышел из режима эхо") else: @@ -54,26 +75,81 @@ class VkBot: else: await self.send_message(message) - def get_weather(self, place): - logger = logging.getLogger("dan63047bot.get_weather") - try: - weather_request = owm.weather_at_place(place) - except pyowm.exceptions.api_response_error.NotFoundError as i: - logger.warning("Ошибка OpenWeather API: "+str(i)) - return "Такого города нет, либо данных о погоде нет" - weather_answer = weather_request.get_weather() - logger.info("Результат поиска погоды через OpenWeather API: "+str(weather_answer)) - return "В городе "+place+" сейчас "+weather_answer.get_detailed_status()+", "+str(round(weather_answer.get_temperature('celsius')['temp']))+"°C" + def game(self, thing): + if thing == "статистика": + with open("data_file.json", "r") as read_file: + data = json.load(read_file) + if str(self._USER_ID) in data: + winrate = (data[str(self._USER_ID)]['wins']/data[str(self._USER_ID)]['games'])*100 + return f"Камень, ножницы, бумага
Сыграно игр: {data[str(self._USER_ID)]['games']}
Из них:
•Побед: {data[str(self._USER_ID)]['wins']}
•Поражений: {data[str(self._USER_ID)]['defeats']}
•Ничей: {data[str(self._USER_ID)]['draws']}
Процент побед: {toFixed(winrate, 2)}%" + else: + return "Похоже, вы ещё никогда не играли в Камень, ножницы, бумага" + elif thing == "камень" or thing == "ножницы" or thing == "бумага": + things = ["камень", "ножницы", "бумага"] + bot_thing = random.choice(things) + if thing == "камень" and bot_thing == "ножницы": + result = 2 + elif thing == "ножницы" and bot_thing == "бумага": + result = 2 + elif thing == "бумага" and bot_thing == "камень": + result = 2 + elif thing == "ножницы" and bot_thing == "камень": + result = 1 + elif thing == "бумага" and bot_thing == "ножницы": + result = 1 + elif thing == "камень" and bot_thing == "бумага": + result = 2 + elif thing == "камень" and bot_thing == "камень": + result = 0 + elif thing == "ножницы" and bot_thing == "ножницы": + result = 0 + elif thing == "бумага" and bot_thing == "бумага": + result = 0 + + if result == 2: + response = f"Камень, ножницы, бумага
{thing} vs. {bot_thing}
Вы выиграли!" + elif result == 1: + response = f"Камень, ножницы, бумага
{thing} vs. {bot_thing}
Вы проиграли!" + elif result == 0: + response = f"Камень, ножницы, бумага
{thing} vs. {bot_thing}
Ничья!" + + with open("data_file.json", 'r') as write_file: + try: + data = json.load(write_file) + except Exception: + data = {} + if str(self._USER_ID) not in data: + data[str(self._USER_ID)] = {} + data[str(self._USER_ID)]["games"] = 0 + data[str(self._USER_ID)]["wins"] = 0 + data[str(self._USER_ID)]["defeats"] = 0 + data[str(self._USER_ID)]["draws"] = 0 + + if result == 2: + data[str(self._USER_ID)]["games"] += 1 + data[str(self._USER_ID)]["wins"] += 1 + elif result == 1: + data[str(self._USER_ID)]["games"] += 1 + data[str(self._USER_ID)]["defeats"] += 1 + elif result == 0: + data[str(self._USER_ID)]["games"] += 1 + data[str(self._USER_ID)]["draws"] += 1 + + with open("data_file.json", "w") as write_file: + json.dump(data, write_file) + return response + else: + return "Неверный аргумент
Использование команды:
!game *камень/ножницы/бумага/статистика*" def get_info_user(self, id): logger = logging.getLogger("dan63047bot.get_info_user") try: user_info = vk.method('users.get', {'user_ids': id, 'fields': 'verified,last_seen,sex'}) except vk_api.exceptions.ApiError as lol: - logger.warning("Ошибка метода users.get: "+str(lol)) - return "Пользователь не найден
"+str(lol) - - logger.info("Результат метода API users.get: "+str(user_info)) + logger.warning("Ошибка метода users.get: " + str(lol)) + return "Пользователь не найден
" + str(lol) + + logger.info("Результат метода API users.get: " + str(user_info)) if user_info[0]['is_closed']: is_closed = "Да" else: @@ -105,7 +181,7 @@ class VkBot: time = datetime.datetime.fromtimestamp(user_info[0]['last_seen']['time']) - answer = user_info[0]['first_name']+" "+user_info[0]['last_name']+"
Его ид: "+str(user_info[0]['id'])+"
Профиль закрыт: "+is_closed+"
Пол: "+sex+"
Последний онлайн: "+time.strftime('%d.%m.%Y в %H:%M:%S')+" ("+platform+")" + answer = user_info[0]['first_name'] + " " + user_info[0]['last_name'] + "
Его ид: " + str(user_info[0]['id']) + "
Профиль закрыт: " + is_closed + "
Пол: " + sex + "
Последний онлайн: " + time.strftime('%d.%m.%Y в %H:%M:%S') + " (" + platform + ")" return answer @@ -114,61 +190,73 @@ class VkBot: try: group_info = vk.method('groups.getById', {'group_id': id, 'fields': 'description,members_count'}) except vk_api.exceptions.ApiError as lol: - logger.warning("Ошибка метода groups.getById: "+str(lol)) - return "Группа не найдена
"+str(lol) - - logger.info("Результат метода API groups.getById: "+str(group_info)) + logger.warning("Ошибка метода groups.getById: " + str(lol)) + return "Группа не найдена
" + str(lol) + + logger.info("Результат метода API groups.getById: " + str(group_info)) if group_info[0]['description'] == "": description = "Отсутствует" else: description = group_info[0]['description'] - answer = group_info[0]['name']+"
Описание: "+description+"
Ид группы: "+str(group_info[0]['id'])+"
Подписчиков: "+str(group_info[0]['members_count']) + answer = group_info[0]['name'] + "
Описание: " + description + "
Ид группы: " + str( + group_info[0]['id']) + "
Подписчиков: " + str(group_info[0]['members_count']) return answer def random_image(self): logger = logging.getLogger("dan63047bot.random_image") - random_images_query = vk_mda.method('photos.get', {'owner_id': -190322075, 'album_id': 269199619, 'count': 1000}) - logger.info("Результат метода photos.get: Получено "+str(random_images_query['count'])+" фото") + random_images_query = vk_mda.method('photos.get', + {'owner_id': -190322075, 'album_id': 269199619, 'count': 1000}) + logger.info("Результат метода photos.get: Получено " + str(random_images_query['count']) + " фото") random_number = random.randrange(random_images_query['count']) - return "photo"+str(random_images_query['items'][random_number]['owner_id'])+"_"+str(random_images_query['items'][random_number]['id']) + return "photo" + str(random_images_query['items'][random_number]['owner_id']) + "_" + str( + random_images_query['items'][random_number]['id']) def wiki_article(self, search): w = wiki.Wikipedia('ru') page = w.page(search) if page.exists(): - answer = page.title+"
"+page.summary + answer = page.title + "
" + page.summary else: answer = "Такой статьи не существует" return answer async def exchange_rates(self): try: - rates_USD = json.loads(requests.get("https://www.nbrb.by/api/exrates/rates/145?periodicity=0", timeout=10).text) - rates_EUR = json.loads(requests.get("https://www.nbrb.by/api/exrates/rates/292?periodicity=0", timeout=10).text) - rates_RUB = json.loads(requests.get("https://www.nbrb.by/api/exrates/rates/298?periodicity=0", timeout=10).text) - return "Текущий курс валют по данным НБ РБ:
"+rates_USD['Cur_Name']+": "+str(rates_USD['Cur_Scale'])+" "+rates_USD['Cur_Abbreviation']+" = "+str(rates_USD['Cur_OfficialRate'])+" BYN
"+rates_EUR['Cur_Name']+": "+str(rates_EUR['Cur_Scale'])+" "+rates_EUR['Cur_Abbreviation']+" = "+str(rates_EUR['Cur_OfficialRate'])+" BYN
"+"Российский рубль"+": "+str(rates_RUB['Cur_Scale'])+" "+rates_RUB['Cur_Abbreviation']+" = "+str(rates_RUB['Cur_OfficialRate'])+" BYN" + rates_USD = json.loads( + requests.get("https://www.nbrb.by/api/exrates/rates/145?periodicity=0", timeout=10).text) + rates_EUR = json.loads( + requests.get("https://www.nbrb.by/api/exrates/rates/292?periodicity=0", timeout=10).text) + rates_RUB = json.loads( + requests.get("https://www.nbrb.by/api/exrates/rates/298?periodicity=0", timeout=10).text) + return "Текущий курс валют по данным НБ РБ:
" + rates_USD['Cur_Name'] + ": " + str( + rates_USD['Cur_Scale']) + " " + rates_USD['Cur_Abbreviation'] + " = " + str( + rates_USD['Cur_OfficialRate']) + " BYN
" + rates_EUR['Cur_Name'] + ": " + str( + rates_EUR['Cur_Scale']) + " " + rates_EUR['Cur_Abbreviation'] + " = " + str( + rates_EUR['Cur_OfficialRate']) + " BYN
" + "Российский рубль" + ": " + str( + rates_RUB['Cur_Scale']) + " " + rates_RUB['Cur_Abbreviation'] + " = " + str( + rates_RUB['Cur_OfficialRate']) + " BYN" except Exception as mda: - return "Невозможно получить данные из НБ РБ: "+str(mda) + return "Невозможно получить данные из НБ РБ: " + str(mda) async def send_message(self, message): respond = {'attachment': None, 'text': None} message = message.split(' ', 1) if message[0] == self._COMMANDS[0]: - respond['attachment'] = self.random_image() + respond['attachment'] = self.random_image() elif message[0] == self._COMMANDS[1]: - respond['text'] = "Ваш ид: "+str(self._USER_ID) + 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 НБ РБ
!echo - бот отправляет вам всё, что вы ему пишите
!h, !help - справка
Дата последнего обновления: 12.04.2020
Проект бота на 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 - бот отправляет вам всё, что вы ему пишите
!game *камень/ножницы/бумага/статистика* - бот будет играть с вами в \"Камень, ножницы, бумага\" и записывать статистику
!h, !help - справка
Дата последнего обновления: 13.04.2020
Проект бота на GitHub: https://github.com/dan63047/dan63047pythonbot" elif message[0] == self._COMMANDS[3]: try: respond['text'] = self.get_info_user(message[1]) except IndexError: respond['text'] = "Отсуствует аргумент" - + elif message[0] == self._COMMANDS[4]: try: respond['text'] = self.get_info_group(message[1]) @@ -177,7 +265,7 @@ class VkBot: elif message[0] == self._COMMANDS[6]: try: - respond['text'] = self.get_weather(message[1]) + respond['text'] = get_weather(message[1]) except IndexError: respond['text'] = "Отсуствует аргумент" @@ -186,20 +274,31 @@ class VkBot: respond['text'] = self.wiki_article(message[1]) except IndexError: respond['text'] = "Отсуствует аргумент" - - elif message[0] == self._COMMANDS[8]: + + elif message[0] == self._COMMANDS[8]: respond['text'] = await self.exchange_rates() elif message[0] == self._COMMANDS[9]: - vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': "Теперь бот работает в режиме эхо. Чтобы это выключить, введить \"!echo off\"", 'random_id': time.time()}) + vk.method('messages.send', {'peer_id': self._CHAT_ID, + 'message': "Теперь бот работает в режиме эхо. Чтобы это выключить, введить \"!echo off\"", + 'random_id': time.time()}) self._ECHO_MODE = True logging.info(f"Бот id{self._CHAT_ID} в режиме эхо") - - + + elif message[0] == self._COMMANDS[10]: + try: + message[1] = message[1].lower() + respond['text'] = self.game(message[1]) + except IndexError: + respond['text'] = "Отсуствует аргумент" + 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']}) + 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): @@ -212,8 +311,9 @@ async def main(): bot[event.message.peer_id] = VkBot(event.message.peer_id, event.message.from_id) await bot[event.message.peer_id].get_message(event.message.text) except Exception as kek: - logging.warning("Беды с ботом: "+str(kek)) - continue + logging.warning("Беды с ботом: " + str(kek)) + continue + logging.info("Бот начал работу")