FreedomBox/modules/first_boot/middleware.py
2014-07-29 18:16:54 +03:00

30 lines
1.2 KiB
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):
# TODO: there should be a better way to tell if we're at the
# firstboot module
if "firstboot" not in request.path:
with sqlite_db(cfg.store_file, table='firstboot') as database:
first_boot_url = reverse('first_boot:index')
if 'state' not 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(first_boot_url)
elif database['state'] < 5:
LOGGER.info('First boot state - %d', database['state'])
return HttpResponseRedirect(reverse('first_boot:state%d' %
database['state']))