Jan Dittberner
c9a9fa11b2
This commit isolates the celery task invocations of the userdbs app into signal handlers. All celery interaction is now asynchronously handled in userdbs.signals.
190 lines
7.3 KiB
Python
190 lines
7.3 KiB
Python
"""
|
|
This module contains the signal handlers of the :py:mod:`userdbs` app.
|
|
|
|
"""
|
|
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
|
|
from django.db.models.signals import post_delete, post_save
|
|
from django.dispatch import receiver
|
|
from passlib.utils import generate_password
|
|
|
|
from mysqltasks.tasks import (create_mysql_database, create_mysql_user,
|
|
delete_mysql_database, delete_mysql_user,
|
|
set_mysql_userpassword)
|
|
from pgsqltasks.tasks import (create_pgsql_database, create_pgsql_user,
|
|
delete_pgsql_database, delete_pgsql_user,
|
|
set_pgsql_userpassword)
|
|
from taskresults.models import TaskResult
|
|
|
|
from .models import DB_TYPES, DatabaseUser, UserDatabase, password_set
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@receiver(password_set, sender=DatabaseUser)
|
|
def handle_dbuser_password_set(sender, instance, password, **kwargs):
|
|
"""
|
|
Signal handler triggered by password changes for
|
|
:py:class:`userdbs.models.DatabaseUser` instances.
|
|
|
|
"""
|
|
if instance.db_type == DB_TYPES.mysql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_dbuser_password_set',
|
|
set_mysql_userpassword.s(instance.name, password),
|
|
'mysql password change')
|
|
_LOGGER.info(
|
|
'MySQL password change has been requested in task %s',
|
|
taskresult.task_id)
|
|
elif instance.db_type == DB_TYPES.pgsql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_dbuser_password_set',
|
|
set_pgsql_userpassword.s(instance.name, password),
|
|
'pgsql password change')
|
|
_LOGGER.info(
|
|
'PostgreSQL password change has been requested in task %s',
|
|
taskresult.task_id)
|
|
else:
|
|
_LOGGER.warning(
|
|
'Password change has been requested for unknown database %s'
|
|
' the request has been ignored.',
|
|
instance.db_type)
|
|
|
|
|
|
@receiver(post_save, sender=DatabaseUser)
|
|
def handle_dbuser_created(sender, instance, created, **kwargs):
|
|
"""
|
|
Signal handler triggered after the creation of or updates to
|
|
:py:class:`userdbs.models.DatabaseUser` instances.
|
|
|
|
"""
|
|
if created:
|
|
password = kwargs.get('password', generate_password())
|
|
# TODO: send GPG encrypted mail with this information
|
|
if instance.db_type == DB_TYPES.mysql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_dbuser_created',
|
|
create_mysql_user.s(instance.name, password),
|
|
'mysql user creation')
|
|
_LOGGER.info(
|
|
'A new MySQL user %s creation has been requested in task %s',
|
|
instance.name, taskresult.task_id)
|
|
elif instance.db_type == DB_TYPES.pgsql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_dbuser_created',
|
|
create_pgsql_user.s(instance.name, password),
|
|
'pgsql user creation')
|
|
_LOGGER.info(
|
|
'A new PostgreSQL user %s creation has been requested in task'
|
|
' %s',
|
|
instance.name, taskresult.task_id)
|
|
else:
|
|
_LOGGER.warning(
|
|
'created DatabaseUser for unknown database type %s',
|
|
instance.db_type)
|
|
_LOGGER.debug(
|
|
'database user %s has been %s',
|
|
instance, created and "created" or "updated")
|
|
|
|
|
|
@receiver(post_delete, sender=DatabaseUser)
|
|
def handle_dbuser_deleted(sender, instance, **kwargs):
|
|
"""
|
|
Signal handler triggered after the deletion of
|
|
:py:class:`userdbs.models.DatabaseUser` instances.
|
|
|
|
"""
|
|
if instance.db_type == DB_TYPES.mysql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_dbuser_deleted',
|
|
delete_mysql_user.s(instance.name),
|
|
'mysql user deletion')
|
|
_LOGGER.info(
|
|
'MySQL user %s deletion has been requested in task %s',
|
|
instance.name, taskresult.task_id)
|
|
elif instance.db_type == DB_TYPES.pgsql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_dbuser_deleted',
|
|
delete_pgsql_user.s(instance.name),
|
|
'pgsql user deletion')
|
|
_LOGGER.info(
|
|
'PostgreSQL user %s deletion has been requested in task %s',
|
|
instance.name, taskresult.task_id)
|
|
else:
|
|
_LOGGER.warning(
|
|
'deleted DatabaseUser %s for unknown database type %s',
|
|
instance.name, instance.db_type)
|
|
_LOGGER.debug(
|
|
'database user %s has been deleted', instance)
|
|
|
|
|
|
@receiver(post_save, sender=UserDatabase)
|
|
def handle_userdb_created(sender, instance, created, **kwargs):
|
|
"""
|
|
Signal handler triggered after the creation of or updates to
|
|
:py:class:`userdbs.models.UserDatabase` instances.
|
|
|
|
"""
|
|
if created:
|
|
if instance.db_user.db_type == DB_TYPES.mysql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_userdb_created',
|
|
create_mysql_database.s(
|
|
instance.db_name, instance.db_user.name),
|
|
'mysql database creation')
|
|
_LOGGER.info(
|
|
'The creation of a new MySQL database %s has been requested in'
|
|
' task %s',
|
|
instance.db_name, taskresult.task_id)
|
|
elif instance.db_user.db_type == DB_TYPES.pgsql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_userdb_created',
|
|
create_pgsql_database.s(
|
|
instance.db_name, instance.db_user.name),
|
|
'pgsql database creation')
|
|
_LOGGER.info(
|
|
'The creation of a new PostgreSQL database %s has been'
|
|
' requested in task %s',
|
|
instance.db_name, taskresult.task_id)
|
|
else:
|
|
_LOGGER.warning(
|
|
'created UserDatabase for unknown database type %s',
|
|
instance.db_user.db_type)
|
|
_LOGGER.debug(
|
|
'database %s has been %s',
|
|
instance, created and "created" or "updated")
|
|
|
|
|
|
@receiver(post_delete, sender=UserDatabase)
|
|
def handle_userdb_deleted(sender, instance, **kwargs):
|
|
"""
|
|
Signal handler triggered after the deletion of
|
|
:py:class:`userdbs.models.UserDatabase` instances.
|
|
|
|
"""
|
|
if instance.db_user.db_type == DB_TYPES.mysql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_userdb_deleted',
|
|
delete_mysql_database.s(instance.db_name, instance.db_user.name),
|
|
'mysql database deletion')
|
|
_LOGGER.info(
|
|
'The deletion of MySQL database %s has been requested in task %s',
|
|
instance.db_name, taskresult.task_id)
|
|
elif instance.db_user.db_type == DB_TYPES.pgsql:
|
|
taskresult = TaskResult.objects.create_task_result(
|
|
'handle_userdb_deleted',
|
|
delete_pgsql_database.s(instance.db_name),
|
|
'pgsql database deletion')
|
|
_LOGGER.info(
|
|
'The deletion of PostgreSQL database %s has been requested in '
|
|
' task %s',
|
|
instance.db_name, taskresult.task_id)
|
|
else:
|
|
_LOGGER.warning(
|
|
'deleted UserDatabase %s of unknown type %s',
|
|
instance.db_name, instance.db_type)
|
|
pass
|
|
_LOGGER.debug(
|
|
'database %s has been deleted', instance)
|