2017-02-08 20:23:21 +00:00
|
|
|
from django.contrib.contenttypes.models import ContentType
|
2016-10-25 20:04:24 +00:00
|
|
|
from django.contrib.auth.models import Permission
|
|
|
|
from django.contrib.auth.models import User, Group
|
2017-02-08 19:53:55 +00:00
|
|
|
from django.core.exceptions import ObjectDoesNotExist
|
2016-10-25 20:04:24 +00:00
|
|
|
from django.db.utils import ProgrammingError
|
2017-02-08 19:53:55 +00:00
|
|
|
from guardian.shortcuts import assign_perm
|
|
|
|
|
|
|
|
from nodeodm.models import ProcessingNode
|
2017-05-19 20:15:26 +00:00
|
|
|
from . import scheduler
|
|
|
|
import logging
|
2017-02-08 19:53:55 +00:00
|
|
|
from .models import Task
|
2017-02-08 20:23:21 +00:00
|
|
|
from webodm import settings
|
2017-05-19 20:15:26 +00:00
|
|
|
from webodm.wsgi import booted
|
2017-02-08 19:53:55 +00:00
|
|
|
|
2016-10-07 23:07:47 +00:00
|
|
|
|
2016-10-25 20:04:24 +00:00
|
|
|
def boot():
|
2017-05-19 20:15:26 +00:00
|
|
|
# booted is a shared memory variable to keep track of boot status
|
|
|
|
# as multiple workers could trigger the boot sequence twice
|
|
|
|
if booted.value: return
|
|
|
|
|
|
|
|
booted.value = True
|
2016-10-07 23:07:47 +00:00
|
|
|
logger = logging.getLogger('app.logger')
|
2017-05-19 20:15:26 +00:00
|
|
|
logger.info("Booting...")
|
|
|
|
|
|
|
|
if settings.DEBUG:
|
|
|
|
logger.warning("Debug mode is ON (for development this is OK)")
|
2016-10-07 23:07:47 +00:00
|
|
|
|
|
|
|
# Check default group
|
2016-10-08 01:00:34 +00:00
|
|
|
try:
|
|
|
|
default_group, created = Group.objects.get_or_create(name='Default')
|
|
|
|
if created:
|
|
|
|
logger.info("Created default group")
|
2016-10-07 23:07:47 +00:00
|
|
|
|
2017-02-08 19:53:55 +00:00
|
|
|
# Assign viewprocessing node object permission to default processing node (if present)
|
|
|
|
# Otherwise non-root users will not be able to process
|
|
|
|
try:
|
|
|
|
pnode = ProcessingNode.objects.get(hostname="node-odm-1")
|
|
|
|
assign_perm('view_processingnode', default_group, pnode)
|
|
|
|
logger.info("Added view_processingnode permissions to default group")
|
|
|
|
except ObjectDoesNotExist:
|
|
|
|
pass
|
|
|
|
|
2017-02-08 20:23:21 +00:00
|
|
|
|
2016-10-12 22:18:37 +00:00
|
|
|
# Add default permissions (view_project, change_project, delete_project, etc.)
|
2016-10-13 16:21:12 +00:00
|
|
|
for permission in ('_project', '_task'):
|
2016-10-08 01:00:34 +00:00
|
|
|
default_group.permissions.add(
|
2017-02-08 20:23:21 +00:00
|
|
|
*list(Permission.objects.filter(codename__endswith=permission))
|
|
|
|
)
|
2016-11-17 23:51:07 +00:00
|
|
|
|
|
|
|
# Add permission to view processing nodes
|
|
|
|
default_group.permissions.add(Permission.objects.get(codename="view_processingnode"))
|
2017-02-08 20:23:21 +00:00
|
|
|
|
2016-10-08 01:00:34 +00:00
|
|
|
# Check super user
|
2016-10-08 03:02:48 +00:00
|
|
|
if User.objects.filter(is_superuser=True).count() == 0:
|
2016-10-08 01:00:34 +00:00
|
|
|
User.objects.create_superuser('admin', 'admin@example.com', 'admin')
|
|
|
|
logger.info("Created superuser")
|
2016-10-24 18:14:35 +00:00
|
|
|
|
2016-10-25 22:08:15 +00:00
|
|
|
# Unlock any Task that might have been locked
|
|
|
|
Task.objects.filter(processing_lock=True).update(processing_lock=False)
|
2017-02-08 19:53:55 +00:00
|
|
|
|
2016-10-25 22:08:15 +00:00
|
|
|
if not settings.TESTING:
|
|
|
|
# Setup and start scheduler
|
|
|
|
scheduler.setup()
|
|
|
|
|
|
|
|
scheduler.update_nodes_info(background=True)
|
2016-10-25 20:04:24 +00:00
|
|
|
|
2016-10-25 22:08:15 +00:00
|
|
|
except ProgrammingError:
|
2016-12-13 14:52:15 +00:00
|
|
|
logger.warning("Could not touch the database. If running a migration, this is expected.")
|