File tree 1 file changed +7
-5
lines changed
1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -16,10 +16,11 @@ def main():
16
16
Main function primarly used for the command line interface
17
17
"""
18
18
parser = argparse .ArgumentParser (prog = "pyauthenticator" )
19
+ config_dict = load_config ()
19
20
parser .add_argument (
20
21
"service" ,
21
22
help = "Service to generate optauth code for. Available services are: "
22
- + str (list_services (config_dict = load_config () )),
23
+ + str (list_services (config_dict = config_dict )),
23
24
)
24
25
parser .add_argument (
25
26
"-qr" ,
@@ -34,14 +35,15 @@ def main():
34
35
)
35
36
args = parser .parse_args ()
36
37
if args .qrcode :
37
- generate_qrcode (key = args .service , config_dict = load_config () )
38
+ generate_qrcode (key = args .service , config_dict = config_dict )
38
39
elif args .add :
39
40
add_service (
40
- key = args .service , qrcode_png_file_name = args .add , config_dict = load_config ()
41
+ key = args .service , qrcode_png_file_name = args .add , config_dict = config_dict
41
42
)
42
43
print (args .service , "added." )
43
44
else :
44
- print (get_two_factor_code (key = args .service , config_dict = load_config () ))
45
+ print (get_two_factor_code (key = args .service , config_dict = config_dict ))
45
46
46
47
47
- main ()
48
+ if __name__ == "__main__" :
49
+ main ()
You can’t perform that action at this time.
0 commit comments