From b037b25ef8fb50d6b698a50095348abcba6c39cc Mon Sep 17 00:00:00 2001 From: workalitkp <62553394+workalitkp@users.noreply.github.com> Date: Sat, 29 May 2021 01:07:23 +0430 Subject: [PATCH 1/2] updating get_messages method using requests sessions instead of sseclient --- client.py | 30 ++++++++++-------------------- 1 file changed, 10 insertions(+), 20 deletions(-) diff --git a/client.py b/client.py index 90f04a2..aa3db19 100644 --- a/client.py +++ b/client.py @@ -1,5 +1,4 @@ import requests -import sseclient import json import os from time import sleep @@ -39,25 +38,16 @@ def get_messages(self): while True: try: - response = requests.get(url, stream=True) - if 'Content-Type' in response.headers: - client = sseclient.SSEClient(response) - - print('connected successfully\n') - - for event in client.events(): - try: - message_event = json.loads(event.data) - yield message_event - except Exception as e: - print(e.args[0]) - continue - else: - print('Invalid bot token OR Invalid connection response from server') - - print('retry to connect after 10 seconds...') - sleep(self.RETRY_DELAY) - + session = requests.Session() + with session.get(url, headers=None, stream=True) as resp: + for line in resp.iter_lines(): + if line: + try: + message_event = json.loads(line.decode("utf-8")[5::])["body"] + yield message_event + except Exception as e: + print(e.args[0]) + continue except Exception as e: print(e.args[0]) print('retry to connect after 10 seconds...') From 1624e8e46ee1e0a88ca5d032b963c0037797aa41 Mon Sep 17 00:00:00 2001 From: workalitkp <62553394+workalitkp@users.noreply.github.com> Date: Sat, 29 May 2021 01:20:16 +0430 Subject: [PATCH 2/2] deleted unnecessary libs --- requirements.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index a944a1c..f229360 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1 @@ requests -sseclient-py