87 lines
4.0 KiB
Python
87 lines
4.0 KiB
Python
from __future__ import absolute_import
|
|
|
|
import tempfile
|
|
|
|
from south.signals import post_migrate
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.contrib.auth import models as auth_models
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.auth.management import create_superuser
|
|
from django.dispatch import receiver
|
|
from django.db.models.signals import post_save
|
|
from django.db import transaction, DatabaseError
|
|
|
|
from navigation.api import register_links, register_top_menu
|
|
|
|
from .conf.settings import (AUTO_CREATE_ADMIN, AUTO_ADMIN_USERNAME,
|
|
AUTO_ADMIN_PASSWORD, TEMPORARY_DIRECTORY)
|
|
from .conf import settings as common_settings
|
|
from .utils import validate_path
|
|
from .models import AutoAdminSingleton
|
|
|
|
|
|
def has_usable_password(context):
|
|
return context['request'].user.has_usable_password
|
|
|
|
password_change_view = {'text': _(u'change password'), 'view': 'password_change_view', 'famfam': 'computer_key', 'condition': has_usable_password}
|
|
current_user_details = {'text': _(u'user details'), 'view': 'current_user_details', 'famfam': 'vcard'}
|
|
current_user_edit = {'text': _(u'edit details'), 'view': 'current_user_edit', 'famfam': 'vcard_edit'}
|
|
|
|
register_links(['current_user_details', 'current_user_edit', 'password_change_view'], [current_user_details, current_user_edit, password_change_view], menu_name='secondary_menu')
|
|
|
|
about_view = {'text': _('about'), 'view': 'about_view', 'famfam': 'information'}
|
|
license_view = {'text': _('license'), 'view': 'license_view', 'famfam': 'script'}
|
|
|
|
register_links(['about_view', 'license_view'], [about_view, license_view], menu_name='secondary_menu')
|
|
|
|
register_top_menu('about', link={'text': _(u'about'), 'view': 'about_view', 'famfam': 'information'}, position=-1)
|
|
|
|
|
|
@receiver(post_migrate, dispatch_uid='create_superuser')
|
|
def create_superuser(sender, **kwargs):
|
|
"""
|
|
From https://github.com/lambdalisue/django-qwert/blob/master/qwert/autoscript/__init__.py
|
|
From http://stackoverflow.com/questions/1466827/ --
|
|
|
|
Prevent interactive question about wanting a superuser created. (This code
|
|
has to go in this otherwise empty "models" module so that it gets processed by
|
|
the "syncdb" command during database creation.)
|
|
|
|
Create our own admin super user automatically.
|
|
"""
|
|
|
|
if AUTO_CREATE_ADMIN and kwargs['app'] == 'common':
|
|
try:
|
|
auth_models.User.objects.get(username=AUTO_ADMIN_USERNAME)
|
|
except auth_models.User.DoesNotExist:
|
|
print '*' * 80
|
|
print 'Creating super admin user -- login: %s, password: %s' % (AUTO_ADMIN_USERNAME, AUTO_ADMIN_PASSWORD)
|
|
print '*' * 80
|
|
assert auth_models.User.objects.create_superuser(AUTO_ADMIN_USERNAME, 'autoadmin@autoadmin.com', AUTO_ADMIN_PASSWORD)
|
|
admin = auth_models.User.objects.get(username=AUTO_ADMIN_USERNAME)
|
|
# Store the auto admin password properties to display the first login message
|
|
auto_admin_properties = AutoAdminSingleton.objects.get()
|
|
auto_admin_properties.account = admin
|
|
auto_admin_properties.password = AUTO_ADMIN_PASSWORD
|
|
auto_admin_properties.password_hash = admin.password
|
|
auto_admin_properties.save()
|
|
else:
|
|
print 'Super admin user already exists. -- login: %s' % AUTO_ADMIN_USERNAME
|
|
|
|
|
|
@receiver(post_save, dispatch_uid='auto_admin_account_passwd_change', sender=User)
|
|
def auto_admin_account_passwd_change(sender, instance, **kwargs):
|
|
with transaction.commit_on_success():
|
|
try:
|
|
auto_admin_properties = AutoAdminSingleton.objects.get()
|
|
if instance == auto_admin_properties.account and instance.password != auto_admin_properties.password_hash:
|
|
# Only delete the auto admin properties when the password has been changed
|
|
auto_admin_properties.delete(force=True)
|
|
except DatabaseError:
|
|
transaction.rollback()
|
|
|
|
|
|
if (validate_path(TEMPORARY_DIRECTORY) == False) or (not TEMPORARY_DIRECTORY):
|
|
setattr(common_settings, 'TEMPORARY_DIRECTORY', tempfile.mkdtemp())
|