134 lines
5.4 KiB
Python
134 lines
5.4 KiB
Python
# Copyright (c) 2016 Huawei, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from django import forms
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.views.decorators.debug import sensitive_variables # noqa
|
|
|
|
from horizon import exceptions
|
|
from horizon import forms as horizon_forms
|
|
from horizon import messages
|
|
|
|
import json
|
|
from karbor_dashboard.api import karbor as karborclient
|
|
|
|
EMPTY_VALUES = (None, '', u'', [], (), {})
|
|
|
|
|
|
class RestoreCheckpointForm(horizon_forms.SelfHandlingForm):
|
|
provider_id = forms.CharField(label=_("Provider ID"),
|
|
widget=forms.HiddenInput(),
|
|
required=False)
|
|
checkpoint_id = forms.CharField(label=_("Checkpoint ID"),
|
|
widget=forms.HiddenInput(),
|
|
required=False)
|
|
use_current_session = forms.BooleanField(
|
|
label=_("Use current session credentials"),
|
|
widget=forms.CheckboxInput(attrs={
|
|
'class': 'disable_input',
|
|
'data-slug': 'use_current_session',
|
|
'data-disable-on-checked': 'true',
|
|
'checked': 'checked'
|
|
}),
|
|
initial=False,
|
|
required=False)
|
|
restore_target = forms.URLField(
|
|
label=_("Restore Target"),
|
|
widget=forms.URLInput(attrs={
|
|
'class': 'disabled_input',
|
|
'data-disable-on': 'use_current_session',
|
|
'data-source-manual': _("Restore Target"),
|
|
'disabled': 'disabled',
|
|
'value': 'Target: Current project'
|
|
}),
|
|
required=False)
|
|
restore_target_username = forms.CharField(
|
|
label=_("Restore Target Username"),
|
|
widget=forms.TextInput(attrs={
|
|
'class': 'disabled_input',
|
|
'data-disable-on': 'use_current_session',
|
|
'data-source-manual': _("Restore Target Username"),
|
|
'disabled': 'disabled',
|
|
'value': 'Target Username: current user'
|
|
}),
|
|
required=False)
|
|
restore_target_password = forms.CharField(
|
|
label=_("Restore Target Password"),
|
|
widget=forms.PasswordInput(attrs={
|
|
'class': 'disabled_input',
|
|
'data-disable-on': 'use_current_session',
|
|
'data-source-manual': _("Restore Target Password"),
|
|
'disabled': 'disabled',
|
|
'hidden': 'hidden'
|
|
}),
|
|
required=False)
|
|
provider = forms.CharField(
|
|
widget=forms.HiddenInput(attrs={"class": "provider"}))
|
|
parameters = forms.CharField(
|
|
widget=forms.HiddenInput(attrs={"class": "parameters"}))
|
|
failure_url = 'horizon:karbor:checkpoints:index'
|
|
|
|
def __init__(self, request, *args, **kwargs):
|
|
super(RestoreCheckpointForm, self).\
|
|
__init__(request, *args, **kwargs)
|
|
|
|
provider_id = str(kwargs["initial"]["provider_id"])
|
|
provider = karborclient.provider_get(request, provider_id)
|
|
self.fields['provider'].initial = json.dumps(provider._info)
|
|
|
|
@sensitive_variables('restore_target_password')
|
|
def handle(self, request, data):
|
|
def all_empty(data_list):
|
|
return all(map(lambda x: x in EMPTY_VALUES, data_list))
|
|
|
|
def all_not_empty(data_list):
|
|
return all(map(lambda x: x not in EMPTY_VALUES, data_list))
|
|
|
|
def empty_to_none(data_):
|
|
return data_ if data_ not in EMPTY_VALUES else None
|
|
|
|
target = empty_to_none(data["restore_target"])
|
|
target_username = empty_to_none(data["restore_target_username"])
|
|
target_password = empty_to_none(data["restore_target_password"])
|
|
use_current_session = empty_to_none(data["use_current_session"])
|
|
|
|
if not use_current_session:
|
|
validate_data = [target, target_username, target_password]
|
|
if not (all_empty(validate_data) or all_not_empty(validate_data)):
|
|
messages.warning(request,
|
|
_('Restore Target, Restore Target Username '
|
|
'and Restore Target Password must be '
|
|
'assigned at the same time or not '
|
|
'assigned.'))
|
|
return False
|
|
|
|
try:
|
|
data_parameters = json.loads(data["parameters"])
|
|
restore_auth = {
|
|
"type": "password",
|
|
"username": target_username,
|
|
"password": target_password,
|
|
}
|
|
new_restore = karborclient.restore_create(
|
|
request,
|
|
provider_id=data["provider_id"],
|
|
checkpoint_id=data["checkpoint_id"],
|
|
restore_target=target,
|
|
parameters=data_parameters,
|
|
restore_auth=restore_auth)
|
|
messages.success(request, _("Checkpoint restore initiated"))
|
|
return new_restore
|
|
except Exception:
|
|
exceptions.handle(request, _('Unable to restore checkpoint.'))
|