sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

feat: add init config

Browse Source
This commit is contained in:
sangge 2023-10-21 14:39:37 +08:00
parent 7b6e45690e
commit 459b03c872
2 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/client.ini Normal file
View File

@ -0,0 +1,3 @@
[settings]
server_address = "127.0.0.1:8000"
version = 1.0

24
src/client.py
View File

@ -18,13 +18,15 @@ app = FastAPI(lifespan=lifespan)
pk = point pk = point
sk = int sk = int
server_address = str
def init(): def init():
global pk, sk global pk, sk, server_address
init_db() init_db()
pk, sk = GenerateKeyPair() pk, sk = GenerateKeyPair()
get_node_list(6) init_config()
get_node_list(6, server_address) # type: ignore
def init_db(): def init_db():
@ -66,6 +68,16 @@ def init_db():
print("Init Database Successful") print("Init Database Successful")
def init_config():
import configparser
global server_address
config = configparser.ConfigParser()
config.read("client.ini")
server_address = config["settings"]["server_address"]
# execute on exit # execute on exit
def clean_env(): def clean_env():
print("Exit app") print("Exit app")
@ -86,7 +98,7 @@ async def receive_messages(C: Tuple[capsule, int], ip: str):
C: capsule and ct C: capsule and ct
ip: sender ip ip: sender ip
return: return:
status_code
""" """
if not C or not ip: if not C or not ip:
raise HTTPException(status_code=400, detail="Invalid input data") raise HTTPException(status_code=400, detail="Invalid input data")
@ -153,9 +165,9 @@ async def send_message(ip: tuple[str, ...]):
async def request_message(ip): async def request_message(ip):
return 0 return 0
# get node list from central server # get node list from central server
def get_node_list(count: int): def get_node_list(count: int, server_addr: str):
server_addr = ""
url = "http://" + server_addr + "/server/send_nodes_list" url = "http://" + server_addr + "/server/send_nodes_list"
payload = {"count": count} payload = {"count": count}
response = requests.post(url, json=payload) response = requests.post(url, json=payload)
@ -182,4 +194,4 @@ def get_node_list(count: int):
if __name__ == "__main__": if __name__ == "__main__":
import uvicorn # pylint: disable=e0401 import uvicorn # pylint: disable=e0401
uvicorn.run("client:app", host="0.0.0.0", port=8000) uvicorn.run("client:app", host="0.0.0.0", port=8003)