augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-24 07:36:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add protocol option

Browse Source
This commit is contained in:
augustin64 2022-07-02 10:56:55 +02:00
parent 3faff1e541
commit 26a4d7a0f6
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/parallel/client.py
View File

@ -4,17 +4,20 @@ Client se connectant au serveur Flask afin de fournir de la puissance de calcul.
""" """
import json import json
import os import os
import sys import shutil
import time
import subprocess import subprocess
import sys
import tempfile
import time
import psutil import psutil
import requests import requests
# Définition de constantes # Définition de constantes
CACHE = "/tmp/parallel/client_cache" # Replace with an absolute path CACHE = tempfile.mkdtemp()
DELTA = os.path.join(CACHE, "delta_shared.bin") DELTA = os.path.join(CACHE, "delta_shared.bin")
RESEAU = os.path.join(CACHE, "reseau_shared.bin") RESEAU = os.path.join(CACHE, "reseau_shared.bin")
PROTOCOL = "https"
if len(sys.argv) > 1: if len(sys.argv) > 1:
HOST = sys.argv[1] HOST = sys.argv[1]
@ -46,7 +49,7 @@ def authenticate():
performance = get_performance() performance = get_performance()
data = {"performance": performance, "secret": SECRET} data = {"performance": performance, "secret": SECRET}
# Les données d'identification seront ensuite stockées dans un cookie de l'objet session # Les données d'identification seront ensuite stockées dans un cookie de l'objet session
req = session.post(f"http://{HOST}/authenticate", json=data) req = session.post(f"{PROTOCOL}://{HOST}/authenticate", json=data)
data = json.loads(req.text) data = json.loads(req.text)
if data["status"] != "ok": if data["status"] != "ok":
@ -60,7 +63,7 @@ def download_network():
""" """
Récupère le réseau depuis le serveur Récupère le réseau depuis le serveur
""" """
with session.get(f"http://{HOST}/get_network", stream=True) as req: with session.get(f"{PROTOCOL}://{HOST}/get_network", stream=True) as req:
req.raise_for_status() req.raise_for_status()
with open(RESEAU, "wb") as file: with open(RESEAU, "wb") as file:
for chunk in req.iter_content(chunk_size=8192): for chunk in req.iter_content(chunk_size=8192):
@ -73,7 +76,7 @@ def send_delta_network(continue_=False):
""" """
with open(DELTA, "rb") as file: with open(DELTA, "rb") as file:
files = {"file": file} files = {"file": file}
req = session.post(f"http://{HOST}/post_network", files=files) req = session.post(f"{PROTOCOL}://{HOST}/post_network", files=files)
req_data = json.loads(req.text) req_data = json.loads(req.text)
# Actions à effectuer en fonction de la réponse # Actions à effectuer en fonction de la réponse
@ -146,8 +149,11 @@ def __main__():
# train_shared s'appelle récursivement sur lui même jusqu'à la fin du programme # train_shared s'appelle récursivement sur lui même jusqu'à la fin du programme
try: try:
train_shared(dataset, start, nb_elem, epochs=1, out=DELTA) train_shared(dataset, start, nb_elem, epochs=1, out=DELTA)
except requests.exceptions.ConnectionError: except requests.exceptions.ConnectionError and json.decoder.JSONDecodeError:
# requests.exceptions.ConnectionError -> Host disconnected
# json.decoder.JSONDecodeError -> Host disconnected but nginx handles it
print("Host disconnected") print("Host disconnected")
shutil.rmtree(CACHE)
if __name__ == "__main__": if __name__ == "__main__":