Merge branch 'master' of https://gitlab.adm.crans.org/nounous/scripts
This commit is contained in:
Pauline Pommeret
commit
5201e3dfdf
2 changed files with 167 additions and 110 deletions
|
|
@ -6,29 +6,29 @@ Copyright (C) Frédéric Pauget
|
|||
Licence : GPLv2
|
||||
"""
|
||||
|
||||
import sys
|
||||
import re
|
||||
import os
|
||||
import tempfile
|
||||
import re
|
||||
import shlex
|
||||
import subprocess
|
||||
import sys
|
||||
if not "/usr/scripts" in sys.path:
|
||||
sys.path.append("/usr/scripts")
|
||||
import cranslib.cransstrings
|
||||
|
||||
# Détermination de l'encodage
|
||||
encoding = None
|
||||
ENCODING = None
|
||||
try:
|
||||
from locale import getpreferredencoding
|
||||
encoding = getpreferredencoding()
|
||||
ENCODING = getpreferredencoding()
|
||||
except:
|
||||
pass
|
||||
|
||||
if not encoding:
|
||||
encoding = sys.stdin.encoding or 'UTF-8'
|
||||
if not ENCODING:
|
||||
ENCODING = sys.stdin.encoding or 'UTF-8'
|
||||
|
||||
# Si aucune locale n'est définie, on se met en...
|
||||
if encoding == "ANSI_X3.4-1968":
|
||||
encoding = 'UTF-8'
|
||||
if ENCODING == "ANSI_X3.4-1968":
|
||||
ENCODING = 'UTF-8'
|
||||
|
||||
if 'TERM' in os.environ and os.environ['TERM'] != 'unknown':
|
||||
el = subprocess.Popen('tput cr ; tput el', shell=True, stdout=subprocess.PIPE).stdout.read()
|
||||
|
|
@ -41,44 +41,89 @@ except (IOError, AttributeError):
|
|||
stdout_atty = False
|
||||
|
||||
def dialog(backtitle, arg, dialogrc=''):
|
||||
""" Affiche la boite de dialogue défine avec les arguments fournis
|
||||
(cf man dialog)
|
||||
si tout se déroule bien retourne :
|
||||
[ 0, [ reponse(s) ] ]
|
||||
si annulatin retourne :
|
||||
[ 1, [] ]
|
||||
si appui sur ESC demande confirmation de l'abandon et exécute sys.exit(0)
|
||||
si erreur dans les arguments raise RuntimeError
|
||||
"""
|
||||
f = tempfile.NamedTemporaryFile()
|
||||
cmd = u'%s /usr/bin/dialog --backtitle "%s" --cancel-label "Retour" %s 2>%s' % (dialogrc, backtitle,arg,f.name)
|
||||
res = os.system(cmd.encode(encoding, 'ignore'))
|
||||
Affiche la boîte de dialogue définie avec les arguments fournis
|
||||
(cf man dialog)
|
||||
|
||||
if res == 256 :
|
||||
# Annuler
|
||||
f.close()
|
||||
return [ 1, [] ]
|
||||
# Lecture du fichier de résultat et effacement
|
||||
try:
|
||||
result= f.readlines()
|
||||
f.close()
|
||||
except :
|
||||
result = [ "n'importe quoi", '']
|
||||
res = 65280
|
||||
# Traitement
|
||||
if res==65280 and result:
|
||||
# Erreur dans les arguments
|
||||
raise RuntimeError( arg, result[1].strip() )
|
||||
elif res==65280 :
|
||||
# Appui sur ESC
|
||||
arg1 = u'--title "Annulation" --yesno "Quitter ?\nLes dernières modifications seront perdues." 6 48'
|
||||
print backtitle
|
||||
cmd = u'%s /usr/bin/dialog --backtitle "%s" %s' % (dialogrc, backtitle,arg1)
|
||||
res = os.system(cmd.encode(encoding ,'ignore') )
|
||||
if res==0 : sys.exit(0)
|
||||
else : return dialog(backtitle,arg)
|
||||
elif not result : result=['']
|
||||
return [ 0, result ]
|
||||
Paramètres:
|
||||
- ``backtitle`` : (str) texte de l'en-tête, situé en haut à gauche
|
||||
- ``arg`` : (str ou list) ensemble des paramètres à passer au
|
||||
programme dialog
|
||||
- ``dialogrc`` : (str) chemin vers le dialogrc à utiliser
|
||||
|
||||
Utilises:
|
||||
- :py:var:`os.environ`
|
||||
- :py:func:`shlex.split`
|
||||
- :py:class:`subprocess.Popen`
|
||||
- :py:func:`sys.exit`
|
||||
|
||||
Peut lever:
|
||||
- :py:exc:`RuntimeError`
|
||||
|
||||
Retournes:
|
||||
- soit un tuple dont le premier élément est le code d'erreur (0 ou 1) et
|
||||
le second la sortie du programme dialog (une liste ou une liste vide)
|
||||
- soit rien du tout parce que sys.exit(0)
|
||||
"""
|
||||
# Tant que gest_crans (au moins) n'a pas été modifié pour que arg soit une
|
||||
# liste, ce kludge est nécessaire.
|
||||
if not isinstance(arg, list):
|
||||
arg = shlex.split(to_encoding(arg))
|
||||
|
||||
commande = [
|
||||
"/usr/bin/dialog",
|
||||
"--backtitle", backtitle,
|
||||
"--cancel-label", "Retour"
|
||||
] + arg
|
||||
|
||||
# On copie l'environnement afin de pouvoir écraser une variable pour
|
||||
# l'appel au programme dialog sans que cela affecte les autres processus.
|
||||
environnement = os.environ.copy()
|
||||
|
||||
# Si dialog est fourni, on modifie effectivement l'environnement
|
||||
if dialogrc:
|
||||
environnement["DIALOGRC"] = dialogrc
|
||||
|
||||
# Le résultat du programme dialog est dans stderr, on la récupère dans
|
||||
# un pipe
|
||||
processus = subprocess.Popen(commande, stderr=subprocess.PIPE, env=environnement)
|
||||
|
||||
# Récupération du contenu du pipe
|
||||
_, sortie = processus.communicate()
|
||||
resultat = sortie.splitlines()
|
||||
|
||||
# Récupération du code d'erreur
|
||||
code_erreur = processus.returncode
|
||||
|
||||
if code_erreur == 1:
|
||||
# L'utilisateur a annulé
|
||||
return (1, [])
|
||||
elif code_erreur == 255 and resultat:
|
||||
# La liste des arguments passés au programme dialog est invalide
|
||||
raise RuntimeError(arg, resultat[1].strip())
|
||||
elif code_erreur == 255:
|
||||
# L'utilisateur a appuyé sur ÉCHAP. Dans ce cas, on lui demande de
|
||||
# confirmer.
|
||||
# "0" et "0" sont là pour ajuster la taille de la fenêtre à son
|
||||
# contenu
|
||||
arg_confirmation = [
|
||||
"--title", "Annulation",
|
||||
"--yesno", "Quitter ?\nLes dernières modifications seront perdues.",
|
||||
"0", "0"
|
||||
]
|
||||
code_confirmation, _ = dialog(backtitle, arg_confirmation, dialogrc)
|
||||
if code_confirmation == 0:
|
||||
# L'utilisateur veut quitter (vraiment).
|
||||
sys.exit(0)
|
||||
else:
|
||||
# L'appui sur ÉCHAP était une fausse manip, on recommence
|
||||
return dialog(backtitle, arg, dialogrc)
|
||||
elif not resultat:
|
||||
# Quand le code d'erreur est nul, il ne faut pas que resultat soit
|
||||
# vide.
|
||||
resultat = ['']
|
||||
|
||||
return (0, resultat)
|
||||
|
||||
def coul(txt, col=None, dialog=False):
|
||||
"""
|
||||
|
|
@ -98,8 +143,7 @@ def coul(txt, col=None, dialog=False):
|
|||
'cyan': 36,
|
||||
'gris': 30,
|
||||
'gras': 50}
|
||||
codecol_dialog = {
|
||||
'rouge': 1,
|
||||
codecol_dialog = {'rouge': 1,
|
||||
'vert': 2,
|
||||
'jaune': 3,
|
||||
'bleu': 4,
|
||||
|
|
@ -127,7 +171,7 @@ OK = coul('OK', 'vert')
|
|||
WARNING = coul('WARNING', 'jaune')
|
||||
ERREUR = coul('ERREUR', 'rouge')
|
||||
|
||||
def to_unicode(txt, enc=encoding):
|
||||
def to_unicode(txt, enc=ENCODING):
|
||||
if isinstance(txt, unicode):
|
||||
return txt
|
||||
else:
|
||||
|
|
@ -139,7 +183,7 @@ def to_unicode(txt, enc=encoding):
|
|||
# donc on le décode en utf-8 \o/
|
||||
return txt.decode("UTF-8")
|
||||
|
||||
def to_encoding(txt, enc=encoding):
|
||||
def to_encoding(txt, enc=ENCODING):
|
||||
return to_unicode(txt).encode(enc, 'ignore')
|
||||
|
||||
def cprint(txt, col='blanc', newline=True):
|
||||
|
|
@ -316,7 +360,7 @@ def prompt(prompt, defaut=u'', couleur='gras'):
|
|||
# On laisse néanmoins la possibilité de sortir sur un Ctrl+C
|
||||
while True:
|
||||
try:
|
||||
v = raw_input(prompt_s.encode(encoding))
|
||||
v = raw_input(prompt_s.encode(ENCODING))
|
||||
v = cranslib.cransstrings.decode_dammit(v).strip()
|
||||
if not v:
|
||||
v = defaut
|
||||
|
|
|
|||
|
|
@ -53,19 +53,31 @@ isbureau = u'Bureau' in droits
|
|||
encoding = sys.stdin.encoding or 'UTF-8'
|
||||
|
||||
if u'Nounou' in droits:
|
||||
# Si on est nounou
|
||||
if os.path.exists(os.path.expanduser('~/.dialogrc')):
|
||||
dialogrc='~/.dialogrc'
|
||||
# Si on a un fichier de configuration du programme dialog dans son
|
||||
# HOME, alors on récupère son chemin.
|
||||
DIALOGRC = '~/.dialogrc'
|
||||
else:
|
||||
dialogrc='/etc/dialog.rc'
|
||||
dlg = dialog.Dialog(DIALOGRC=dialogrc)
|
||||
dialog_theme='DIALOGRC='+dialogrc
|
||||
# Sinon on utilise celui du système.
|
||||
DIALOGRC = '/etc/dialog.rc'
|
||||
|
||||
dlg = dialog.Dialog(DIALOGRC=DIALOGRC)
|
||||
else:
|
||||
# Si on est pas nounou, on est libre de faire ce que l'on veut avec la
|
||||
# variable d'environnement DIALOGRC.
|
||||
DIALOGRC = ''
|
||||
|
||||
dlg = dialog.Dialog()
|
||||
dialog_theme=''
|
||||
|
||||
|
||||
def dialog(arg):
|
||||
return affich_tools.dialog(u'Gestion des adhérents et machines du Crans', arg, dialog_theme)
|
||||
"""
|
||||
Raccourci permettant d'appeler :py:func:`affich_tools.dialog`.
|
||||
|
||||
.. seealso:: :py:func:`affich_tools.dialog`
|
||||
"""
|
||||
return affich_tools.dialog(u'Gestion des adhérents et machines du Crans', arg, DIALOGRC)
|
||||
|
||||
in_facture = None
|
||||
#########################################################################
|
||||
|
|
@ -92,7 +104,8 @@ def set_bases(adher):
|
|||
|
||||
# Affichage
|
||||
annul, result = dialog(arg)
|
||||
if annul: return 1
|
||||
if annul:
|
||||
return 1
|
||||
|
||||
# Traitement
|
||||
err = ''
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue