yiekheng/cm_bot_v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

+ Add error handler for set security pin

Browse Source
This commit is contained in:
Wong Yiek Heng 2025-10-04 11:06:28 +08:00
parent ad85ec0bfc
commit 4f309f40e6
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cm_bot.py
View File

@ -410,6 +410,9 @@ class CM_BOT:
data=security_data, data=security_data,
headers=self.set_security_pin_headers headers=self.set_security_pin_headers
) )
if 'java.lang.NullPointerException' in response.text:
return False
return True
def transfer_credit(self, t_username: str, t_password: str, amount: float): def transfer_credit(self, t_username: str, t_password: str, amount: float):
token = self.get_transfer_token() token = self.get_transfer_token()

4
cm_bot_hal.py
View File

@ -137,7 +137,9 @@ class CM_BOT_HAL:
username = f_username, username = f_username,
password = password password = password
) )
cm_bot.set_security_pin(self.security_pin) if cm_bot.set_security_pin(self.security_pin) == False:
cm_bot.logout()
raise Exception(f'Agent acc: {f_username} already has security pin!')
cm_bot.logout() cm_bot.logout()
result = self.update_user_status_to_done(f_username) result = self.update_user_status_to_done(f_username)