advatroniks/Telegram-Scrapper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add fix

Browse Source
This commit is contained in:
harold 2025-05-05 17:04:50 +05:00
parent 7f59b35d62
commit 7bc7e5b09f
2 changed files with 26 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
telegram-application/src/core/tg_service/messages_handler.py
View File

@ -31,7 +31,6 @@ async def message_listener(client: Client, message: Message):
text=message.text, text=message.text,
message_time=message.date, message_time=message.date,
) )
async with lock:
await check_chunk_state_and_publish( await check_chunk_state_and_publish(
data=DATA, data=DATA,
message_schema=message_schema, message_schema=message_schema,

2
telegram-application/src/core/tg_service/service.py
View File

@ -6,6 +6,7 @@ from src.core.database import TgChat, User
from src.core.rabbitmq.connect import message_handler_publisher from src.core.rabbitmq.connect import message_handler_publisher
from src.core.redis_helper.redis_connect import redis_client from src.core.redis_helper.redis_connect import redis_client
from src.core.tg_service.constants import MESSAGE_CHANG_SIZE from src.core.tg_service.constants import MESSAGE_CHANG_SIZE
from src.core.tg_service.messages_handler import lock
from src.core.tg_service.schemas import UserFromMessageSchema, MessagesForSendToWorkersSchema, MessageFromChatSchema from src.core.tg_service.schemas import UserFromMessageSchema, MessagesForSendToWorkersSchema, MessageFromChatSchema
from src.core.database.connect import db_helper from src.core.database.connect import db_helper
from src.core.tg_service import crud as tg_crud from src.core.tg_service import crud as tg_crud
@ -65,6 +66,7 @@ async def check_chunk_state_and_publish(
): ):
messages_chunk = data.get(chat_id) messages_chunk = data.get(chat_id)
async with lock:
if messages_chunk is None: if messages_chunk is None:
data[chat_id] = [message_schema] data[chat_id] = [message_schema]