Ajout d'une commande chsh

users/management/commands/chsh.py

@@ -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."))

users/models.py

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