This commit is contained in:
dan63047 2020-04-13 21:25:48 +03:00
parent 05306e1bba
commit fd0c168bd3
1 changed files with 147 additions and 47 deletions

View File

@ -20,6 +20,11 @@ root_logger.addHandler(handler)
longpoll = VkBotLongPoll(vk, group_id) longpoll = VkBotLongPoll(vk, group_id)
def toFixed(numObj, digits=0):
return f"{numObj:.{digits}f}"
class MyVkLongPoll(VkBotLongPoll): class MyVkLongPoll(VkBotLongPoll):
def listen(self): def listen(self):
while True: while True:
@ -31,6 +36,20 @@ class MyVkLongPoll(VkBotLongPoll):
time.sleep(15) time.sleep(15)
continue 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: class VkBot:
def __init__(self, peer_id, user_id): def __init__(self, peer_id, user_id):
@ -40,12 +59,14 @@ class VkBot:
self._CHAT_ID = peer_id self._CHAT_ID = peer_id
self._ECHO_MODE = False 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): async def get_message(self, message):
if self._ECHO_MODE: if self._ECHO_MODE:
if message == "!echo off": 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 self._ECHO_MODE = False
logging.info(f"Бот id{self._CHAT_ID} вышел из режима эхо") logging.info(f"Бот id{self._CHAT_ID} вышел из режима эхо")
else: else:
@ -54,16 +75,71 @@ class VkBot:
else: else:
await self.send_message(message) await self.send_message(message)
def get_weather(self, place): def game(self, thing):
logger = logging.getLogger("dan63047bot.get_weather") 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: try:
weather_request = owm.weather_at_place(place) data = json.load(write_file)
except pyowm.exceptions.api_response_error.NotFoundError as i: except Exception:
logger.warning("Ошибка OpenWeather API: "+str(i)) data = {}
return "Такого города нет, либо данных о погоде нет" if str(self._USER_ID) not in data:
weather_answer = weather_request.get_weather() data[str(self._USER_ID)] = {}
logger.info("Результат поиска погоды через OpenWeather API: "+str(weather_answer)) data[str(self._USER_ID)]["games"] = 0
return "В городе "+place+" сейчас "+weather_answer.get_detailed_status()+", "+str(round(weather_answer.get_temperature('celsius')['temp']))+"°C" 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): def get_info_user(self, id):
logger = logging.getLogger("dan63047bot.get_info_user") logger = logging.getLogger("dan63047bot.get_info_user")
@ -123,15 +199,18 @@ class VkBot:
else: else:
description = group_info[0]['description'] 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']) answer = group_info[0]['name'] + "<br>Описание: " + description + "<br>Ид группы: " + str(
group_info[0]['id']) + "<br>Подписчиков: " + str(group_info[0]['members_count'])
return answer return answer
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['count']) + " фото") logger.info("Результат метода photos.get: Получено " + str(random_images_query['count']) + " фото")
random_number = random.randrange(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): def wiki_article(self, search):
w = wiki.Wikipedia('ru') w = wiki.Wikipedia('ru')
@ -144,10 +223,19 @@ class VkBot:
async def exchange_rates(self): async def exchange_rates(self):
try: try:
rates_USD = json.loads(requests.get("https://www.nbrb.by/api/exrates/rates/145?periodicity=0", timeout=10).text) rates_USD = json.loads(
rates_EUR = json.loads(requests.get("https://www.nbrb.by/api/exrates/rates/292?periodicity=0", timeout=10).text) requests.get("https://www.nbrb.by/api/exrates/rates/145?periodicity=0", timeout=10).text)
rates_RUB = json.loads(requests.get("https://www.nbrb.by/api/exrates/rates/298?periodicity=0", timeout=10).text) rates_EUR = json.loads(
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" 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: except Exception as mda:
return "Невозможно получить данные из НБ РБ: " + str(mda) return "Невозможно получить данные из НБ РБ: " + str(mda)
@ -161,7 +249,7 @@ class VkBot:
respond['text'] = "Ваш ид: " + str(self._USER_ID) respond['text'] = "Ваш ид: " + str(self._USER_ID)
elif message[0] == self._COMMANDS[2] or message[0] == self._COMMANDS[5]: 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" 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]: elif message[0] == self._COMMANDS[3]:
try: try:
@ -177,7 +265,7 @@ class VkBot:
elif message[0] == self._COMMANDS[6]: elif message[0] == self._COMMANDS[6]:
try: try:
respond['text'] = self.get_weather(message[1]) respond['text'] = get_weather(message[1])
except IndexError: except IndexError:
respond['text'] = "Отсуствует аргумент" respond['text'] = "Отсуствует аргумент"
@ -191,15 +279,26 @@ class VkBot:
respond['text'] = await self.exchange_rates() respond['text'] = await self.exchange_rates()
elif message[0] == self._COMMANDS[9]: 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 self._ECHO_MODE = True
logging.info(f"Бот id{self._CHAT_ID} в режиме эхо") 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']: 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}') logging.info(f'Ответ бота в чат id{self._CHAT_ID}: {respond}')
async def main(): async def main():
bot = {} bot = {}
for event in MyVkLongPoll.listen(longpoll): for event in MyVkLongPoll.listen(longpoll):
@ -215,6 +314,7 @@ async def main():
logging.warning("Беды с ботом: " + str(kek)) logging.warning("Беды с ботом: " + str(kek))
continue continue
logging.info("Бот начал работу") logging.info("Бот начал работу")
asyncio.run(main()) asyncio.run(main())