Compare commits
2 commits
6388760ebb
...
56fbb3c346
| Author | SHA1 | Date | |
|---|---|---|---|
| 56fbb3c346 | |||
| a41d7ece98 |
1 changed files with 25 additions and 16 deletions
41
bot.py
41
bot.py
|
|
@ -4,6 +4,7 @@ import datetime
|
|||
import time
|
||||
|
||||
from vkbottle.bot import Bot, Message
|
||||
from vkbottle_types.objects import MessagesGetConversationMembers
|
||||
|
||||
import config
|
||||
from config import db_load, db_save
|
||||
|
|
@ -24,20 +25,26 @@ async def get_user_info(user_id):
|
|||
return None
|
||||
|
||||
|
||||
def user_is_admin(user_id: int, chat_members: MessagesGetConversationMembers):
|
||||
for member in chat_members.items:
|
||||
if member.member_id != user_id:
|
||||
continue
|
||||
return member.is_admin
|
||||
return False
|
||||
|
||||
|
||||
@bot.on.chat_message(text="!старт")
|
||||
async def stats_handler(message: Message):
|
||||
chat_id = str(message.peer_id)
|
||||
user_id = str(message.from_id)
|
||||
chat_members = await bot.api.messages.get_conversation_members(peer_id=message.peer_id, extended=False)
|
||||
|
||||
# TODO
|
||||
if user_id != '5326732':
|
||||
if not user_is_admin(message.from_id, chat_members):
|
||||
await message.answer('О таком меня может попросить только администратор беседы.')
|
||||
return
|
||||
|
||||
chat_id = str(message.peer_id)
|
||||
if chat_id not in config.DB['chats']:
|
||||
config.DB['chats'][chat_id] = {'users': {}}
|
||||
|
||||
chat_members = await bot.api.messages.get_conversation_members(peer_id=message.peer_id, extended=False)
|
||||
|
||||
for member in chat_members.items:
|
||||
if member.member_id < 0:
|
||||
continue
|
||||
|
|
@ -176,20 +183,22 @@ async def counters_reset_task():
|
|||
target_date = now.date()
|
||||
target_datetime = datetime.datetime.combine(target_date, target_time)
|
||||
|
||||
await wait_until(target_datetime)
|
||||
while True:
|
||||
await wait_until(target_datetime)
|
||||
|
||||
print('Resetting daily counters...')
|
||||
for chat_id in config.DB['chats']:
|
||||
for user_id in config.DB['chats'][chat_id]['users']:
|
||||
config.DB['chats'][chat_id]['users'][user_id]['messages_today'] = 0
|
||||
|
||||
if target_datetime.day == 1:
|
||||
print('Resetting monthly counters...')
|
||||
print('Resetting daily counters...')
|
||||
for chat_id in config.DB['chats']:
|
||||
for user_id in config.DB['chats'][chat_id]['users']:
|
||||
config.DB['chats'][chat_id]['users'][user_id]['messages_month'] = 0
|
||||
config.DB['chats'][chat_id]['users'][user_id]['messages_today'] = 0
|
||||
|
||||
db_save()
|
||||
if target_datetime.day == 1:
|
||||
print('Resetting monthly counters...')
|
||||
for chat_id in config.DB['chats']:
|
||||
for user_id in config.DB['chats'][chat_id]['users']:
|
||||
config.DB['chats'][chat_id]['users'][user_id]['messages_month'] = 0
|
||||
|
||||
db_save()
|
||||
target_datetime = target_datetime + datetime.timedelta(days=1)
|
||||
|
||||
|
||||
async def startup_task():
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue