221 lines
11 KiB
Python
221 lines
11 KiB
Python
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)
|
||
|
||
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):
|
||
|
||
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"]
|
||
|
||
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 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 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)
|
||
|
||
logger.info("Результат метода API users.get: "+str(user_info))
|
||
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>!h, !help - справка<br>Дата последнего обновления: 12.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'] = self.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} в режиме эхо")
|
||
|
||
|
||
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())
|