Coverage for sfkit/cli.py: 100%
29 statements
« prev ^ index » next coverage.py v7.2.7, created at 2023-08-01 13:35 -0400
« prev ^ index » next coverage.py v7.2.7, created at 2023-08-01 13:35 -0400
1from sfkit.auth.auth import auth
2from sfkit.auth.setup_networking import setup_networking
3from sfkit.encryption.generate_personal_keys import generate_personal_keys
4from sfkit.parser import get_parser
5from sfkit.protocol.register_data import register_data
6from sfkit.protocol.run_protocol import run_protocol
9def main() -> None:
10 parser = get_parser()
11 args = parser.parse_args()
13 if args.command == "auth":
14 auth()
15 elif args.command == "networking":
16 ports = args.ports or ""
17 ip_address = args.ip_address or ""
18 setup_networking(ports, ip_address)
19 elif args.command == "generate_keys":
20 generate_personal_keys()
21 elif args.command == "register_data":
22 geno_binary_file_prefix = args.geno_binary_file_prefix or ""
23 data_path = args.data_path or ""
24 register_data(geno_binary_file_prefix, data_path)
25 elif args.command == "run_protocol":
26 phase: str = "" # args.phase or ""
27 demo: bool = args.demo or False
28 visualize_results: str = args.visualize_results or ""
29 results_path: str = args.results_path or ""
30 retry: bool = args.retry or False
31 run_protocol(phase, demo, visualize_results, results_path, retry)
32 else:
33 parser.print_help()