mirror of
https://gitlab.federez.net/re2o/re2o
synced 2024-05-20 09:32:29 +00:00
100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
# ⁻*- mode: python; coding: utf-8 -*-
|
|
# Re2o est un logiciel d'administration développé initiallement au rezometz. Il
|
|
# se veut agnostique au réseau considéré, de manière à être installable en
|
|
# quelques clics.
|
|
#
|
|
# Copyright © 2018 Lev-Arcady Sellem
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
"""re2o.script_utils
|
|
A set of utility scripts that can be used as standalone to interact easily
|
|
with Re2o throught the CLI
|
|
"""
|
|
|
|
import os
|
|
from os.path import dirname
|
|
import sys
|
|
import pwd
|
|
|
|
from getpass import getpass
|
|
from reversion import revisions as reversion
|
|
|
|
from django.core.wsgi import get_wsgi_application
|
|
from django.core.management.base import CommandError
|
|
from django.db import transaction
|
|
from django.utils.html import strip_tags
|
|
|
|
from users.models import User
|
|
|
|
proj_path = dirname(dirname(__file__))
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "re2o.settings")
|
|
sys.path.append(proj_path)
|
|
os.chdir(proj_path)
|
|
|
|
application = get_wsgi_application()
|
|
|
|
|
|
def get_user(pseudo):
|
|
"""Cherche un utilisateur re2o à partir de son pseudo"""
|
|
user = User.objects.filter(pseudo=pseudo)
|
|
if len(user) == 0:
|
|
raise CommandError("Utilisateur invalide")
|
|
if len(user) > 1:
|
|
raise CommandError("Plusieurs utilisateurs correspondant à ce "
|
|
"pseudo. Ceci NE DEVRAIT PAS arriver")
|
|
return user[0]
|
|
|
|
|
|
def get_system_user():
|
|
"""Retourne l'utilisateur système ayant lancé la commande"""
|
|
return pwd.getpwuid(int(os.getenv("SUDO_UID") or os.getuid())).pw_name
|
|
|
|
|
|
def form_cli(Form, user, action, *args, **kwargs):
|
|
"""
|
|
Remplit un formulaire à partir de la ligne de commande
|
|
Form : le formulaire (sous forme de classe) à remplir
|
|
user : l'utilisateur re2o faisant la modification
|
|
action : l'action réalisée par le formulaire (pour les logs)
|
|
Les arguments suivants sont transmis tels quels au formulaire.
|
|
"""
|
|
data = {}
|
|
dumb_form = Form(user=user, *args, **kwargs)
|
|
for key in dumb_form.fields:
|
|
if not dumb_form.fields[key].widget.input_type == 'hidden':
|
|
if dumb_form.fields[key].widget.input_type == 'password':
|
|
data[key] = getpass("%s : " % dumb_form.fields[key].label)
|
|
else:
|
|
data[key] = input("%s : " % dumb_form.fields[key].label)
|
|
|
|
form = Form(data, user=user, *args, **kwargs)
|
|
if not form.is_valid():
|
|
sys.stderr.write("Erreurs : \n")
|
|
for err in form.errors:
|
|
# Oui, oui, on gère du HTML là où d'autres ont eu la
|
|
# lumineuse idée de le mettre
|
|
sys.stderr.write(
|
|
"\t%s : %s\n" % (err, strip_tags(form.errors[err]))
|
|
)
|
|
raise CommandError("Formulaire invalide")
|
|
|
|
with transaction.atomic(), reversion.create_revision():
|
|
form.save()
|
|
reversion.set_user(user)
|
|
reversion.set_comment(action)
|
|
|
|
sys.stdout.write("%s : effectué. La modification peut prendre "
|
|
"quelques minutes pour s'appliquer.\n" % action)
|