# 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.core.urlresolvers import reverse from django.core.urlresolvers import reverse_lazy from django.utils.translation import ugettext_lazy as _ from calendar import monthrange from datetime import date from datetime import timedelta from horizon import exceptions from horizon import forms as horizon_forms from horizon import tables as horizon_tables from horizon.utils import memoized from karbor_dashboard.api import karbor as karborclient from karbor_dashboard.checkpoints import forms from karbor_dashboard.checkpoints import tables from karbor_dashboard.checkpoints import utils from karborclient.v1 import protectables from oslo_utils import uuidutils class IndexView(horizon_tables.DataTableView): table_class = tables.CheckpointsTable template_name = 'checkpoints/index.html' page_title = _("Checkpoints") @memoized.memoized_method def get_provider_list(self): return karborclient.provider_list(self.request) @memoized.memoized_method def get_plan_list(self): return karborclient.plan_list(self.request) @memoized.memoized_method def get_filter_list(self): filters = {} # Get all filter for key in utils.FILTER_LIST: filters[key] = self.request.POST.get(key, u"All") # Remove the "All" of provider_filter provider_filter = utils.FILTER_LIST[0] try: providers = self.get_provider_list() filters[provider_filter] = \ self.request.POST.get(provider_filter, providers[0].id) except Exception: exceptions.handle( self.request, _('Unable to retrieve anyone provider.')) # Get arguments from the providers page if self.kwargs.get("provider_id", None) is not None: filters[provider_filter] = self.kwargs["provider_id"] return filters def get_context_data(self, **kwargs): context = super(IndexView, self).get_context_data(**kwargs) try: context["provider_list"] = self.get_provider_list() except Exception: exceptions.handle( self.request, _('Unable to retrieve anyone provider.')) context["plan_list"] = self.get_plan_list() context["date_list"] = utils.DATE_CHOICES context["url"] = reverse("horizon:karbor:checkpoints:index") context = dict(context, **self.get_filter_list()) return context @memoized.memoized_method def get_search_opts(self): def _total_days(year, month, num_months): days = 0 i = 0 while i < num_months: days += monthrange(year, month)[1] month = month - 1 if month > 1 else 12 year = year if month > 1 else year - 1 i += 1 return days search_opts = {} filters = self.get_filter_list() provider_id = filters.get(utils.FILTER_LIST[0], None) plan_id = filters.get(utils.FILTER_LIST[1], u"All") if plan_id != u"All": search_opts["plan_id"] = plan_id now = date.today() date_filter = filters.get(utils.FILTER_LIST[2], None) if date_filter == utils.TODAY: delta = timedelta(days=1) elif date_filter == utils.LASTESTONEWEEK: delta = timedelta(weeks=1) elif date_filter == utils.LASTESTTWOWEEKS: delta = timedelta(weeks=2) elif date_filter == utils.LASTESTONEMONTH: days = _total_days(now.year, now.month, 1) delta = timedelta(days=days) elif date_filter == utils.LASTESTTHREEMONTHS: days = _total_days(now.year, now.month, 3) delta = timedelta(days=days) else: delta = None if delta: search_opts["start_date"] = now - delta search_opts["end_date"] = now return provider_id, search_opts def has_prev_data(self, table): return self._prev def has_more_data(self, table): return self._more def get_data(self): prev_marker = self.request.GET.get( tables.CheckpointsTable._meta.prev_pagination_param, None) if prev_marker is not None: marker = prev_marker else: marker = self.request.GET.get( tables.CheckpointsTable._meta.pagination_param, None) reversed_order = prev_marker is not None checkpoints = [] try: # Get provider id and search_opts provider_id, search_opts = self.get_search_opts() if provider_id is None: raise Exception() checkpoints, self._more, self._prev = \ karborclient.checkpoint_list_paged( self.request, provider_id=provider_id, search_opts=search_opts, marker=marker, paginate=True, sort_dir='asc', sort_key='id', reversed_order=reversed_order) provider = karborclient.provider_get(self.request, provider_id) for checkpoint in checkpoints: setattr(checkpoint, "provider_name", provider.name) setattr(checkpoint, "provider_id", provider_id) except Exception: self._prev = False self._more = False exceptions.handle(self.request, _('Unable to retrieve checkpoints list.')) return checkpoints def get_table(self): super(IndexView, self).get_table() provider_id, _ = self.get_search_opts() setattr(self.table, 'provider_id', provider_id) return self.table class CheckpointsRestoreView(horizon_forms.ModalFormView): template_name = 'checkpoints/restore.html' modal_header = _("Restore Checkpoint") form_id = "restore_checkpoint_form" form_class = forms.RestoreCheckpointForm submit_label = _("Restore Checkpoint") submit_url = 'horizon:karbor:checkpoints:restore' success_url = reverse_lazy('horizon:karbor:restores:index') page_title = _("Restore Checkpoint") def get_initial(self): return {"provider_id": self.kwargs['provider_id'], "checkpoint_id": self.kwargs['checkpoint_id']} def get_context_data(self, **kwargs): context = super(CheckpointsRestoreView, self). \ get_context_data(**kwargs) provider_id = self.kwargs['provider_id'] checkpoint_id = self.kwargs['checkpoint_id'] context['provider_id'] = provider_id context['checkpoint_id'] = checkpoint_id context["instances"] = self.get_resources() context['submit_url'] = reverse(self.submit_url, args=(provider_id, checkpoint_id)) return context @memoized.memoized_method def get_resources(self): results = [] try: provider_id = self.kwargs['provider_id'] checkpoint_id = self.kwargs['checkpoint_id'] checkpoint = karborclient.checkpoint_get(self.request, provider_id, checkpoint_id) graphnodes = utils.deserialize_resource_graph( checkpoint.resource_graph) self.get_results(graphnodes, None, results) except Exception: exceptions.handle( self.request, _('Unable to retrieve checkpoint details.'), redirect=reverse("horizon:karbor:checkpoints:index")) return results def get_results(self, graphnodes, showparentid, results): for graphnode in graphnodes: if graphnode is not None: # add graph node to results resource = {} resource["id"] = graphnode.value.id resource["type"] = graphnode.value.type resource["name"] = graphnode.value.name resource["showid"] = uuidutils.generate_uuid() resource["showparentid"] = showparentid result = protectables.Instances(self, resource) results.append(result) # add child graph nodes to results self.get_results(graphnode.child_nodes, result.showid, results ) class DetailView(horizon_tables.DataTableView): table_class = tables.DetailTable template_name = 'checkpoints/detail.html' page_title = _("{{ checkpoint.protection_plan.name }}") @memoized.memoized_method def get_checkpoint_data(self): try: provider_id = self.kwargs['provider_id'] checkpoint_id = self.kwargs['checkpoint_id'] checkpoint = karborclient.checkpoint_get(self.request, provider_id, checkpoint_id) except Exception: checkpoint = [] msg = _('checkpoint list can not be retrieved.') exceptions.handle(self.request, msg) return checkpoint def get_context_data(self, **kwargs): context = super(DetailView, self).get_context_data(**kwargs) checkpoint = self.get_checkpoint_data() context["checkpoint"] = checkpoint provider_id = self.kwargs['provider_id'] provider = karborclient.provider_get(self.request, provider_id) context["provider_name"] = provider.name context["resources"] = self.get_resources() context["url"] = reverse("horizon:karbor:protectionplans:index") return context @memoized.memoized_method def get_resources(self): results = [] try: provider_id = self.kwargs['provider_id'] checkpoint_id = self.kwargs['checkpoint_id'] checkpoint = karborclient.checkpoint_get(self.request, provider_id, checkpoint_id) graphnodes = utils.deserialize_resource_graph( checkpoint.resource_graph) self.get_results(graphnodes, None, results) except Exception: exceptions.handle( self.request, _('Unable to retrieve checkpoint details.'), redirect=reverse("horizon:karbor:checkpoints:index")) return results def get_results(self, graphnodes, showparentid, results): for graphnode in graphnodes: if graphnode is not None: # add graph node to results resource = {} resource["id"] = graphnode.value.id resource["type"] = graphnode.value.type resource["name"] = graphnode.value.name resource["showid"] = uuidutils.generate_uuid() resource["showparentid"] = showparentid result = protectables.Instances(self, resource) results.append(result) # add child graph nodes to results self.get_results(graphnode.child_nodes, result.showid, results )