FreedomBox/modules/first_boot/middleware.py

26 lines
984 B
Python

import logging
from django.core.urlresolvers import reverse
from django.http.response import HttpResponseRedirect
from withsqlite.withsqlite import sqlite_db
import cfg
LOGGER = logging.getLogger(__name__)
class FirstBootMiddleware:
""" Forward to firstboot page if firstboot isn't finished yet """
def process_request(self, request):
with sqlite_db(cfg.store_file, table='firstboot') as database:
if not 'state' in database:
# Permanent redirect causes the browser to cache the redirect,
# preventing the user from navigating to /plinth until the
# browser is restarted.
return HttpResponseRedirect(reverse('first_boot:index'))
if database['state'] < 5:
LOGGER.info('First boot state - %d', database['state'])
return HttpResponseRedirect(reverse('first_boot:state%d' %
database['state']))