diff --git a/modules/societegenerale/sgpe/pages.py b/modules/societegenerale/sgpe/pages.py index d49879551c9267f851666e82cefa62d2b182fad3..91eb231b800d447091fe86b0edbe98b7d9e8b7a1 100644 --- a/modules/societegenerale/sgpe/pages.py +++ b/modules/societegenerale/sgpe/pages.py @@ -141,8 +141,7 @@ def get_keyboard_infos(self): def get_keyboard_data(self): infos = self.get_keyboard_infos() - infos['grid'] = self.decode_grid(infos) - + infos['grid'] = self.get_grid_data(infos) url = self.get_url('/vk/gen_ui?modeClavier=0&cryptogramme=' + infos['crypto']) img = Captcha(BytesIO(self.browser.open(url).content), infos) @@ -158,6 +157,10 @@ def get_keyboard_data(self): 'img': img, } + def get_grid_data(self, infos): + # For ent the grid is already decoded + return infos['grid'] + def get_authentication_url(self): return self.browser.absurl('/authent.html') @@ -178,6 +181,10 @@ def login(self, login, password): class MainProPage(LoginEntPage): + def get_grid_data(self, infos): + # The grid are in b64 + return self.decode_grid(infos) + def get_authentication_url(self): return self.browser.absurl('/sec/vk/authent.json')