From 5783a38faf323497eeeb90e7b5c5e8fecfe31004 Mon Sep 17 00:00:00 2001 From: Wong Yiek Heng Date: Mon, 6 Oct 2025 07:26:38 +0800 Subject: [PATCH] Transfer hot fix --- cm_bot.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/cm_bot.py b/cm_bot.py index 30a63a0..5f2c88f 100644 --- a/cm_bot.py +++ b/cm_bot.py @@ -1,3 +1,4 @@ +import datetime import requests, re from bs4 import BeautifulSoup @@ -439,7 +440,13 @@ class CM_BOT: headers=self.get_user_credit_headers ) soup = BeautifulSoup(response.content, 'html.parser') - return float(soup.find('table', {'class': 'generalContent'}).find(text=re.compile('Credit Available')).parent.parent.find_all('td')[2].text) + try: + return float(soup.find('table', {'class': 'generalContent'}).find(text=re.compile('Credit Available')).parent.parent.find_all('td')[2].text) + except: + print(f"Error getting credit.") + with open(f'{datetime.datetime.now().strftime('%Y%m%d_%H%M')}.html', 'wb') as f: + f.write(response.content) + return 0 def get_transfer_token(self): response = self.session.get(f'{self.base_url}/cm/transfer')