new command !subscribe

This commit is contained in:
dan63047 2020-05-23 10:48:05 +03:00
parent 6f44f7179c
commit d7a14b7d3d
2 changed files with 54 additions and 19 deletions

View File

@ -36,6 +36,7 @@ owm = pyowm.OWM('OpenWeather_api_key', language='ru') # Ключ OpenWeather API
* **!game *камень/ножницы/бумага/статистика*** — бот играет с вами в "Камень, ножницы, бумага" и ведет статискику игр, которую записывает в файл json * **!game *камень/ножницы/бумага/статистика*** — бот играет с вами в "Камень, ножницы, бумага" и ведет статискику игр, которую записывает в файл json
* **!midnight** — бот будет уведомлять о каждом миднайте по Московскому времени. Введите ещё раз, чтобы отменить это * **!midnight** — бот будет уведомлять о каждом миднайте по Московскому времени. Введите ещё раз, чтобы отменить это
* **!reminder *set/list/delete*** — напоминалка. С агрументом *set* бот запускает установку напоминания, с аргументом *list* отправляет вам все ваши запланированные напоминания, с аргументом *delete* запускает удаление напоминания * **!reminder *set/list/delete*** — напоминалка. С агрументом *set* бот запускает установку напоминания, с аргументом *list* отправляет вам все ваши запланированные напоминания, с аргументом *delete* запускает удаление напоминания
* **!subscribe** — бот будет уведомлять вас о каждом новом посте. Введите ещё раз, чтобы отменить это
* **!debug** — бот отправляет информацию о своём состоянии * **!debug** — бот отправляет информацию о своём состоянии
* **!debug *log*** — бот отправляет последние 10 строк из своего лога. Доступно только вам * **!debug *log*** — бот отправляет последние 10 строк из своего лога. Доступно только вам
* **!debug *bots*** — бот отправляет информацию о обьектах бота в памяти. Доступно только вам * **!debug *bots*** — бот отправляет информацию о обьектах бота в памяти. Доступно только вам

View File

