Transfer hot fix

This commit is contained in:
Wong Yiek Heng 2025-10-06 07:26:38 +08:00
parent 706c5ee896
commit 5783a38faf

View File

@ -1,3 +1,4 @@
import datetime
import requests, re import requests, re
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
@ -439,7 +440,13 @@ class CM_BOT:
headers=self.get_user_credit_headers headers=self.get_user_credit_headers
) )
soup = BeautifulSoup(response.content, 'html.parser') soup = BeautifulSoup(response.content, 'html.parser')
try:
return float(soup.find('table', {'class': 'generalContent'}).find(text=re.compile('Credit Available')).parent.parent.find_all('td')[2].text) 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): def get_transfer_token(self):
response = self.session.get(f'{self.base_url}/cm/transfer') response = self.session.get(f'{self.base_url}/cm/transfer')