tpre-python/src/client_cli.py
Smart-SangGe bfb5c27bcb
All checks were successful
Deploy App / Deploy to Web Server (push) Successful in 1m28s
fix: fix all bug
2023-10-26 11:30:03 +08:00

42 lines
1.1 KiB
Python

import argparse
import requests
import json
def send_post_request(ip_addr, message_name):
url = f"http://localhost:8002/request_message"
data = {"dest_ip": ip_addr, "message_name": message_name}
response = requests.post(url, json=data)
return response.text
def get_pk(ip_addr):
url = f"http://" + ip_addr + ":8002/get_pk"
response = requests.get(url, timeout=1)
print(response.text)
json_pk = json.loads(response.text)
payload = {"pkx": json_pk["pkx"], "pky": json_pk["pky"], "ip": ip_addr}
response = requests.post("http://localhost:8002/recieve_pk", json=payload)
return response.text
def main():
parser = argparse.ArgumentParser(description="Send POST request to a specified IP.")
parser.add_argument("ip_addr", help="IP address to send request to.")
parser.add_argument("message_name", help="Message name to send.")
args = parser.parse_args()
response = get_pk(args.ip_addr)
print(response)
response = send_post_request(args.ip_addr, args.message_name)
print(response)
if __name__ == "__main__":
main()