shaba/awx
1
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2024-11-02 01:21:21 +03:00
Code Releases Activity

copy NotificationTemplate code for migration and use Credential model directly

Browse Source
This commit is contained in:
Wayne Witzel III 2017-06-20 10:39:07 -04:00
parent bd52e4bb22
commit 3750f2ebf3
1 changed files with 31 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
awx/main/migrations/_reencrypt.py
View File

@ -1,13 +1,32 @@
from django.utils.translation import ugettext_lazy as _
from awx.main import utils from awx.main import utils
from awx.conf.migrations._reencrypt import ( from awx.conf.migrations._reencrypt import (
decrypt_field, decrypt_field,
should_decrypt_field, should_decrypt_field,
) )
from awx.main.notifications.email_backend import CustomEmailBackend
from awx.main.notifications.slack_backend import SlackBackend
from awx.main.notifications.twilio_backend import TwilioBackend
from awx.main.notifications.pagerduty_backend import PagerDutyBackend
from awx.main.notifications.hipchat_backend import HipChatBackend
from awx.main.notifications.webhook_backend import WebhookBackend
from awx.main.notifications.irc_backend import IrcBackend
from awx.main.models.credential import Credential
__all__ = ['replace_aesecb_fernet'] __all__ = ['replace_aesecb_fernet']
NOTIFICATION_TYPES = [('email', _('Email'), CustomEmailBackend),
('slack', _('Slack'), SlackBackend),
('twilio', _('Twilio'), TwilioBackend),
('pagerduty', _('Pagerduty'), PagerDutyBackend),
('hipchat', _('HipChat'), HipChatBackend),
('webhook', _('Webhook'), WebhookBackend),
('irc', _('IRC'), IrcBackend)]
def replace_aesecb_fernet(apps, schema_editor): def replace_aesecb_fernet(apps, schema_editor):
_notification_templates(apps) _notification_templates(apps)
_credentials(apps) _credentials(apps)
@ -17,8 +36,10 @@ def replace_aesecb_fernet(apps, schema_editor):
def _notification_templates(apps): def _notification_templates(apps):
NotificationTemplate = apps.get_model('main', 'NotificationTemplate') NotificationTemplate = apps.get_model('main', 'NotificationTemplate')
for nt in NotificationTemplate.objects.all(): for nt in NotificationTemplate.objects.all():
for field in filter(lambda x: nt.notification_class.init_parameters[x]['type'] == "password", CLASS_FOR_NOTIFICATION_TYPE = dict([(x[0], x[2]) for x in NOTIFICATION_TYPES])
nt.notification_class.init_parameters): notification_class = CLASS_FOR_NOTIFICATION_TYPE[nt.notification_type]
for field in filter(lambda x: notification_class.init_parameters[x]['type'] == "password",
notification_class.init_parameters):
if should_decrypt_field(nt.notification_configuration[field]): if should_decrypt_field(nt.notification_configuration[field]):
value = decrypt_field(nt, 'notification_configuration', subfield=field) value = decrypt_field(nt, 'notification_configuration', subfield=field)
nt.notification_configuration[field] = value nt.notification_configuration[field] = value
@ -26,21 +47,14 @@ def _notification_templates(apps):
def _credentials(apps): def _credentials(apps):
# this monkey-patch is necessary to make the implicit role generation save # TODO: Try to not use the model directly imported from our
# signal use the correct Role model (the version active at this point in # source (should use apps.get_model) to make the migration less britle.
# migration, not the one at HEAD) for credential in Credential.objects.all():
orig_current_apps = utils.get_current_apps
try:
utils.get_current_apps = lambda: apps
for credential in apps.get_model('main', 'Credential').objects.all():
for field_name, value in credential.inputs.items(): for field_name, value in credential.inputs.items():
if should_decrypt_field(value): if should_decrypt_field(value):
value = decrypt_field(credential, field_name) value = decrypt_field(credential, field_name)
credential.inputs[field_name] = value credential.inputs[field_name] = value
credential.save() credential.save()
finally:
utils.get_current_apps = orig_current_apps
def _unified_jobs(apps): def _unified_jobs(apps):