ytsm/app/YtManagerApp/views/first_time.py

import logging

from django.conf import settings
from django.contrib.auth import authenticate, login
from django.contrib.auth.models import User
from django.http import HttpResponseForbidden
from django.shortcuts import redirect
from django.urls import reverse_lazy
from django.views.generic import FormView

from YtManagerApp.management.appconfig import appconfig
from YtManagerApp.management.jobs.synchronize import SynchronizeJob
from YtManagerApp.scheduler import scheduler
from YtManagerApp.views.forms.first_time import WelcomeForm, ApiKeyForm, PickAdminUserForm, ServerConfigForm, DoneForm, \
    UserCreationForm, LoginForm

logger = logging.getLogger("FirstTimeWizard")


class WizardStepMixin:

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def get(self, request, *args, **kwargs):

        # Prevent access if application is already initialized
        if appconfig.initialized:
            logger.debug(f"Attempted to access {request.path}, but first time setup already run. Redirected to home "
                         f"page.")
            return redirect('home')

        return super().get(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        if appconfig.initialized:
            logger.debug(f"Attempted to post {request.path}, but first time setup already run.")
            return HttpResponseForbidden()
        return super().post(request, *args, **kwargs)


#
# Step 0: welcome screen
#
class Step0WelcomeView(WizardStepMixin, FormView):
    template_name = 'YtManagerApp/first_time_setup/step0_welcome.html'
    form_class = WelcomeForm
    success_url = reverse_lazy('first_time_1')

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context.update({
            'config_errors': settings.CONFIG_ERRORS,
            'config_warnings': settings.CONFIG_WARNINGS,
        })
        return context


#
# Step 1: setup API key
#
class Step1ApiKeyView(WizardStepMixin, FormView):
    template_name = 'YtManagerApp/first_time_setup/step1_apikey.html'
    form_class = ApiKeyForm
    success_url = reverse_lazy('first_time_2')

    def get_initial(self):
        initial = super().get_initial()
        initial['api_key'] = appconfig.youtube_api_key
        return initial

    def form_valid(self, form):
        key = form.cleaned_data['api_key']
        # TODO: validate key
        if key is not None and len(key) > 0:
            appconfig.youtube_api_key = key

        return super().form_valid(form)


#
# Step 2: create admin user
#
class Step2SetupAdminUserView(WizardStepMixin, FormView):
    template_name = 'YtManagerApp/first_time_setup/step2_admin.html'
    success_url = reverse_lazy('first_time_3')

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def get_form_class(self):
        have_users = User.objects.count() > 0
        have_admin = User.objects.filter(is_superuser=True).count() > 0

        # Check if an admin user already exists
        if have_admin:
            logger.debug("Admin user already exists and is not logged in!")
            return LoginForm

        elif have_users and 'register' not in self.kwargs:
            logger.debug("There are users but no admin!")
            return PickAdminUserForm

        logger.debug("No admin user exists, will register a new account!")
        return UserCreationForm

    def get(self, request, *args, **kwargs):

        # Skip if admin is already logged in
        if request.user.is_authenticated and request.user.is_superuser:
            logger.debug("Admin user already exists and is logged in!")
            return redirect(self.success_url)

        return super().get(request, *args, **kwargs)

    def form_valid(self, form):
        if isinstance(form, LoginForm):
            form.apply_session_expiry(self.request)
            login(self.request, form.get_user())

        elif isinstance(form, UserCreationForm):
            user = form.save()
            user.is_staff = True
            user.is_superuser = True
            user.save()

            username = form.cleaned_data.get('username')
            password = form.cleaned_data.get('password1')
            user = authenticate(username=username, password=password)
            login(self.request, user)

        elif isinstance(form, PickAdminUserForm):
            user = form.cleaned_data['admin_user']
            user.is_staff = True
            user.is_superuser = True
            user.save()

            return redirect('first_time_2', assigned_success='1')

        return super().form_valid(form)


#
# Step 3: configure server
#
class Step3ConfigureView(WizardStepMixin, FormView):
    template_name = 'YtManagerApp/first_time_setup/step3_configure.html'
    form_class = ServerConfigForm
    success_url = reverse_lazy('first_time_done')

    def get_initial(self):
        initial = super().get_initial()
        initial['allow_registrations'] = appconfig.allow_registrations
        initial['sync_schedule'] = appconfig.sync_schedule
        initial['auto_download'] = self.request.user.preferences['auto_download']
        initial['download_location'] = self.request.user.preferences['download_path']
        return initial

    def form_valid(self, form):
        allow_registrations = form.cleaned_data['allow_registrations']
        if allow_registrations is not None:
            appconfig.allow_registrations = allow_registrations

        sync_schedule = form.cleaned_data['sync_schedule']
        if sync_schedule is not None and len(sync_schedule) > 0:
            appconfig.sync_schedule = sync_schedule

        auto_download = form.cleaned_data['auto_download']
        if auto_download is not None:
            self.request.user.preferences['auto_download'] = auto_download

        download_location = form.cleaned_data['download_location']
        if download_location is not None and len(download_location) > 0:
            self.request.user.preferences['download_path'] = download_location

        # Set initialized to true
        appconfig.initialized = True

        # Start scheduler if not started
        scheduler.initialize()
        SynchronizeJob.schedule_global_job()

        return super().form_valid(form)


#
# Done screen
#
class DoneView(FormView):
    template_name = 'YtManagerApp/first_time_setup/done.html'
    form_class = DoneForm
    success_url = reverse_lazy('home')