proper csrf method, remove temporary hack

dev
Thomas Lynch 2 years ago
parent a00ea88036
commit 043de1bda4
  1. 20
      session.py

@ -36,20 +36,12 @@ class ModSession(Session):
def update_csrf(self): def update_csrf(self):
try: try:
res = self.get(url=f'{self.imageboard_url}/csrf.json',
# res = self.get(url=f'{self.imageboard_url}/csrf.json', headers={'Referer': f'{self.imageboard_url}/csrf.json'}).json()
# headers={'Referer': f'{self.imageboard_url}/csrf.json'}).json() if 'token' in res:
# if 'token' in res: self.csrf_token = res['token']
# self.csrf_token = res['token'] else:
# else: raise Exception('Unable to update csrf token')
# raise Exception('Unable to update csrf token')
#temporary hack to get csrf token to test on <=0.1.10
res2 = self.get(url=f'{self.imageboard_url}/account.html',
headers={'Referer': f'{self.imageboard_url}/account.html'})
csrfi = res2.text.index('_csrf')
self.csrf_token = res2.text[csrfi+14:csrfi+50]
except requests.RequestException as e: except requests.RequestException as e:
logging.error(f'Exception {e} occurred while updating csrf token') logging.error(f'Exception {e} occurred while updating csrf token')
raise Exception('Unable to update csrf token') raise Exception('Unable to update csrf token')

Loading…
Cancel
Save