diff --git a/README.md b/README.md
index c319d89..db51d2b 100644
--- a/README.md
+++ b/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*** — бот отправляет информацию о обьектах бота в памяти. Доступно только вам
diff --git a/dan63047bot.py b/dan63047bot.py
index 6e9543d..3f69efa 100644
--- a/dan63047bot.py
+++ b/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}
-
- if midnight:
- self._MIDNIGHT_EVENT = True
- else:
- self._MIDNIGHT_EVENT = False
+ self._MIDNIGHT_EVENT = midnight
+ self._NEW_POST = new_post
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'] = "Я бот, призванный доставлять неудобства.
Команды:
!my_id - сообщит ваш id в ВК
!user_id *id* - сообщит информацию о этом пользователе
!group_id *id* - сообщит информацию о этой группе
!image - отправляет рандомную картинку из альбома
!weather *город* - отправляет текущую погоду в городе (данные из OpenWeather API)
!wiki *запрос* - отправляет информацию об этом из Wikipedia
!byn - отправляет текущий курс валют, полученный из API НБ РБ
!echo - бот отправляет вам всё, что вы ему пишите
!game *камень/ножницы/бумага/статистика* - бот будет играть с вами в \"Камень, ножницы, бумага\" и записывать статистику
!midnight - бот будет уведомлять вас о 00:00 по Москве. Отправьте ещё раз, чтобы бот больше вас не уведомлял
!reminder *set/list/delete* - напоминалка. set устанавливает напоминание, delete удаляет, list выдаёт список ваших напоминаний
!h, !help - справка
Дата последнего обновления: 21.05.2020 (!reminder)
Проект бота на GitHub: https://github.com/dan63047/dan63047pythonbot"
+ 'text'] = "Я бот, призванный доставлять неудобства.
Команды:
!my_id - сообщит ваш id в ВК
!user_id *id* - сообщит информацию о этом пользователе
!group_id *id* - сообщит информацию о этой группе
!image - отправляет рандомную картинку из альбома
!weather *город* - отправляет текущую погоду в городе (данные из OpenWeather API)
!wiki *запрос* - отправляет информацию об этом из Wikipedia
!byn - отправляет текущий курс валют, полученный из API НБ РБ
!echo - бот отправляет вам всё, что вы ему пишите
!game *камень/ножницы/бумага/статистика* - бот будет играть с вами в \"Камень, ножницы, бумага\" и записывать статистику
!midnight - бот будет уведомлять вас о 00:00 по Москве. Отправьте ещё раз, чтобы бот больше вас не уведомлял
!reminder *set/list/delete* - напоминалка. set устанавливает напоминание, delete удаляет, list выдаёт список ваших напоминаний
!subscribe - бот будет уведомлять вас новых постах в группе. Отправьте ещё раз, чтобы бот больше вас не уведомлял
!h, !help - справка
Дата последнего обновления: 23.05.2020 (!subscribe)
Проект бота на 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"]
@@ -297,6 +296,19 @@ class VkBot:
self.change_await("reminder delete")
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:
- atch = "nothing"
- log(False, f'Новое сообщение: peer_id: {event.message.peer_id}, user_id: {event.message.from_id}, text: "{event.message.text}", attachments: {atch}')
+ log_msg += atch
+ log(False, log_msg)
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: