diff --git a/modules/creditdunord/pages.py b/modules/creditdunord/pages.py index 58751dcb80106e6e09800674759e92ed663b998a..2f05dd8bb321785da3b70c83ca27083d94f586c1 100755 --- a/modules/creditdunord/pages.py +++ b/modules/creditdunord/pages.py @@ -173,12 +173,14 @@ def on_load(self): def get_labels(self): synthesis_labels = ["Synthèse"] - loan_labels = ["Crédits en cours", "Crédits perso et immo", "Crédits"] - for element in Dict('donnees/1/submenu')(self.doc): - if CleanText(Dict('label'))(element) in synthesis_labels: - synthesis_label = CleanText(Dict('link'))(element).split("/")[-1] - if CleanText(Dict('label'))(element) in loan_labels: - loan_label = CleanText(Dict('link'))(element).split("/")[-1] + loan_labels = ["Crédits en cours", "Crédits perso et immo", "Crédits", "Crédits Personnels et immobiliers"] + keys = [key for key in Dict('donnees')(self.doc) if key.get('label') in ['Crédits', 'Comptes et cartes']] + for key in keys: + for element in Dict('submenu')(key): + if CleanText(Dict('label'))(element) in synthesis_labels: + synthesis_label = CleanText(Dict('link'))(element).split("/")[-1] + if CleanText(Dict('label'))(element) in loan_labels: + loan_label = CleanText(Dict('link'))(element).split("/")[-1] return (synthesis_label, loan_label)