chsh.py 2.97 KB
Newer Older
Krokmou's avatar
Krokmou committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 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."))