dan63
/
pythonbot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added Wiki API

This commit is contained in:
dan63047 2020-04-06 13:26:50 +03:00
parent 5dbc155322
commit 338b344079
1 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
dan63047bot.py
View File

@ -4,6 +4,7 @@ import requests
import logging import logging
import pyowm import pyowm
import random import random
import wikipediaapi as wiki
from config import vk, owm, vk_mda from config import vk, owm, vk_mda
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from vk_api.longpoll import VkLongPoll, VkEventType from vk_api.longpoll import VkLongPoll, VkEventType
@ -18,7 +19,7 @@ class VkBot:
self._USER_ID = user_id self._USER_ID = user_id
self._CHAT_ID = peer_id self._CHAT_ID = peer_id
self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather"] self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki"]
@staticmethod @staticmethod
def _clean_all_tag_from_str(string_line): def _clean_all_tag_from_str(string_line):
@ -110,13 +111,22 @@ class VkBot:
def random_image(self): def random_image(self):
logger = logging.getLogger("dan63047bot.random_image") logger = logging.getLogger("dan63047bot.random_image")
random_images_query = vk_mda.method('photos.get', {'owner_id': -190322075, 'album_id': 269199619, 'count': 1000}) random_images_query = vk_mda.method('photos.get', {'owner_id': -190322075, 'album_id': 269199619, 'count': 1000})
logger.info("Результат метода photos.get: "+str(random_images_query)) logger.info("Результат метода photos.get: Получено "+str(random_images_query['count'])+" фото")
random_number = random.randrange(random_images_query['count']-1) 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+"<br>"+page.summary
else:
answer = "Такой статьи не существует"
return answer
def new_message(self, message): def new_message(self, message):
respond = {'attachment': None, 'text': None} respond = {'attachment': None, 'text': None}
message = message.split(' ') message = message.split(' ', 1)
if message[0] == self._COMMANDS[0]: if message[0] == self._COMMANDS[0]:
respond['attachment'] = self.random_image() respond['attachment'] = self.random_image()
@ -134,16 +144,20 @@ class VkBot:
elif message[0] == self._COMMANDS[4]: elif message[0] == self._COMMANDS[4]:
try: try:
respond['text'] = respond['text'] = self.get_info_group(message[1]) respond['text'] = self.get_info_group(message[1])
except IndexError: except IndexError:
respond['text'] = "Отсуствует аргумент" respond['text'] = "Отсуствует аргумент"
elif message[0] == self._COMMANDS[6]: elif message[0] == self._COMMANDS[6]:
try: try:
respond['text'] = respond['text'] = self.get_weather(message[1]) 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: except IndexError:
respond['text'] = "Отсуствует аргумент" respond['text'] = "Отсуствует аргумент"
except AttributeError as lol:
respond['text'] = "Пайтон в ахуе: "+str(lol)
return respond return respond