pythonbot/dan63047bot.py

import vk_api
import datetime
import time
import requests
import logging
import pyowm
import random
import json
import asyncio
import wikipediaapi as wiki
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.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)


def toFixed(numObj, digits=0):
    return f"{numObj:.{digits}f}"


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


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):

        logging.info(f"Создан объект бота! id{peer_id}")
        self._USER_ID = user_id
        self._CHAT_ID = peer_id
        self._ECHO_MODE = False

        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()})
                self._ECHO_MODE = False
                logging.info(f"Бот id{self._CHAT_ID} вышел из режима эхо")
            else:
                vk.method('messages.send', {'peer_id': self._CHAT_ID, 'message': message, 'random_id': time.time()})
                logging.info(f"Эхо-бот id{self._CHAT_ID}: {message}")
        else:
            await self.send_message(message)

    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"Камень, ножницы, бумага<br>Сыграно игр: {data[str(self._USER_ID)]['games']}<br>Из них:<br>•Побед: {data[str(self._USER_ID)]['wins']}<br>•Поражений: {data[str(self._USER_ID)]['defeats']}<br>•Ничей: {data[str(self._USER_ID)]['draws']}<br>Процент побед: {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"Камень, ножницы, бумага<br>{thing} vs. {bot_thing}<br>Вы выиграли!"
            elif result == 1:
                response = f"Камень, ножницы, бумага<br>{thing} vs. {bot_thing}<br>Вы проиграли!"
            elif result == 0:
                response = f"Камень, ножницы, бумага<br>{thing} vs. {bot_thing}<br>Ничья!"

            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 "Неверный аргумент<br>Использование команды:<br>!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 "Пользователь не найден<br>" + str(lol)

        if "deactivated" in user_info[0]:
            if user_info[0]['deactivated'] == 'banned':
                return user_info[0]['first_name'] + " " + user_info[0]['last_name'] + " забанен"
            elif user_info[0]['deactivated'] == 'deleted':
                return "Профиль был удалён"

        if user_info[0]['is_closed']:
            is_closed = "Да"
        else:
            is_closed = "Нет"

        if user_info[0]['sex'] == 1:
            sex = "Женский"
        elif user_info[0]['sex'] == 2:
            sex = "Мужской"
        else:
            sex = "Неизвестно"

        if user_info[0]['last_seen']['platform'] == 1:
            platform = "m.vk.com"
        elif user_info[0]['last_seen']['platform'] == 2:
            platform = "iPhone"
        elif user_info[0]['last_seen']['platform'] == 3:
            platform = "iPad"
        elif user_info[0]['last_seen']['platform'] == 4:
            platform = "Android"
        elif user_info[0]['last_seen']['platform'] == 5:
            platform = "Windows Phone"
        elif user_info[0]['last_seen']['platform'] == 6:
            platform = "Windows 10"
        elif user_info[0]['last_seen']['platform'] == 7:
            platform = "vk.com"
        else:
            platform = "тип платформы неизвестен"

        time = datetime.datetime.fromtimestamp(user_info[0]['last_seen']['time'])

        answer = user_info[0]['first_name'] + " " + user_info[0]['last_name'] + "<br>Его ид: " + str(user_info[0]['id']) + "<br>Профиль закрыт: " + is_closed + "<br>Пол: " + sex + "<br>Последний онлайн: " + time.strftime('%d.%m.%Y в %H:%M:%S') + " (" + platform + ")"

        return answer

    def get_info_group(self, id):
        logger = logging.getLogger("dan63047bot.get_info_group")
        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 "Группа не найдена<br>" + 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'] + "<br>Описание: " + description + "<br>Ид группы: " + str(
            group_info[0]['id']) + "<br>Подписчиков: " + 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_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'])

    def wiki_article(self, search):
        w = wiki.Wikipedia('ru')
        page = w.page(search)
        if page.exists():
            answer = page.title + "<br>" + 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 "Текущий курс валют по данным НБ РБ:<br>" + rates_USD['Cur_Name'] + ": " + str(
                rates_USD['Cur_Scale']) + " " + rates_USD['Cur_Abbreviation'] + " = " + str(
                rates_USD['Cur_OfficialRate']) + " BYN<br>" + rates_EUR['Cur_Name'] + ": " + str(
                rates_EUR['Cur_Scale']) + " " + rates_EUR['Cur_Abbreviation'] + " = " + str(
                rates_EUR['Cur_OfficialRate']) + " BYN<br>" + "Российский рубль" + ": " + str(
                rates_RUB['Cur_Scale']) + " " + rates_RUB['Cur_Abbreviation'] + " = " + str(
                rates_RUB['Cur_OfficialRate']) + " BYN"
        except Exception as 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()

        elif message[0] == self._COMMANDS[1]:
            respond['text'] = "Ваш ид: " + str(self._USER_ID)

        elif message[0] == self._COMMANDS[2] or message[0] == self._COMMANDS[5]:
            respond['text'] = "Я бот, призванный доставлять неудобства. <br>Команды:<br>!my_id - сообщит ваш id в ВК<br>!user_id *id* - сообщит информацию о этом пользователе<br>!group_id *id* - сообщит информацию о этой группе<br>!image - отправляет рандомную картинку из альбома<br>!weather *город* - отправляет текущую погоду в городе (данные из OpenWeather API)<br>!wiki *запрос* - отправляет информацию об этом из Wikipedia<br>!byn - отправляет текущий курс валют, полученный из API НБ РБ<br>!echo - бот отправляет вам всё, что вы ему пишите<br>!game *камень/ножницы/бумага/статистика* - бот будет играть с вами в \"Камень, ножницы, бумага\" и записывать статистику<br>!h, !help - справка<br>Дата последнего обновления: 13.04.2020<br>Проект бота на 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])
            except IndexError:
                respond['text'] = "Отсуствует аргумент"

        elif message[0] == self._COMMANDS[6]:
            try:
                respond['text'] = get_weather(message[1])
            except IndexError:
                respond['text'] = "Отсуствует аргумент"

        elif message[0] == self._COMMANDS[7]:
            try:
                respond['text'] = self.wiki_article(message[1])
            except IndexError:
                respond['text'] = "Отсуствует аргумент"

        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()})
            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']})
            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].get_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].get_message(event.message.text)
        except Exception as kek:
            logging.warning("Беды с ботом: " + str(kek))
            continue


logging.info("Бот начал работу")

asyncio.run(main())