bbohard
/
pexpectation
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

branchement conditionnel des questions

This commit is contained in:
Benjamin Bohard 2021-08-31 08:54:40 +02:00
parent 8ee744416c
commit 9887189b38
2 changed files with 62 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
eole_module_expectations/register_zephir_pexpectations.py
View File

@ -8,73 +8,52 @@ already_registered = Expectation("""1 -> Désinscrire ce serveur du serveur Zép
3 -> Ne rien faire 3 -> Ne rien faire
Entrez le numéro de votre choix :""", response='3', name='already_registered') Entrez le numéro de votre choix :""", response='3', name='already_registered')
network_configuration = Expectation(""" Procédure d'enregistrement sur le serveur Zéphir 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') 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') 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_address = Expectation("""adresse_ip {variable} :""", response='192.168.1.2', name='')
network_netmask = Expectation("""masque de réseau pour {interface} :""", response='255.255.255.0', name='network_netmask') 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') gateway = Expectation("""adresse de la passerelle :""", response='192.168.1.1', name='gateway')
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')
"""Entrez l'adresse (nom DNS) du serveur Zéphir :""" zephir_admin_password = Expectation("""Mot de passe pour l'application Zéphir pour {variable} :""", response='eole', name='zephir_admin_password')
"""Entrez votre login pour l'application Zéphir (rien pour sortir) :""" new_server = Expectation("""créer le serveur dans la base du serveur Zéphir (O/N) :""", response='N', name='new_server')
"""Mot de passe pour l'application Zéphir pour admin :""" rne = Expectation("""entrez le RNE de l'établissement correspondant au serveur,
(rien pour saisir directement un n° de serveur) :""", response='', name='rne')
"""créer le serveur dans la base du serveur Zéphir (O/N) :""" 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
"""entrez le RNE de l'établissement correspondant au serveur matériel (Standard PC (Q35 + ICH9, 2009) par défaut) :""", response='', name='hardware')
(rien pour saisir directement un n° de serveur) :""" 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')
"""entrez le n° identifiant le serveur l'application Zéphir :""" 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')
"""Mise à jour des informations sur le matériel final = Expectation("""1 -> Ne rien faire
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
2 -> Utiliser la configuration définie sur le serveur Zéphir 2 -> Utiliser la configuration définie sur le serveur Zéphir
3 -> Non disponible 3 -> Non disponible
4 -> Modifier la variante du serveur 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("""############################################################################# expectations.add_expectation(already_registered)
# Initialisation du mot de passe de l'administrateur de base (admin_zephir) # expectations.add_expectation(network_configuration)
############################################################################# expectations.add_expectation(new_server)
Mot de passe :""", response='eole', name='admin_zephir_password') 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') network_configuration.set_next_expectation(interface_name, triggers=['O', 'Oui', 'OUI'])
exp1.set_next_expectation(exp1_1) interface_name.set_next_expectation(network_address)
expectations.add_expectation(exp1) 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 new_server.set_next_expectation(rne, triggers=['N', 'Non', 'NON'])
Attribuez leur des droits sur l'application via l'interface web 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') hardware.set_next_expectation(processor)
exp2_2 = Expectation("""Saisissez à nouveau le mot de passe : """, response='eole', name='other_zephir_account_password') processor.set_next_expectation(harddrive)
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)

51
pexpectation.py
View File

@ -1,7 +1,6 @@
import re import re
import pexpect import pexpect
ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])') ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
class ExpectationCollection: class ExpectationCollection:
def __init__(self): def __init__(self):
self.expectations_lookup = {} self.expectations_lookup = {}
@ -24,8 +23,15 @@ class ExpectationCollection:
def get_expectations_by_pattern(self, pattern): def get_expectations_by_pattern(self, pattern):
return self.expectations_lookup.get(pattern, []) return self.expectations_lookup.get(pattern, [])
def get_expectation_by_name(self, name): def get_expectations_by_name(self, name):
return [expectation.name for expectation in self.get_expectations(recursive=True)] 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): def get_expectations(self, recursive=False):
expectations = [exp for exps in self.expectations_lookup.values() for exp in exps] expectations = [exp for exps in self.expectations_lookup.values() for exp in exps]
@ -35,7 +41,7 @@ class ExpectationCollection:
return expectations return expectations
def count_expectations(self): 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): def merge(self, collection):
for expectation in collection.get_expectations(): 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.context = [p.strip() for p in multiline_pattern.strip().split('\n') if p.strip()]
self.pattern = self.context[-1] self.pattern = self.context[-1]
self.response = response self.response = response
self.next = None self.next = {}
def get_pattern(self): def get_pattern(self, previous_answer=None):
return re.compile(f'(.*){re.escape(self.pattern)}(.*)') return re.compile(f'(.*){re.escape(self.pattern.format(variable=previous_answer))}(.*)')
def set_next_expectation(self, expectation): def set_next_expectation(self, expectation, triggers=None):
self.next = expectation if not isinstance(triggers, list):
triggers = [triggers]
for trigger in triggers:
self.next[trigger] = expectation
def set_response(self, response): def set_response(self, response):
print(f'Setting {response} for {self.pattern}')
self.response = response self.response = response
def expect(self, spawned): def expect(self, spawned, previous_answer=None):
print(f'-> expecting next "{self.pattern}"') print(f'-> expecting next "{self.pattern.format(variable=previous_answer}"')
p = spawned.expect([pexpect.EOF, pexpect.TIMEOUT, self.get_pattern()]) p = spawned.expect([pexpect.EOF, pexpect.TIMEOUT, self.get_pattern(previous_answer=previous_answer)])
if p not in [0, 1]: if p not in [0, 1]:
self.answer(spawned) self.answer(spawned)
else: else:
@ -71,8 +81,10 @@ class Expectation:
print(f'-> answering "{self.response}" to "{spawned.after}"') print(f'-> answering "{self.response}" to "{spawned.after}"')
if self.response is not None: if self.response is not None:
spawned.sendline(self.response) spawned.sendline(self.response)
if self.next: if self.response in self.next:
self.next.expect(spawned) 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): def is_the_one(self, context):
#print(f'testing if it is the one {self.get_pattern()} for {context}') #print(f'testing if it is the one {self.get_pattern()} for {context}')
@ -86,14 +98,7 @@ class Expectation:
return False return False
return True return True
def count_expectations(self):
count = 1
if self.next:
count += self.next.count_expectations()
return count
def get_following_expectations(self): def get_following_expectations(self):
if self.next: for next_expectation in self.next.values():
return [self.next] + self.next.get_following_expectations() return [next_expectation] + next_expectation.get_following_expectations()
return [] return []