Coverage for sfkit/cli.py: 100%

29 statements  

« 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 

7 

8 

9def main() -> None: 

10 parser = get_parser() 

11 args = parser.parse_args() 

12 

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()