File tree Expand file tree Collapse file tree 1 file changed +10
-7
lines changed Expand file tree Collapse file tree 1 file changed +10
-7
lines changed Original file line number Diff line number Diff line change @@ -179,13 +179,16 @@ def __init__(self, config):
179
179
180
180
def login (self ):
181
181
def get_session_key (soup ):
182
- try :
183
- session_key = soup .find ("input" , {"name" : "sesskey" })["value" ]
184
- except TypeError :
185
- logger .critical ("Can't retrieve session key" )
186
- logger .info (soup )
187
- sys .exit (1 )
188
- return session_key
182
+ script = soup .find ("script" , string = lambda text : text and 'sesskey' in text )
183
+ js_text = script .text
184
+ match = re .search (r'"sesskey":"(.*?)"' , js_text )
185
+ if match :
186
+ return match .group (1 )
187
+ else :
188
+ logger .critical ("Can't retrieve session key from JavaScript config" )
189
+ exit (1 )
190
+
191
+
189
192
190
193
self .session = requests .Session ()
191
194
cookie_file = Path (self .config .get ("cookie_file" , "./session" ))
You can’t perform that action at this time.
0 commit comments