Other fixes

This commit is contained in:
Pünkösd Marcell 2020-03-29 19:30:59 +02:00
parent 04edb91a39
commit 805e735da5

17
app.py
View File

@ -16,15 +16,15 @@ __copyright__ = "Copyright 2020, GoldenPogácsa Team"
__module_name__ = "app" __module_name__ = "app"
__version__text__ = "1" __version__text__ = "1"
sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10") sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10")
def main(): def main():
r = redis.Redis(host='localhost', port=6379, db=0) r = redis.Redis(host='localhost', port=6379, db=0)
ip_list = os.environ['INITIAL_SERVERS'].split(',') ip_list = os.environ['INITIAL_SERVERS'].split(',')
ip_list_redis = json.loads(r.get('consumer_list').decode('utf-8')) consumer_list_redis = json.loads((r.get('consumer_list') or b'{}').decode('utf-8'))
temp_dict = {} temp_dict = {}
@ -37,15 +37,16 @@ def main():
if response.status_code == 200: if response.status_code == 200:
temp_dict[response.json()['uuid']] = {'ip': ip} temp_dict[response.json()['uuid']] = {'ip': ip}
ip_list_redis.update(temp_dict) consumer_list_redis.update(temp_dict)
r.set('consumer_list', json.dumps(ip_list_redis).encode('utf-8')) r.set('consumer_list', json.dumps(consumer_list_redis).encode('utf-8'))
while True: while True:
ip_list_redis = json.loads(r.get('consumer_list').decode('utf-8')) consumer_list_redis = json.loads((r.get('consumer_list') or b'{}').decode('utf-8'))
for ip in ip_list_redis: for uuid, info in consumer_list_redis.items():
ip = info['ip']
try: try:
response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']}) response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']})
except requests.exceptions.ConnectionError: except requests.exceptions.ConnectionError:
@ -54,8 +55,8 @@ def main():
if response.status_code == 200: if response.status_code == 200:
temp_dict[response.json()['uuid']] = {'ip': ip} temp_dict[response.json()['uuid']] = {'ip': ip}
ip_list_redis.update(temp_dict) consumer_list_redis.update(temp_dict)
r.set('consumer_list', json.dumps(ip_list_redis).encode('utf-8')) r.set('consumer_list', json.dumps(consumer_list_redis).encode('utf-8'))
time.sleep(30) time.sleep(30)