diff --git a/eole_module_expectations/register_zephir_pexpectations.py b/eole_module_expectations/register_zephir_pexpectations.py index c85eb2e..6fe9e3b 100644 --- a/eole_module_expectations/register_zephir_pexpectations.py +++ b/eole_module_expectations/register_zephir_pexpectations.py @@ -8,73 +8,52 @@ already_registered = Expectation("""1 -> Désinscrire ce serveur du serveur Zép 3 -> Ne rien faire Entrez le numéro de votre choix :""", response='3', name='already_registered') - network_configuration = Expectation(""" Procédure d'enregistrement sur le serveur Zéphir Voulez-vous établir une configuration réseau minimale (O/N) :""", response='N', name='network_configuration') - interface_name = Expectation("""interface connectée sur l'extérieur""", response='ens0', name='interface_name') -network_address = Expectation("""adresse_ip {interface} :""", response='192.168.1.2', name='') -network_netmask = Expectation("""masque de réseau pour {interface} :""", response='255.255.255.0', name='network_netmask') +network_address = Expectation("""adresse_ip {variable} :""", response='192.168.1.2', name='') +network_netmask = Expectation("""masque de réseau pour {variable} :""", response='255.255.255.0', name='network_netmask') gateway = Expectation("""adresse de la passerelle :""", response='192.168.1.1', name='gateway') - - -"""Entrez l'adresse (nom DNS) du serveur Zéphir :""" -"""Entrez votre login pour l'application Zéphir (rien pour sortir) :""" -"""Mot de passe pour l'application Zéphir pour admin :""" - -"""créer le serveur dans la base du serveur Zéphir (O/N) :""" - -"""entrez le RNE de l'établissement correspondant au serveur -(rien pour saisir directement un n° de serveur) :""" - -"""entrez le n° identifiant le serveur l'application Zéphir :""" - -"""Mise à jour des informations sur le matériel -matériel (Standard PC (Q35 + ICH9, 2009) par défaut) :""" -"""processeur ( Intel Core Processor (Skylake, IBRS) 3191 MHz par défaut) :""" -"""disque dur (19 Go par défaut) :""" - -"""(une procédure d'enregistrement à déjà eu lieu pour ce serveur) -continuer l'enregistrement (O/N) ?""" - -"""1 -> Ne rien faire +zephir_address = Expectation("""Entrez l'adresse (nom DNS) du serveur Zéphir :""", response='zephir', name='zephir_address') +zephir_admin = Expectation("""Entrez votre login pour l'application Zéphir (rien pour sortir) :""", response='admin_zephir', name='zephir_admin') +zephir_admin_password = Expectation("""Mot de passe pour l'application Zéphir pour {variable} :""", response='eole', name='zephir_admin_password') +new_server = Expectation("""créer le serveur dans la base du serveur Zéphir (O/N) :""", response='N', name='new_server') +rne = Expectation("""entrez le RNE de l'établissement correspondant au serveur, +(rien pour saisir directement un n° de serveur) :""", response='', name='rne') +server_id = Expectation("""entrez le n° identifiant le serveur l'application Zéphir :""", response='1', name='server_id') +hardware = Expectation("""Mise à jour des informations sur le matériel +matériel (Standard PC (Q35 + ICH9, 2009) par défaut) :""", response='', name='hardware') +processor = Expectation("""processeur ( Intel Core Processor (Skylake, IBRS) 3191 MHz par défaut) :""", response='', name='processor') +harddrive = Expectation("""disque dur (19 Go par défaut) :""", response='', name='harddrive') +key_available = Expectation("""(une procédure d'enregistrement à déjà eu lieu pour ce serveur) +continuer l'enregistrement (O/N) ?""", response='O', name='key_available') +final = Expectation("""1 -> Ne rien faire 2 -> Utiliser la configuration définie sur le serveur Zéphir 3 -> Non disponible 4 -> Modifier la variante du serveur - Entrez le numéro de votre choix :""" + Entrez le numéro de votre choix :""", response='2', name='final') -exp1 = Expectation("""############################################################################# -# Initialisation du mot de passe de l'administrateur de base (admin_zephir) # -############################################################################# -Mot de passe :""", response='eole', name='admin_zephir_password') +expectations.add_expectation(already_registered) +expectations.add_expectation(network_configuration) +expectations.add_expectation(new_server) +expectations.add_expectation(zephir_address) +expectations.add_expectation(hardware) +expectations.add_expectation(key_available) +expectations.add_expectation(final) -exp1_1 = Expectation("""Confirmation du mot de passe :""", response='eole', name='admin_zephir_password') -exp1.set_next_expectation(exp1_1) -expectations.add_expectation(exp1) +network_configuration.set_next_expectation(interface_name, triggers=['O', 'Oui', 'OUI']) +interface_name.set_next_expectation(network_address) +network_address.set_next_expectation(network_netmask) +network_netmask.set_next_expectation(gateway) -exp2 = Expectation("""Vous pouvez maintenant créer des utilisateurs si vous le souhaitez -Attribuez leur des droits sur l'application via l'interface web +new_server.set_next_expectation(rne, triggers=['N', 'Non', 'NON']) +rne.set_next_expectation(server_id, triggers='') -nom d'utilisateur a créer (rien pour terminer) : """, response='admin', name='other_zephir_account') +zephir_address.set_next_expectation(zephir_admin) +zephir_admin.set_next_expectation(zephir_admin_password) -exp2_1 = Expectation("""Mot de passe du nouvel utilisateur : """, response='eole', name='other_zephir_account_password') -exp2_2 = Expectation("""Saisissez à nouveau le mot de passe : """, response='eole', name='other_zephir_account_password') -exp2_3 = Expectation("""nom d'utilisateur a créer (rien pour terminer) : """, response='', name='empty_zephir_account') - -exp2_2.set_next_expectation(exp2_3) -exp2_1.set_next_expectation(exp2_2) -exp2.set_next_expectation(exp2_1) -expectations.add_expectation(exp2) - -expectations.add_expectation_from_descr("""* Vérification des données (md5) : Eole 2.6.1""") -expectations.add_expectation_from_descr("""* Vérification des données (md5) : Eole 2.6.2""") -expectations.add_expectation_from_descr("""* Vérification des données (md5) : Eole 2.7.0""") -expectations.add_expectation_from_descr("""* Vérification des données (md5) : Eole 2.7.1""") -expectations.add_expectation_from_descr("""* Vérification des données (md5) : Eole 2.7.2""") -expectations.add_expectation_from_descr("""* Vérification des données (md5) : Eole 2.8.0""") -expectations.add_expectation_from_descr("""Start Systemd services""") - -expectations.merge(eolebase_expectations) +hardware.set_next_expectation(processor) +processor.set_next_expectation(harddrive) diff --git a/pexpectation.py b/pexpectation.py index af00fe6..46875a5 100644 --- a/pexpectation.py +++ b/pexpectation.py @@ -1,7 +1,6 @@ import re import pexpect ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])') - class ExpectationCollection: def __init__(self): self.expectations_lookup = {} @@ -24,8 +23,15 @@ class ExpectationCollection: def get_expectations_by_pattern(self, pattern): return self.expectations_lookup.get(pattern, []) - def get_expectation_by_name(self, name): - return [expectation.name for expectation in self.get_expectations(recursive=True)] + def get_expectations_by_name(self, name): + return [expectation for expectation in self.get_expectations(recursive=True) if expectation.name == name] + + def get_exposed_expectation_names(self): + return [expectation.name for expectation in self.get_expectations(recursive=True) if expectation.name] + + def set_expectation_response_by_name(self, name, response): + for expectation in self.get_expectations_by_name(name): + expectation.set_response(response) def get_expectations(self, recursive=False): expectations = [exp for exps in self.expectations_lookup.values() for exp in exps] @@ -35,7 +41,7 @@ class ExpectationCollection: return expectations def count_expectations(self): - return sum([exp.count_expectations() for exp in self.get_expectations()]) + return len(self.get_expectations(recursive=True)) def merge(self, collection): for expectation in collection.get_expectations(): @@ -48,20 +54,24 @@ class Expectation: self.context = [p.strip() for p in multiline_pattern.strip().split('\n') if p.strip()] self.pattern = self.context[-1] self.response = response - self.next = None + self.next = {} - def get_pattern(self): - return re.compile(f'(.*){re.escape(self.pattern)}(.*)') + def get_pattern(self, previous_answer=None): + return re.compile(f'(.*){re.escape(self.pattern.format(variable=previous_answer))}(.*)') - def set_next_expectation(self, expectation): - self.next = expectation + def set_next_expectation(self, expectation, triggers=None): + if not isinstance(triggers, list): + triggers = [triggers] + for trigger in triggers: + self.next[trigger] = expectation def set_response(self, response): + print(f'Setting {response} for {self.pattern}') self.response = response - def expect(self, spawned): - print(f'-> expecting next "{self.pattern}"') - p = spawned.expect([pexpect.EOF, pexpect.TIMEOUT, self.get_pattern()]) + def expect(self, spawned, previous_answer=None): + print(f'-> expecting next "{self.pattern.format(variable=previous_answer}"') + p = spawned.expect([pexpect.EOF, pexpect.TIMEOUT, self.get_pattern(previous_answer=previous_answer)]) if p not in [0, 1]: self.answer(spawned) else: @@ -71,8 +81,10 @@ class Expectation: print(f'-> answering "{self.response}" to "{spawned.after}"') if self.response is not None: spawned.sendline(self.response) - if self.next: - self.next.expect(spawned) + if self.response in self.next: + self.next[self.response].expect(spawned, previous_answer=self.response) + elif None in self.next: + self.next[None].expect(spawned, previous_answer=self.response) def is_the_one(self, context): #print(f'testing if it is the one {self.get_pattern()} for {context}') @@ -86,14 +98,7 @@ class Expectation: return False return True - def count_expectations(self): - count = 1 - if self.next: - count += self.next.count_expectations() - return count - def get_following_expectations(self): - if self.next: - return [self.next] + self.next.get_following_expectations() + for next_expectation in self.next.values(): + return [next_expectation] + next_expectation.get_following_expectations() return [] -