mirror of
https://github.com/openmax-server/server.git
synced 2026-05-26 13:31:43 +03:00
39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
import pydantic
|
||
import json
|
||
import time
|
||
from classes.baseprocessor import BaseProcessor
|
||
from oneme.models import SyncFoldersPayloadModel
|
||
|
||
class FoldersProcessors(BaseProcessor):
|
||
async def folders_get(self, payload, seq, writer, senderPhone):
|
||
"""Синхронизация папок с сервером"""
|
||
# Валидируем данные пакета
|
||
try:
|
||
SyncFoldersPayloadModel.model_validate(payload)
|
||
except pydantic.ValidationError as error:
|
||
self.logger.error(f"Возникли ошибки при валидации пакета: {error}")
|
||
await self._send_error(seq, self.opcodes.FOLDERS_GET, self.error_types.INVALID_PAYLOAD, writer)
|
||
return
|
||
|
||
# Ищем папки в бд
|
||
async with self.db_pool.acquire() as conn:
|
||
async with conn.cursor() as cursor:
|
||
await cursor.execute("SELECT folders FROM user_data WHERE phone = %s", (int(senderPhone),))
|
||
result_folders = await cursor.fetchone()
|
||
user_folders = json.loads(result_folders.get("folders"))
|
||
|
||
# Создаем данные пакета
|
||
payload = {
|
||
"folderSync": int(time.time() * 1000),
|
||
"folders": self.static.ALL_CHAT_FOLDER + user_folders.get("folders"),
|
||
"foldersOrder": self.static.ALL_CHAT_FOLDER_ORDER + user_folders.get("foldersOrder"),
|
||
"allFilterExcludeFolders": user_folders.get("allFilterExcludeFolders")
|
||
}
|
||
|
||
# Собираем пакет
|
||
packet = self.proto.pack_packet(
|
||
cmd=self.proto.CMD_OK, seq=seq, opcode=self.opcodes.FOLDERS_GET, payload=payload
|
||
)
|
||
|
||
# Отправляем
|
||
await self._send(writer, packet) |