new command !subscribe

README.md

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

dan63047bot.py

@@ -44,7 +44,7 @@ def load_users():
                 users[i] = users_not_json[i]
             users_file.close()
         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:
         log(True, f"Проблема с загрузкой users.json: {str(lol)}")
 
@@ -93,18 +93,15 @@ def get_weather(place):
 
 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}")
         self._CHAT_ID = peer_id
         self._AWAITING_INPUT_MODE = awaiting
         self._ACCESS_LEVEL = access
         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):
             self._OWNER = True
@@ -112,7 +109,7 @@ class VkBot:
             self._OWNER = False
 
         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):
         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):
         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:
             current_time = datetime.datetime.fromtimestamp(time.time() + 10800)
             image = None
@@ -154,6 +151,10 @@ class VkBot:
 
             self.send(midnight_output, image)
             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):
         if self._AWAITING_INPUT_MODE:
@@ -196,7 +197,7 @@ class VkBot:
 
             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>!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]:
                 try:
@@ -249,15 +250,13 @@ class VkBot:
             elif message[0] == self._COMMANDS[12]:
                 if self._ACCESS_LEVEL or int(user_id) == int(owner_id):
                     if self._MIDNIGHT_EVENT:
-                        self._MIDNIGHT_EVENT = False
+                        self.change_midnight(False)
                         self.send("Уведомление о миднайте выключено")
                         log(False, f"Бот id{self._CHAT_ID}: Юзер отписался от ивента \"Миднайт\"")
                     else:
-                        self._MIDNIGHT_EVENT = True
+                        self.change_midnight(True)
                         self.send("Бот будет уведомлять вас о каждом миднайте")
                         log(False, f"Бот id{self._CHAT_ID}: Юзер подписался на ивент \"Миднайт\"")
-                    users[self._CHAT_ID]["midnight"] = self._MIDNIGHT_EVENT
-                    update_users_json(users)
                 else:
                     respond['text'] = errors_array["access"]
 
@@ -298,6 +297,19 @@ class VkBot:
                 except IndexError:
                     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']:
                 self.send(respond['text'], respond['attachment'])
 
@@ -566,6 +578,24 @@ class VkBot:
             users[self._CHAT_ID]['access']= self._ACCESS_LEVEL
         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):
         message = vk.method('messages.send',
                             {'peer_id': self._CHAT_ID, 'message': message, 'random_id': time.time(),
@@ -580,26 +610,30 @@ def bots():
     for event in MyVkLongPoll.listen(longpoll):
         try:
             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:
-                    atch = ''
+                    atch = ', attachments: '
                     for i in event.message.attachments:
                         if i['type'] == "sticker":
                             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:
                             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
                 if int(event.message.peer_id) in bot:
                     bot[event.message.peer_id].get_message(event.message.text, event.message.from_id)
                 else:
                     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)
                     bot[event.message.peer_id].get_message(event.message.text, event.message.from_id)
             elif event.type == VkBotEventType.WALL_POST_NEW:
                 log(False, f"На стене группы опубликован новый пост id{event.object.id}: {event.object.text}")
+                for i in users:
+                    bot[int(i)].event("post", event.object)
             else:
                 log(False, str(event))
         except Exception as kek: