mudanças no método post do handle_request

Author: GiovannaMafra

data/keys.json

@@ -1 +1,4 @@
-[]
+[
+    {"chave": 1234},
+    {"chave": 4567}
+]

src/main_server.py

@@ -1,4 +1,6 @@
 from utils import http_parser_request
+from utils import readjson
+from utils import writejson
 
 from socket import socket, AF_INET, SOCK_STREAM
 from threading import Thread
@@ -20,8 +22,16 @@ def http_get(socket_client):
 
     socket_client.send(msgHtml.encode())
 
-def http_post():
-    ...
+def http_post(socket_client):
+    #duvida de como seria o reply
+    #client ainda não votou
+    msgHeader = 'HTTP/1.1 200 OK \r\n' \
+                'Host: voting.com\r\n' \
+                'Content-Type: text/html\r\n' \
+                '\r\n'
+                #'voto': registrado
+   
+    socket_client.send(msgHeader.encode())
 
 def handle_request(socket_client):
     req = socket_client.recv(2048).decode()
@@ -35,7 +45,24 @@ def handle_request(socket_client):
         http_get(socket_client)
 
     elif(request['Method'] == 'POST'):
-        http_post()
+        votou = False
+        #adicionar o voto
+        for cliente in readjson("keys"):
+            if cliente['chave'] == request['Key']:
+                #caso a pessoa já votou
+                votou = True
+                #reply de já votou http_post()
+
+
+        #caso a pessoa ainda não votou
+        if votou == False:
+            #adiciona a chave no keys.json
+            keys_dados = readjson("keys")
+            keys_dados.append({"chave": request['Key']})
+            writejson("keys", keys_dados)
+
+            #responder o voto registrado
+            http_post(socket_client)
 
     return
 

src/utils.py

@@ -1,11 +1,19 @@
+import os
 import json
 
-def readjson():
-    f = open('./data/dns.json', 'r')
+def readjson(path):
+    file = os.path.join("./data", f"{path}.json")
+    f = open(file, 'r')
     data = json.load(f)
     f.close()
     return data
 
+def writejson(data, path):
+    file = os.path.join("./data", f"{path}.json")
+    f = open(file, 'w')
+    json.dump(data, f)
+    f.close()
+
 def http_parser_request(html_string):
 
     map = dict()