feat: ignore reaction messages

This commit is contained in:
Krzysztof Rudnicki 2024-06-24 21:10:10 +02:00
parent 6a897a80a2
commit e8efcfffcb

21
main.py
View File

@ -189,17 +189,28 @@ async def get_attachments():
print(f"Request failed with status code: {response.status_code}") print(f"Request failed with status code: {response.status_code}")
print(response) print(response)
async def is_message_reaction(message):
message_json = json.loads(message)
inside_message = message_json.get('envelope', {}).get('dataMessage', {}).get('reaction', {})
if inside_message == {}:
inside_message = message_json.get('envelope', {}).get('syncMessage', {}).get('reaction', {})
if inside_message != {}:
return True
return False
async def listen_to_server(counter): async def listen_to_server(counter):
uri = f"ws://localhost:9922/v1/receive/{PHONE_NUMBER}?send_read_receipts=false" uri = f"ws://localhost:9922/v1/receive/{PHONE_NUMBER}?send_read_receipts=false"
async with websockets.connect(uri) as websocket: async with websockets.connect(uri) as websocket:
print(f"Connected to signal server") print(f"Connected to signal server")
try: try:
async for message in websocket: async for message in websocket:
message_content = extract_message_content(message) if is_message_reaction(message) == False:
await send_to_group(message_content, counter, message) print("message: ", message)
if message_content.get('destinationNumber', {}) == PHONE_NUMBER: message_content = extract_message_content(message)
await count_messages(json.loads(message).get('envelope', {}), counter) await send_to_group(message_content, counter, message)
await trigger_command(message_content, PHONE_NUMBER) 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: except websockets.ConnectionClosed as e:
print(f"Connection closed: {e}") print(f"Connection closed: {e}")