Compare commits
4 Commits
963ce24e4e
..
main
| Author | SHA1 | Date | |
|---|---|---|---|
| c0edc77a11 | |||
| bbd9c839d5 | |||
| 9584a99133 | |||
| 629c2c2739 |
@@ -0,0 +1,10 @@
|
|||||||
|
TELEGRAM_BOT_TOKEN=your_token_here
|
||||||
|
|
||||||
|
# Полное отключение логов и любого хранения (БД, файлы логов, сохранение message_id)
|
||||||
|
# DISABLE_PERSISTENCE=1
|
||||||
|
|
||||||
|
# Или по отдельности:
|
||||||
|
# DISABLE_LOGGING=1 — нет логов в консоль и в файл
|
||||||
|
# DISABLE_STORAGE=1 — нет SQLite, /del не работает, группы /set не сохраняются
|
||||||
|
|
||||||
|
SCHEDULE_DRIVE_FOLDER_ID=1WhUFHGkS4qC_e84KRArF4ooXHJr8mL5T
|
||||||
@@ -12,6 +12,8 @@ logger = getLogger(__name__)
|
|||||||
|
|
||||||
def register_handlers(dp: Dispatcher, state: BotState, bot: Bot) -> int:
|
def register_handlers(dp: Dispatcher, state: BotState, bot: Bot) -> int:
|
||||||
async def init_db():
|
async def init_db():
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
return
|
||||||
async with aiosqlite.connect(Config.DAYS_TO_DB_PATH) as db:
|
async with aiosqlite.connect(Config.DAYS_TO_DB_PATH) as db:
|
||||||
await db.execute("""
|
await db.execute("""
|
||||||
CREATE TABLE IF NOT EXISTS days_to_new_year (
|
CREATE TABLE IF NOT EXISTS days_to_new_year (
|
||||||
@@ -24,6 +26,8 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot) -> int:
|
|||||||
logger.info("База данных инициализирована")
|
logger.info("База данных инициализирована")
|
||||||
|
|
||||||
async def save_days_to_db(user_id: int, days: int):
|
async def save_days_to_db(user_id: int, days: int):
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
return
|
||||||
logger.debug(f"Сохраняем user_id={user_id}, days={days}")
|
logger.debug(f"Сохраняем user_id={user_id}, days={days}")
|
||||||
async with aiosqlite.connect(Config.DAYS_TO_DB_PATH) as db:
|
async with aiosqlite.connect(Config.DAYS_TO_DB_PATH) as db:
|
||||||
await db.execute("""
|
await db.execute("""
|
||||||
@@ -34,6 +38,8 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot) -> int:
|
|||||||
logger.info(f"Запись сохранена: user_id={user_id}, days={days}")
|
logger.info(f"Запись сохранена: user_id={user_id}, days={days}")
|
||||||
|
|
||||||
async def get_last_days(user_id: int) -> int | None:
|
async def get_last_days(user_id: int) -> int | None:
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
return None
|
||||||
async with aiosqlite.connect(Config.DAYS_TO_DB_PATH) as db:
|
async with aiosqlite.connect(Config.DAYS_TO_DB_PATH) as db:
|
||||||
async with db.execute(
|
async with db.execute(
|
||||||
"SELECT days FROM days_to_new_year WHERE user_id = ?", (int(user_id),)
|
"SELECT days FROM days_to_new_year WHERE user_id = ?", (int(user_id),)
|
||||||
@@ -110,8 +116,8 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot) -> int:
|
|||||||
save_message(msg.chat.id, msg.message_id)
|
save_message(msg.chat.id, msg.message_id)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.exception(f"Ошибка при обработке uid={uid}: {e}")
|
logger.exception(f"Ошибка при обработке uid={uid}: {e}")
|
||||||
logger.info("Завершён цикл periodic_task, спим 6 часов")
|
logger.info("Завершён цикл periodic_task, спим 24 часов")
|
||||||
await asyncio.sleep(21600) # каждые 6 часов
|
await asyncio.sleep(86400) # каждые 24 часов
|
||||||
|
|
||||||
asyncio.create_task(periodic_task())
|
asyncio.create_task(periodic_task())
|
||||||
return 0
|
return 0
|
||||||
|
|||||||
@@ -1,11 +1,20 @@
|
|||||||
import os
|
import os
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
|
||||||
|
def _env_bool(name: str, default: bool = False) -> bool:
|
||||||
|
raw = os.getenv(name)
|
||||||
|
if raw is None:
|
||||||
|
return default
|
||||||
|
return raw.strip().lower() in ("1", "true", "yes", "on")
|
||||||
|
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
# Загружаем .env
|
|
||||||
load_dotenv()
|
|
||||||
|
|
||||||
# API
|
# API
|
||||||
API_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
|
API_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
|
||||||
@@ -40,9 +49,23 @@ class Config:
|
|||||||
"SCHEDULE_DRIVE_FOLDER_ID", "1WhUFHGkS4qC_e84KRArF4ooXHJr8mL5T"
|
"SCHEDULE_DRIVE_FOLDER_ID", "1WhUFHGkS4qC_e84KRArF4ooXHJr8mL5T"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Отключение логов и хранения (см. .env.example)
|
||||||
|
# DISABLE_PERSISTENCE=1 — выключает и логи, и все БД/файлы сразу
|
||||||
|
_NO_PERSISTENCE = _env_bool("DISABLE_PERSISTENCE")
|
||||||
|
DISABLE_LOGGING = (
|
||||||
|
_env_bool("DISABLE_LOGGING")
|
||||||
|
if os.getenv("DISABLE_LOGGING") is not None
|
||||||
|
else _NO_PERSISTENCE
|
||||||
|
)
|
||||||
|
DISABLE_STORAGE = (
|
||||||
|
_env_bool("DISABLE_STORAGE")
|
||||||
|
if os.getenv("DISABLE_STORAGE") is not None
|
||||||
|
else _NO_PERSISTENCE
|
||||||
|
)
|
||||||
|
|
||||||
# Пути
|
# Пути
|
||||||
LOG_FILE = "storage/log/bot.log"
|
LOG_FILE = Path("storage/log/bot.log")
|
||||||
DAYS_TO_DB_PATH = "addons/x_days_to/days_to_new_year.db"
|
DAYS_TO_DB_PATH = Path("addons/x_days_to/days_to_new_year.db")
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
+45
-6
@@ -17,8 +17,14 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
|
|||||||
@saving
|
@saving
|
||||||
@admin_required(3)
|
@admin_required(3)
|
||||||
async def send_log(message: Message):
|
async def send_log(message: Message):
|
||||||
|
if Config.DISABLE_LOGGING:
|
||||||
|
await message.answer("📝 Логирование отключено (DISABLE_LOGGING=1).")
|
||||||
|
return
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
await message.answer("📝 Файл логов не ведётся (DISABLE_STORAGE=1).")
|
||||||
|
return
|
||||||
try:
|
try:
|
||||||
log_file = types.FSInputFile(Config.LOG_FILE)
|
log_file = types.FSInputFile(str(Config.LOG_FILE))
|
||||||
await message.answer_document(log_file, caption="📑 Логи бота")
|
await message.answer_document(log_file, caption="📑 Логи бота")
|
||||||
except FileNotFoundError:
|
except FileNotFoundError:
|
||||||
await message.answer("Файл логов пока не создан.")
|
await message.answer("Файл логов пока не создан.")
|
||||||
@@ -31,7 +37,12 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
|
|||||||
from utils.mac_metrics import get_macbook_battery_level, get_process_usage
|
from utils.mac_metrics import get_macbook_battery_level, get_process_usage
|
||||||
|
|
||||||
try:
|
try:
|
||||||
stats = analyze_bot_logs(Config.LOG_FILE)
|
if Config.DISABLE_LOGGING or Config.DISABLE_STORAGE:
|
||||||
|
await message.answer(
|
||||||
|
"📊 Аналитика по логам недоступна: логирование или хранение отключено в .env"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
stats = analyze_bot_logs(str(Config.LOG_FILE))
|
||||||
batt = await get_macbook_battery_level()
|
batt = await get_macbook_battery_level()
|
||||||
usage = await get_process_usage()
|
usage = await get_process_usage()
|
||||||
status_text = (
|
status_text = (
|
||||||
@@ -53,7 +64,12 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
|
|||||||
async def stat(message: Message):
|
async def stat(message: Message):
|
||||||
from utils.analytics import analyze_bot_logs
|
from utils.analytics import analyze_bot_logs
|
||||||
|
|
||||||
stats = analyze_bot_logs(Config.LOG_FILE)
|
if Config.DISABLE_LOGGING or Config.DISABLE_STORAGE:
|
||||||
|
await message.answer(
|
||||||
|
"📊 Аналитика по логам недоступна: логирование или хранение отключено в .env"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
stats = analyze_bot_logs(str(Config.LOG_FILE))
|
||||||
await message.answer(
|
await message.answer(
|
||||||
create_statistics_text(stats), reply_to_message_id=message.message_id
|
create_statistics_text(stats), reply_to_message_id=message.message_id
|
||||||
)
|
)
|
||||||
@@ -61,6 +77,12 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
|
|||||||
@dp.message(Command("del"))
|
@dp.message(Command("del"))
|
||||||
@admin_required(1)
|
@admin_required(1)
|
||||||
async def delete_all_messages(message: Message):
|
async def delete_all_messages(message: Message):
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
await message.answer(
|
||||||
|
"📭 Хранение сообщений отключено (DISABLE_STORAGE=1).",
|
||||||
|
reply_to_message_id=message.message_id,
|
||||||
|
)
|
||||||
|
return
|
||||||
messages = load_messages()
|
messages = load_messages()
|
||||||
if not messages:
|
if not messages:
|
||||||
sent = await message.answer(
|
sent = await message.answer(
|
||||||
@@ -91,16 +113,33 @@ def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
|
|||||||
async def power_control(message: types.Message):
|
async def power_control(message: types.Message):
|
||||||
args = message.text.split()
|
args = message.text.split()
|
||||||
if len(args) < 2:
|
if len(args) < 2:
|
||||||
|
days = state.watcher_days_ahead
|
||||||
status = "включена" if state.watcher_work else "выключена"
|
status = "включена" if state.watcher_work else "выключена"
|
||||||
await message.answer(f"⏱️ Слежка расписания: {status}")
|
await message.answer(f"⏱️ Слежка расписания: {status} (на {days} дн.)")
|
||||||
return
|
return
|
||||||
|
|
||||||
command = args[1].lower()
|
command = args[1].lower()
|
||||||
watcher_service = WatcherService(state, bot)
|
watcher_service = WatcherService(state, bot)
|
||||||
|
|
||||||
if command == "on" and not state.watcher_work:
|
if command == "on":
|
||||||
|
# Проверяем, есть ли параметр количества дней
|
||||||
|
days = 1
|
||||||
|
if len(args) > 2:
|
||||||
|
try:
|
||||||
|
days = int(args[2])
|
||||||
|
if days < 1:
|
||||||
|
await message.answer("❌ Количество дней должно быть >= 1")
|
||||||
|
return
|
||||||
|
except ValueError:
|
||||||
|
await message.answer("❌ Неверный формат дней. Используйте: /power on 3")
|
||||||
|
return
|
||||||
|
|
||||||
|
state.watcher_days_ahead = days
|
||||||
|
if not state.watcher_work:
|
||||||
await watcher_service.start()
|
await watcher_service.start()
|
||||||
await message.answer("✅ Слежка расписания включена")
|
await message.answer(f"✅ Слежка расписания включена (на {days} дн.)")
|
||||||
|
else:
|
||||||
|
await message.answer(f"✅ Количество дней изменено на {days} дн.")
|
||||||
elif command == "off" and state.watcher_work:
|
elif command == "off" and state.watcher_work:
|
||||||
await watcher_service.stop()
|
await watcher_service.stop()
|
||||||
await message.answer("❌ Слежка расписания выключена")
|
await message.answer("❌ Слежка расписания выключена")
|
||||||
|
|||||||
@@ -1,17 +1,11 @@
|
|||||||
from asyncio import run
|
from asyncio import run
|
||||||
from logging import basicConfig, FileHandler, StreamHandler, INFO, getLogger
|
from logging import getLogger
|
||||||
|
|
||||||
from bot.core import TelegramBot
|
from bot.core import TelegramBot
|
||||||
from config import Config
|
from config import Config
|
||||||
|
from utils.logging_config import setup_logging
|
||||||
|
|
||||||
# Настройка логирования
|
setup_logging()
|
||||||
basicConfig(
|
|
||||||
level=INFO,
|
|
||||||
format="%(asctime)s [%(levelname)s] %(message)s",
|
|
||||||
datefmt="%Y-%m-%d %H:%M:%S",
|
|
||||||
handlers=[FileHandler(Config.LOG_FILE, encoding="utf-8"), StreamHandler()],
|
|
||||||
force=True,
|
|
||||||
)
|
|
||||||
|
|
||||||
logger = getLogger(__name__)
|
logger = getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ class BotState:
|
|||||||
last_pinned: Dict[str, int] = None
|
last_pinned: Dict[str, int] = None
|
||||||
watcher_work: bool = False
|
watcher_work: bool = False
|
||||||
watcher_task: Optional[Task] = None
|
watcher_task: Optional[Task] = None
|
||||||
|
watcher_days_ahead: int = 1
|
||||||
file_id_cache: Dict[str, str] = None
|
file_id_cache: Dict[str, str] = None
|
||||||
last_day: Dict[str, int] = None
|
last_day: Dict[str, int] = None
|
||||||
last_clip_hash: Dict[str, str] = None
|
last_clip_hash: Dict[str, str] = None
|
||||||
|
|||||||
@@ -36,6 +36,12 @@ class ScheduleService:
|
|||||||
target = target.replace(day=int(day_offset))
|
target = target.replace(day=int(day_offset))
|
||||||
return target.replace(hour=0, minute=0, second=0, microsecond=0)
|
return target.replace(hour=0, minute=0, second=0, microsecond=0)
|
||||||
|
|
||||||
|
|
||||||
|
def _next_target_date(self, day_offset: int = 0) -> datetime:
|
||||||
|
return (datetime.now() + timedelta(days=day_offset)).replace(hour=0, minute=0, second=0, microsecond=0)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def _load_pdf_for_date(
|
async def _load_pdf_for_date(
|
||||||
self, day_offset: int = 0
|
self, day_offset: int = 0
|
||||||
) -> Tuple[Optional[bytes], Optional[str], int, int]:
|
) -> Tuple[Optional[bytes], Optional[str], int, int]:
|
||||||
@@ -54,6 +60,25 @@ class ScheduleService:
|
|||||||
url = f"https://drive.google.com/file/d/{drive_file.file_id}/view"
|
url = f"https://drive.google.com/file/d/{drive_file.file_id}/view"
|
||||||
return self._pdf_cache[drive_file.file_id], url, day, month
|
return self._pdf_cache[drive_file.file_id], url, day, month
|
||||||
|
|
||||||
|
async def _load_pdf_for_watcher(
|
||||||
|
self, day_offset: int = 1
|
||||||
|
) -> Tuple[Optional[bytes], Optional[str], int, int]:
|
||||||
|
target = self._next_target_date(day_offset)
|
||||||
|
day, month = target.day, target.month
|
||||||
|
|
||||||
|
drive_file = await self.drive.find_for_date(target)
|
||||||
|
if not drive_file:
|
||||||
|
return None, None, day, month
|
||||||
|
|
||||||
|
if drive_file.file_id not in self._pdf_cache:
|
||||||
|
self._pdf_cache[drive_file.file_id] = await self.drive.download_pdf(
|
||||||
|
drive_file.file_id
|
||||||
|
)
|
||||||
|
|
||||||
|
url = f"https://drive.google.com/file/d/{drive_file.file_id}/view"
|
||||||
|
return self._pdf_cache[drive_file.file_id], url, day, month
|
||||||
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def exact_group_regex(group: str) -> re.Pattern:
|
def exact_group_regex(group: str) -> re.Pattern:
|
||||||
pattern = rf"(^|{BOUNDARY}){re.escape(group)}({BOUNDARY}|$)"
|
pattern = rf"(^|{BOUNDARY}){re.escape(group)}({BOUNDARY}|$)"
|
||||||
@@ -105,7 +130,7 @@ class ScheduleService:
|
|||||||
return f"📅 Расписание для {day} числа:\n<pre>{body}</pre>"
|
return f"📅 Расписание для {day} числа:\n<pre>{body}</pre>"
|
||||||
|
|
||||||
async def is_published_for(self, day_offset: int = 0) -> bool:
|
async def is_published_for(self, day_offset: int = 0) -> bool:
|
||||||
target = self._resolve_target_date(day_offset)
|
target = self._next_target_date(day_offset)
|
||||||
return await self.drive.find_for_date(target) is not None
|
return await self.drive.find_for_date(target) is not None
|
||||||
|
|
||||||
async def get_schedule(
|
async def get_schedule(
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
from random import randint
|
from random import randint
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
from aiogram import Bot, types
|
from aiogram import Bot, types
|
||||||
|
|
||||||
from config import Config
|
from config import Config
|
||||||
from logging import getLogger
|
from logging import getLogger
|
||||||
from models.state import BotState
|
from models.state import BotState
|
||||||
@@ -47,6 +47,20 @@ class WatcherService:
|
|||||||
Config.WATCHER_RANDOM_DELAY_MAX,
|
Config.WATCHER_RANDOM_DELAY_MAX,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _get_target_date_with_weekend_handling(days_ahead: int) -> datetime:
|
||||||
|
"""
|
||||||
|
Получить целевую дату с учетом выходных.
|
||||||
|
Если целевая дата - воскресенье, переносится на понедельник.
|
||||||
|
"""
|
||||||
|
target = (datetime.now() + timedelta(days=days_ahead)).replace(
|
||||||
|
hour=0, minute=0, second=0, microsecond=0
|
||||||
|
)
|
||||||
|
# weekday() returns 6 for Sunday
|
||||||
|
if target.weekday() == 6:
|
||||||
|
target += timedelta(days=1)
|
||||||
|
return target
|
||||||
|
|
||||||
async def _watcher_loop(self):
|
async def _watcher_loop(self):
|
||||||
"""Основной цикл слежки за появлением PDF на Google Drive."""
|
"""Основной цикл слежки за появлением PDF на Google Drive."""
|
||||||
while self.state.watcher_work:
|
while self.state.watcher_work:
|
||||||
@@ -71,12 +85,14 @@ class WatcherService:
|
|||||||
Возвращает True, если расписание ещё недоступно ни для одной группы.
|
Возвращает True, если расписание ещё недоступно ни для одной группы.
|
||||||
Возвращает False, если хотя бы одной группе отправили расписание.
|
Возвращает False, если хотя бы одной группе отправили расписание.
|
||||||
"""
|
"""
|
||||||
target = self.schedule_service._resolve_target_date(0)
|
days_ahead = self.state.watcher_days_ahead
|
||||||
|
target = self._get_target_date_with_weekend_handling(days_ahead)
|
||||||
logger.info(
|
logger.info(
|
||||||
f"Проверяем Google Drive на расписание за {target.strftime('%d.%m.%Y')}"
|
f"Проверяем Google Drive на расписание за {target.strftime('%d.%m.%Y')} "
|
||||||
|
f"(дней вперед: {days_ahead})"
|
||||||
)
|
)
|
||||||
|
|
||||||
if not await self.schedule_service.is_published_for(0):
|
if not await self.schedule_service.is_published_for(days_ahead):
|
||||||
return True
|
return True
|
||||||
|
|
||||||
found_any = False
|
found_any = False
|
||||||
@@ -84,14 +100,15 @@ class WatcherService:
|
|||||||
logger.info(
|
logger.info(
|
||||||
f"Проверяем расписание для {group} на {target.strftime('%d.%m.%Y')}"
|
f"Проверяем расписание для {group} на {target.strftime('%d.%m.%Y')}"
|
||||||
)
|
)
|
||||||
if await self._check_group_schedule(group, chat_id):
|
if await self._check_group_schedule(group, chat_id, days_ahead):
|
||||||
found_any = True
|
found_any = True
|
||||||
|
|
||||||
return not found_any
|
return not found_any
|
||||||
|
|
||||||
async def _check_group_schedule(self, group: str, chat_id: int) -> bool:
|
async def _check_group_schedule(self, group: str, chat_id: int, days_ahead: int) -> bool:
|
||||||
|
target = self._get_target_date_with_weekend_handling(days_ahead)
|
||||||
text, url, data_day, data_month = await self.schedule_service.get_schedule(
|
text, url, data_day, data_month = await self.schedule_service.get_schedule(
|
||||||
group, 0
|
group, target.day
|
||||||
)
|
)
|
||||||
if not self.schedule_service.is_schedule_missing(text):
|
if not self.schedule_service.is_schedule_missing(text):
|
||||||
msg = await self.bot.send_message(
|
msg = await self.bot.send_message(
|
||||||
|
|||||||
+4
-1
@@ -1,6 +1,9 @@
|
|||||||
import sqlite3
|
import sqlite3
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
DIR = "/Users/mac/myfirstprogramm/storage/message.db"
|
from config import Config
|
||||||
|
|
||||||
|
DIR = Path(__file__).resolve().parent / "message.db"
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
db = sqlite3.connect(DIR)
|
db = sqlite3.connect(DIR)
|
||||||
|
|||||||
@@ -1,7 +1,13 @@
|
|||||||
from .DB import get_db
|
from config import Config
|
||||||
|
|
||||||
|
|
||||||
def save_message(chat_id: int, message_id: int):
|
def save_message(chat_id: int, message_id: int):
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
return
|
||||||
|
if True:
|
||||||
|
return
|
||||||
|
from .DB import get_db
|
||||||
|
|
||||||
db = get_db()
|
db = get_db()
|
||||||
cur = db.cursor()
|
cur = db.cursor()
|
||||||
cur.execute("INSERT INTO message VALUES (?, ?)", (int(chat_id), int(message_id)))
|
cur.execute("INSERT INTO message VALUES (?, ?)", (int(chat_id), int(message_id)))
|
||||||
@@ -11,6 +17,12 @@ def save_message(chat_id: int, message_id: int):
|
|||||||
|
|
||||||
|
|
||||||
def load_messages():
|
def load_messages():
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
return []
|
||||||
|
if True:
|
||||||
|
return []
|
||||||
|
from .DB import get_db
|
||||||
|
|
||||||
db = get_db()
|
db = get_db()
|
||||||
cur = db.cursor()
|
cur = db.cursor()
|
||||||
cur.execute("SELECT * FROM message")
|
cur.execute("SELECT * FROM message")
|
||||||
@@ -21,6 +33,12 @@ def load_messages():
|
|||||||
|
|
||||||
|
|
||||||
def clear_messages():
|
def clear_messages():
|
||||||
|
if Config.DISABLE_STORAGE:
|
||||||
|
return
|
||||||
|
if True:
|
||||||
|
return
|
||||||
|
from .DB import get_db
|
||||||
|
|
||||||
db = get_db()
|
db = get_db()
|
||||||
cur = db.cursor()
|
cur = db.cursor()
|
||||||
cur.execute("DELETE FROM message")
|
cur.execute("DELETE FROM message")
|
||||||
|
|||||||
@@ -1,6 +1,11 @@
|
|||||||
from .DB import get_db
|
from config import Config
|
||||||
|
|
||||||
|
_DEFAULT_GROUP = "30тс"
|
||||||
|
|
||||||
|
|
||||||
|
def save_user(user_id: int, group: str = _DEFAULT_GROUP):
|
||||||
|
from .DB import get_db
|
||||||
|
|
||||||
def save_user(user_id: int, group: str = "30тс"):
|
|
||||||
db = get_db()
|
db = get_db()
|
||||||
cur = db.cursor()
|
cur = db.cursor()
|
||||||
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
|
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
|
||||||
@@ -8,19 +13,18 @@ def save_user(user_id: int, group: str = "30тс"):
|
|||||||
cur.close()
|
cur.close()
|
||||||
db.close()
|
db.close()
|
||||||
|
|
||||||
def set_group(user_id: int, group: str = "30тс"):
|
|
||||||
|
def set_group(user_id: int, group: str = _DEFAULT_GROUP):
|
||||||
|
from .DB import get_db
|
||||||
|
|
||||||
db = get_db()
|
db = get_db()
|
||||||
cur = db.cursor()
|
cur = db.cursor()
|
||||||
|
|
||||||
# проверяем, есть ли пользователь
|
|
||||||
cur.execute("SELECT 1 FROM users WHERE user_id = ?", (user_id,))
|
cur.execute("SELECT 1 FROM users WHERE user_id = ?", (user_id,))
|
||||||
exists = cur.fetchone()
|
exists = cur.fetchone()
|
||||||
|
|
||||||
if exists:
|
if exists:
|
||||||
# если есть — обновляем группу
|
|
||||||
cur.execute("UPDATE users SET user_group = ? WHERE user_id = ?", (group, user_id))
|
cur.execute("UPDATE users SET user_group = ? WHERE user_id = ?", (group, user_id))
|
||||||
else:
|
else:
|
||||||
# если нет — регистрируем нового пользователя
|
|
||||||
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
|
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
|
||||||
|
|
||||||
db.commit()
|
db.commit()
|
||||||
@@ -28,7 +32,9 @@ def set_group(user_id: int, group: str = "30тс"):
|
|||||||
db.close()
|
db.close()
|
||||||
|
|
||||||
|
|
||||||
def get_group(user_id: int, default: str = "30тс") -> str:
|
def get_group(user_id: int, default: str = _DEFAULT_GROUP) -> str:
|
||||||
|
from .DB import get_db
|
||||||
|
|
||||||
db = get_db()
|
db = get_db()
|
||||||
cur = db.cursor()
|
cur = db.cursor()
|
||||||
cur.execute("SELECT user_group FROM users WHERE user_id = ?", (user_id,))
|
cur.execute("SELECT user_group FROM users WHERE user_id = ?", (user_id,))
|
||||||
@@ -36,7 +42,6 @@ def get_group(user_id: int, default: str = "30тс") -> str:
|
|||||||
if row:
|
if row:
|
||||||
group = row[0]
|
group = row[0]
|
||||||
else:
|
else:
|
||||||
# если пользователя нет — регистрируем с дефолтной группой
|
|
||||||
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, default))
|
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, default))
|
||||||
db.commit()
|
db.commit()
|
||||||
group = default
|
group = default
|
||||||
|
|||||||
@@ -0,0 +1,33 @@
|
|||||||
|
import logging
|
||||||
|
from logging import CRITICAL, NullHandler, getLogger
|
||||||
|
|
||||||
|
from config import Config
|
||||||
|
|
||||||
|
|
||||||
|
def setup_logging() -> None:
|
||||||
|
"""Настройка логирования. При DISABLE_LOGGING — полное отключение."""
|
||||||
|
root = getLogger()
|
||||||
|
|
||||||
|
if Config.DISABLE_LOGGING:
|
||||||
|
root.handlers.clear()
|
||||||
|
root.addHandler(NullHandler())
|
||||||
|
root.setLevel(CRITICAL)
|
||||||
|
logging.disable(CRITICAL)
|
||||||
|
return
|
||||||
|
|
||||||
|
from logging import INFO, FileHandler, StreamHandler, basicConfig
|
||||||
|
|
||||||
|
handlers: list[logging.Handler] = [StreamHandler()]
|
||||||
|
|
||||||
|
if not Config.DISABLE_STORAGE:
|
||||||
|
log_path = Config.LOG_FILE
|
||||||
|
log_path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
handlers.append(FileHandler(log_path, encoding="utf-8"))
|
||||||
|
|
||||||
|
basicConfig(
|
||||||
|
level=INFO,
|
||||||
|
format="%(asctime)s [%(levelname)s] %(message)s",
|
||||||
|
datefmt="%Y-%m-%d %H:%M:%S",
|
||||||
|
handlers=handlers,
|
||||||
|
force=True,
|
||||||
|
)
|
||||||
Reference in New Issue
Block a user