@ -44,7 +44,7 @@ def load_users():
users[i] = users_not_json[i] users[i] = users_not_json[i]
users_file.close() users_file.close()
for i in users: for i in users:
bot[int(i)] = VkBot(i, users[i]["midnight"], users[i]['await'], int(users[i]['access'])) bot[int(i)] = VkBot(i, users[i]["midnight"], users[i]['await'], int(users[i]['access']), users[i]['new_post'])
except Exception as lol: except Exception as lol:
log(True, f"Проблема с загрузкой users.json: {str(lol)}") log(True, f"Проблема с загрузкой users.json: {str(lol)}")
@ -93,18 +93,15 @@ def get_weather(place):
class VkBot: class VkBot:
def __init__(self, peer_id, midnight=False, awaiting=None, access=1): def __init__(self, peer_id, midnight=False, awaiting=None, access=1, new_post=False):
log(False, f"Создан объект бота! id{peer_id}") log(False, f"Создан объект бота! id{peer_id}")
self._CHAT_ID = peer_id self._CHAT_ID = peer_id
self._AWAITING_INPUT_MODE = awaiting self._AWAITING_INPUT_MODE = awaiting
self._ACCESS_LEVEL = access self._ACCESS_LEVEL = access
self._SET_UP_REMINDER = {"task": None, "time": None} self._SET_UP_REMINDER = {"task": None, "time": None}
self._MIDNIGHT_EVENT = midnight
if midnight: self._NEW_POST = new_post
self._MIDNIGHT_EVENT = True
else:
self._MIDNIGHT_EVENT = False
if int(self._CHAT_ID) == int(owner_id): if int(self._CHAT_ID) == int(owner_id):
self._OWNER = True self._OWNER = True
@ -112,7 +109,7 @@ class VkBot:
self._OWNER = False self._OWNER = False
self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki", "!byn", self._COMMANDS = ["!image", "!my_id", "!h", "!user_id", "!group_id", "!help", "!weather", "!wiki", "!byn",
"!echo", "!game", "!debug", "!midnight", "!access", "!turnoff", "!reminder"] "!echo", "!game", "!debug", "!midnight", "!access", "!turnoff", "!reminder", "!subscribe"]
def __str__(self): def __str__(self):
return f"peer_id: {str(self._CHAT_ID)}, m: {str(self._MIDNIGHT_EVENT)}, await: {str(self._AWAITING_INPUT_MODE)}, tasks: {len(users[self._CHAT_ID]['tasks'])}" return f"peer_id: {str(self._CHAT_ID)}, m: {str(self._MIDNIGHT_EVENT)}, await: {str(self._AWAITING_INPUT_MODE)}, tasks: {len(users[self._CHAT_ID]['tasks'])}"
@ -120,7 +117,7 @@ class VkBot:
def __del__(self): def __del__(self):
log(False, f"Бот id{str(self._CHAT_ID)} удалён") log(False, f"Бот id{str(self._CHAT_ID)} удалён")
def event(self, event): def event(self, event, something=None):
if event == "midnight" and self._MIDNIGHT_EVENT: if event == "midnight" and self._MIDNIGHT_EVENT:
current_time = datetime.datetime.fromtimestamp(time.time() + 10800) current_time = datetime.datetime.fromtimestamp(time.time() + 10800)
image = None image = None
@ -154,6 +151,10 @@ class VkBot:
self.send(midnight_output, image) self.send(midnight_output, image)
log(False, f"Бот id{self._CHAT_ID} оповестил о миднайте") log(False, f"Бот id{self._CHAT_ID} оповестил о миднайте")
elif event == "post" and self._NEW_POST:
post = f"wall{str(something['from_id'])}_{str(something['id'])}"
self.send(f"Вышел новый пост", post)
log(False, f"Бот id{self._CHAT_ID}: Уведомил о новом посте")
def get_message(self, message, user_id): def get_message(self, message, user_id):
if self._AWAITING_INPUT_MODE: if self._AWAITING_INPUT_MODE:
@ -196,7 +197,7 @@ class VkBot:
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[ 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>!midnight - бот будет уведомлять вас о 00:00 по Москве. Отправьте ещё раз, чтобы бот больше вас не уведомлял<br>!reminder *set/list/delete* - напоминалка. set устанавливает напоминание, delete удаляет, list выдаёт список ваших напоминаний<br>!h, !help - справка<br>Дата последнего обновления: 21.05.2020 (!reminder)<br>Проект бота на GitHub: https://github.com/dan63047/dan63047pythonbot" '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>!midnight - бот будет уведомлять вас о 00:00 по Москве. Отправьте ещё раз, чтобы бот больше вас не уведомлял<br>!reminder *set/list/delete* - напоминалка. set устанавливает напоминание, delete удаляет, list выдаёт список ваших напоминаний<br>!subscribe - бот будет уведомлять вас новых постах в группе. Отправьте ещё раз, чтобы бот больше вас не уведомлял<br>!h, !help - справка<br>Дата последнего обновления: 23.05.2020 (!subscribe)<br>Проект бота на GitHub: https://github.com/dan63047/dan63047pythonbot"
elif message[0] == self._COMMANDS[3]: elif message[0] == self._COMMANDS[3]:
try: try:
@ -249,15 +250,13 @@ class VkBot:
elif message[0] == self._COMMANDS[12]: elif message[0] == self._COMMANDS[12]:
if self._ACCESS_LEVEL or int(user_id) == int(owner_id): if self._ACCESS_LEVEL or int(user_id) == int(owner_id):
if self._MIDNIGHT_EVENT: if self._MIDNIGHT_EVENT:
self._MIDNIGHT_EVENT = False self.change_midnight(False)
self.send("Уведомление о миднайте выключено") self.send("Уведомление о миднайте выключено")
log(False, f"Бот id{self._CHAT_ID}: Юзер отписался от ивента \"Миднайт\"") log(False, f"Бот id{self._CHAT_ID}: Юзер отписался от ивента \"Миднайт\"")
else: else:
self._MIDNIGHT_EVENT = True self.change_midnight(True)
self.send("Бот будет уведомлять вас о каждом миднайте") self.send("Бот будет уведомлять вас о каждом миднайте")
log(False, f"Бот id{self._CHAT_ID}: Юзер подписался на ивент \"Миднайт\"") log(False, f"Бот id{self._CHAT_ID}: Юзер подписался на ивент \"Миднайт\"")
users[self._CHAT_ID]["midnight"] = self._MIDNIGHT_EVENT
update_users_json(users)
else: else:
respond['text'] = errors_array["access"] respond['text'] = errors_array["access"]
@ -298,6 +297,19 @@ class VkBot:
except IndexError: except IndexError:
respond["text"] = errors_array['miss_argument'] respond["text"] = errors_array['miss_argument']
elif message[0] == self._COMMANDS[16]:
if self._ACCESS_LEVEL or int(user_id) == int(owner_id):
if self._NEW_POST:
self.change_new_post(False)
self.send("Уведомление о новом посте выключено")
log(False, f"Бот id{self._CHAT_ID}: Юзер отписался от новых постов")
else:
self.change_new_post(True)
self.send("Бот будет уведомлять вас о каждом новом посте")
log(False, f"Бот id{self._CHAT_ID}: Юзер подписался на уведомления о новых постах")
else:
respond['text'] = errors_array["access"]
if respond['text'] or respond['attachment']: if respond['text'] or respond['attachment']:
self.send(respond['text'], respond['attachment']) self.send(respond['text'], respond['attachment'])
@ -566,6 +578,24 @@ class VkBot:
users[self._CHAT_ID]['access']= self._ACCESS_LEVEL users[self._CHAT_ID]['access']= self._ACCESS_LEVEL
update_users_json(users) update_users_json(users)
def change_new_post(self, new_post):
self._NEW_POST = new_post
try:
users[self._CHAT_ID]['new_post']= self._NEW_POST
except KeyError:
users[self._CHAT_ID].setdefault("new_post", None)
users[self._CHAT_ID]['new_post']= self._NEW_POST
update_users_json(users)
def change_midnight(self, midnight):
self._MIDNIGHT_EVENT = midnight
try:
users[self._CHAT_ID]['midnight']= self._MIDNIGHT_EVENT
except KeyError:
users[self._CHAT_ID].setdefault("midnight", None)
users[self._CHAT_ID]['midnight']= self._MIDNIGHT_EVENT
update_users_json(users)
def send(self, message=None, attachment=None): def send(self, message=None, attachment=None):
message = vk.method('messages.send', message = vk.method('messages.send',
{'peer_id': self._CHAT_ID, 'message': message, 'random_id': time.time(), {'peer_id': self._CHAT_ID, 'message': message, 'random_id': time.time(),
@ -580,26 +610,30 @@ def bots():
for event in MyVkLongPoll.listen(longpoll): for event in MyVkLongPoll.listen(longpoll):
try: try:
if event.type == VkBotEventType.MESSAGE_NEW: if event.type == VkBotEventType.MESSAGE_NEW:
log_msg = f'Новое сообщение: peer_id: {event.message.peer_id}, user_id: {event.message.from_id}, text: "{event.message.text}"'
if event.message.attachments: if event.message.attachments:
atch = '' atch = ', attachments: '
for i in event.message.attachments: for i in event.message.attachments:
if i['type'] == "sticker": if i['type'] == "sticker":
atch += f"sticker_id{i[i['type']]['sticker_id']}" atch += f"sticker_id{i[i['type']]['sticker_id']}"
elif i['type'] == "wall":
atch += i['type'] + str(i[i['type']]['from_id']) + "_" + str(i[i['type']]['id']) + " "
else: else:
atch += i['type'] + str(i[i['type']]['owner_id']) + "_" + str(i[i['type']]['id']) + " " atch += i['type'] + str(i[i['type']]['owner_id']) + "_" + str(i[i['type']]['id']) + " "
else: log_msg += atch
atch = "nothing" log(False, log_msg)
log(False, f'Новое сообщение: peer_id: {event.message.peer_id}, user_id: {event.message.from_id}, text: "{event.message.text}", attachments: {atch}')
debug_array['messages_get'] += 1 debug_array['messages_get'] += 1
if int(event.message.peer_id) in bot: if int(event.message.peer_id) in bot:
bot[event.message.peer_id].get_message(event.message.text, event.message.from_id) bot[event.message.peer_id].get_message(event.message.text, event.message.from_id)
else: else:
bot[event.message.peer_id] = VkBot(event.message.peer_id) bot[event.message.peer_id] = VkBot(event.message.peer_id)
users[event.message.peer_id] = {"midnight": False, "tasks": {}, "await": None, "access": 1} users[event.message.peer_id] = {"midnight": False, "tasks": {}, "await": None, "access": 1, "new_post": False}
update_users_json(users) update_users_json(users)
bot[event.message.peer_id].get_message(event.message.text, event.message.from_id) bot[event.message.peer_id].get_message(event.message.text, event.message.from_id)
elif event.type == VkBotEventType.WALL_POST_NEW: elif event.type == VkBotEventType.WALL_POST_NEW:
log(False, f"На стене группы опубликован новый пост id{event.object.id}: {event.object.text}") log(False, f"На стене группы опубликован новый пост id{event.object.id}: {event.object.text}")
for i in users:
bot[int(i)].event("post", event.object)
else: else:
log(False, str(event)) log(False, str(event))
except Exception as kek: except Exception as kek: