[Django] clear_permissions.py

# -*- coding: utf-8 -*-

# @Author: Antonio Rodrigo <rodrigo>
# @Filename: clear_permissions.py
# @Last modified time: 27-12-2017

from __future__ import unicode_literals

# Django imports
from django.contrib.auth.models import Group
from django.core.management import BaseCommand

# App imports
from login.models import User

# standard library imports
from optparse import make_option
import os

class Command(BaseCommand):
    help = "This command removes all permissions from an user"

    def add_arguments(self, parser):
        parser.add_argument('username', nargs='+', type=str, help='username')

    def handle(self, *args, **options):
        username = options.get('username', 0)[0]

        if (username == '' or username == None):
            print('''Insert the user.\n
                     python manage.py django_clear_permissions [user]
                 ''')
            sys.exit(0)
        else:
            try:
                users = User.objects.get(username=username)
                users.user_permissions.clear()
                print("Permissions from user {} cleared successfully!".format(username))
            except User.DoesNotExist:
                print('Invalid user')