diff --git a/app.py b/app.py index 8f56a9a..cfa109c 100644 --- a/app.py +++ b/app.py @@ -1,7 +1,8 @@ #!/usr/bin/env python import sentry_sdk import time -import request +import requests +import requests.exceptions import os import redis import json @@ -15,9 +16,10 @@ __copyright__ = "Copyright 2020, GoldenPogácsa Team" __module_name__ = "app" __version__text__ = "1" + sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10") -if __name__ == "__main__": +def main(): r = redis.Redis(host='localhost', port=6379, db=0) ip_list = os.environ['INITIAL_SERVERS'].split(',') @@ -26,11 +28,14 @@ if __name__ == "__main__": temp_dict = {} - for i in ip_list: - response = request.post(http=i + '/sync', json={'uuid': os.environ['LOCAL_UUID']}) + for ip in ip_list: + try: + response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']}) + except requests.exceptions.ConnectionError: + continue if response.status_code == 200: - temp_dict[response.json()['uuid']] = {'ip': i} + temp_dict[response.json()['uuid']] = {'ip': ip} ip_list_redis.update(temp_dict) @@ -40,14 +45,20 @@ if __name__ == "__main__": ip_list_redis = json.loads(r.get('consumer_list').decode('utf-8')) - for i in ip_list_redis: - response = request.post(http=i + '/sync', json={'uuid': os.environ['LOCAL_UID']}) + for ip in ip_list_redis: + try: + response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']}) + except requests.exceptions.ConnectionError: + continue + if response.status_code == 200: - temp_dict[response.json()['uuid']] = {'ip': i} + temp_dict[response.json()['uuid']] = {'ip': ip} ip_list_redis.update(temp_dict) r.set('consumer_list', json.dumps(ip_list_redis).encode('utf-8')) time.sleep(30) - print("Consumer-Scheduler") + +if __name__ == "__main__": + main() diff --git a/requirements.txt b/requirements.txt index 9e669fa..b60d5ad 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ sentry_sdk -request +requests redis \ No newline at end of file