11import argparse
22import json
33import logging
4+ from helper .utils import hide_email
45from msreward .msr import MSR
56import os
67import platform
@@ -69,7 +70,8 @@ def parse_args():
6970 action = 'store_true' ,
7071 dest = 'exit_on_finish' ,
7172 default = False ,
72- help = f'Script will exit when finishes, otherwise will remain open and wait for user to press enter to end.' )
73+ help = 'Script will exit when finishes, otherwise will remain open and wait for user to press enter to end.' )
74+
7375 _parser = arg_parser .parse_args ()
7476 if _parser .all_mode :
7577 _parser .mobile_mode = True
@@ -82,8 +84,7 @@ def get_login_info():
8284 with open ('ms_rewards_login_dict.json' , 'r' ) as f :
8385 return json .load (f )
8486
85-
86- if __name__ == '__main__' :
87+ def run_bot ():
8788 check_python_version ()
8889 if os .path .exists ("drivers/chromedriver.exe" ):
8990 update_driver ()
@@ -101,15 +102,15 @@ def get_login_info():
101102 login_cred = get_login_info ()
102103 logging .info (msg = f'logins retrieved, { len (login_cred )} account(s):' )
103104 for cred in login_cred :
104- logging .info (msg = f'{ cred ["email" ]} ' )
105+ logging .info (msg = f'{ hide_email ( cred ["email" ]) } ' )
105106
106107 msrs = [MSR (x ['email' ], x ['password' ], x ['secret' ] if 'secret' in x else None , parser .headless )
107108 for x in login_cred ]
108109
109110 for msr in msrs :
110111 logging .info (
111112 msg = '--------------------------------------------------' )
112- logging .info (msg = f'Current account: { msr .email } ' )
113+ logging .info (msg = f'Current account: { hide_email ( msr .email ) } ' )
113114 msr .work (flag_pc = parser .pc_mode , flag_mob = parser .mobile_mode ,
114115 flag_quiz = parser .quiz_mode )
115116
@@ -121,3 +122,7 @@ def get_login_info():
121122 logging .info (msg = '--------------------------------------------------' )
122123 if not parser .exit_on_finish :
123124 input ('Press any key to exit...' )
125+
126+
127+ if __name__ == '__main__' :
128+ run_bot ()
0 commit comments