From 4e67b0356500c258560e5f0e9592cc2bd095d091 Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Mon, 23 Oct 2023 17:30:17 +0800
Subject: [PATCH] sytle: modify code

---
 src/client.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/client.py b/src/client.py
index c857f0d..303f19d 100644
--- a/src/client.py
+++ b/src/client.py
@@ -97,7 +97,7 @@ class C(BaseModel):
     Tuple: Tuple[capsule, int]
     ip: str
 
-# receive messages from node
+# receive messages from nodes
 @app.post("/receive_messages")
 async def receive_messages(message: C):
     """
@@ -181,13 +181,17 @@ async def send_messages(
 ):
     global pk, sk
     id_list = []
+    # calculate id of nodes
     for node_ip in node_ips:
         ip_parts = node_ip.split(".")
         id = 0
         for i in range(4):
             id += int(ip_parts[i]) << (24 - (8 * i))
         id_list.append(id)
+    
+    # generate rk
     rk_list = GenerateReKey(sk, pk_B, len(node_ips), shreshold, tuple(id_list))  # type: ignore
+    
     capsule_ct = Encrypt(pk, message)  # type: ignore
 
     for i in range(len(node_ips)):
@@ -315,8 +319,6 @@ def get_own_ip() -> str:
 # get node list from central server
 def get_node_list(count: int, server_addr: str):
     url = "http://" + server_addr + "/server/send_nodes_list?count=" + str(count)
-    # payload = {"count": count}
-    # response = requests.post(url, json=payload)
     response = requests.get(url)
     # Checking the response
     if response.status_code == 200: