mirror of
https://github.com/kuhyx/signal-bot.git
synced 2026-07-04 13:23:07 +02:00
fix: add should_count to message count inside send to group
This commit is contained in:
parent
0205627e26
commit
3c79bfb4eb
8
main.py
8
main.py
@ -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:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user