bot_of_pray/python-bot/utils/admin_utils.py

49 lines
1.5 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import os
from utils.logger import logger
#from utils.user_data import get_user_info
#from utils.user_data import get_user_data
from utils.message_sender import send_message
def has_records(data,typer):
for user_id, records in data.items():
if typer in records:
return True
return False
def load_admins(): # utils/admin_utils.py
admin_file= os.getenv('TELEGRAM_ADMINS')
try:
with open(admin_file, 'r') as file:
return [line.strip() for line in file]
except FileNotFoundError:
logger.error("Файл с ID администраторов не найден")
return []
ADMINS = load_admins()
def is_admin(user_id: str) -> bool: # utils/admin_utils.py
admins = load_admins()
return str(user_id) in admins
def register_handlers(app):
app.add_handler(CommandHandler("admin", admin))
# другие базовые команды
def count_memorial_notes(data):
total_count = 0
# Проходим по всем пользователям
for user_id, user_data in data.items():
# Считаем количество записок "О здравии"
if "О здравии" in user_data:
total_count += 1
# Считаем количество записок "За упокой"
if "За упокой" in user_data:
total_count += 1
return total_count
# Пример использования:
# result = count_memorial_notes(USER_DATA)
# print(f"Общее количество записок: {result}")