mirror of
https://gitlab2.federez.net/re2o/re2o
synced 2024-12-23 23:43:47 +00:00
Ajout d'une commande chsh
This commit is contained in:
parent
beb7c48c56
commit
57833142eb
2 changed files with 81 additions and 0 deletions
77
users/management/commands/chsh.py
Executable file
77
users/management/commands/chsh.py
Executable file
|
@ -0,0 +1,77 @@
|
|||
# ⁻*- 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 Jean-Baptiste Daval
|
||||
#
|
||||
# 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.
|
||||
|
||||
import os, sys, pwd
|
||||
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
from django.db import transaction
|
||||
from reversion import revisions as reversion
|
||||
|
||||
from users.models import User, ListShell
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'Change the default shell of a user'
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument('target_username', nargs='?')
|
||||
|
||||
def handle(self, *args, **options):
|
||||
|
||||
def get_user(user_pseudo):
|
||||
"""Return the user queried by pseudo, and exit the script if not found."""
|
||||
user = User.objects.filter(pseudo=user_pseudo)
|
||||
if not user:
|
||||
raise CommandError("Utilisateur invalide")
|
||||
return user[0]
|
||||
|
||||
current_username = pwd.getpwuid(int(os.getenv("SUDO_UID") or os.getuid())).pw_name
|
||||
current_user = get_user(current_username)
|
||||
|
||||
target_username = options["target_username"] or current_username
|
||||
target_user = get_user(target_username)
|
||||
|
||||
#L'utilisateur n'a pas le droit de changer le shell
|
||||
ok, msg = target_user.can_change_shell(current_user)
|
||||
if not ok:
|
||||
raise CommandError(msg)
|
||||
|
||||
shells = ListShell.objects.all()
|
||||
self.stdout.write("Choisissez un shell pour l'utilisateur %s :" % target_user.pseudo)
|
||||
for shell in shells:
|
||||
self.stdout.write("%d - %s (%s)" % (shell.id, shell.get_pretty_name(), shell.shell))
|
||||
shell_id = input("Entrez un nombre : ")
|
||||
|
||||
try:
|
||||
shell_id = int(shell_id)
|
||||
except:
|
||||
raise CommandError("Choix invalide")
|
||||
|
||||
shell = ListShell.objects.filter(id=shell_id)
|
||||
if not shell:
|
||||
raise CommandError("Choix invalide")
|
||||
|
||||
target_user.shell = shell.first()
|
||||
with transaction.atomic(), reversion.create_revision():
|
||||
target_user.save()
|
||||
reversion.set_user(current_user)
|
||||
reversion.set_comment("Shell modifié")
|
||||
|
||||
self.stdout.write(self.style.SUCCESS("Shell modifié. La modification peut prendre quelques minutes pour s'appliquer."))
|
|
@ -1248,6 +1248,10 @@ class ListShell(models.Model):
|
|||
|
||||
shell = models.CharField(max_length=255, unique=True)
|
||||
|
||||
def get_pretty_name(self):
|
||||
"""Return the canonical name of the shell"""
|
||||
return self.shell.split("/")[-1]
|
||||
|
||||
def __str__(self):
|
||||
return self.shell
|
||||
|
||||
|
|
Loading…
Reference in a new issue