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
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)

51
pexpectation.py
View File

@ -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 []