fix: add should_count to message count inside send to group

This commit is contained in:
Krzysztof Rudnicki 2024-10-05 10:15:34 +02:00
parent 0205627e26
commit 3c79bfb4eb

View File

@ -136,7 +136,7 @@ USER_MESSAGE_COUNT = {}
async def count_messages(message_content, counter):
if message_content:
if message_content and should_count(message_content):
uuid = extract_source_uuid(message_content)
source_name = extract_source_name(message_content)
await counter.update_string_map(uuid, source_name)
@ -181,6 +181,7 @@ async def trigger_command(message_content, recipient):
async def send_to_group(message_content, counter, message):
if message_group_id(message_content) == GROUP_ID:
await count_messages(json.loads(message).get('envelope', {}), counter)
await trigger_command(message_content, GROUP_ID_SEND)
async def remove_attachment(attachment_id):
@ -213,8 +214,6 @@ def is_message_reaction(message):
return False
def should_count(message_content):
if message_content.get('destinationNumber', {}) != PHONE_NUMBER:
return False
if message_content.get('sticker') != None:
print("not counting because message has a sticker")
return False
@ -231,8 +230,7 @@ async def listen_to_server(counter):
message_content = extract_message_content(message)
await send_to_group(message_content, counter, message)
print("message_content.get('sticker') == None: ", message_content.get('sticker') == None, message_content.get('sticker'))
if should_count(message_content):
print("counting message")
if message_content.get('destinationNumber', {}) != PHONE_NUMBER:
await count_messages(json.loads(message).get('envelope', {}), counter)
await trigger_command(message_content, PHONE_NUMBER)
except websockets.ConnectionClosed as e: