nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-01-28 08:03:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

backups: Simplify getting included apps during restoring

Browse Source 
Signed-off-by: Joseph Nuthalapati <njoseph@thoughtworks.com>
This commit is contained in:
Joseph Nuthalapati 2018-09-23 15:45:49 -07:00
parent f2528c434d
commit f172925d9d
No known key found for this signature in database
GPG Key ID: 5398F00A2FA43C35
1 changed files with 8 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
plinth/modules/backups/views.py
View File

@ -32,7 +32,6 @@ from django.views.generic import FormView, TemplateView
from plinth.modules import backups
from . import backups as backups_api
from . import find_exported_archive, get_export_apps
from .forms import CreateArchiveForm, ExportArchiveForm, RestoreForm
@ -134,29 +133,20 @@ class RestoreView(SuccessMessageMixin, FormView):
success_url = reverse_lazy('backups:index')
success_message = _('Restored files from backup.')
def collect_data(self, label, name):
def _get_included_apps(self):
"""Save some data used to instantiate the form."""
self.label = unquote(label)
self.name = unquote(name)
self.filename = find_exported_archive(self.label, self.name)
self.installed_apps = backups_api.get_all_apps_for_backup()
self.included_apps = get_export_apps(self.filename)
def get(self, request, *args, **kwargs):
"""Save request parameters for later."""
self.collect_data(kwargs['label'], kwargs['name'])
return super().get(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
"""Save request parameters for later."""
self.collect_data(kwargs['label'], kwargs['name'])
return super().post(request, *args, **kwargs)
label = unquote(self.kwargs['label'])
name = unquote(self.kwargs['name'])
filename = backups.find_exported_archive(label, name)
return backups.get_export_apps(filename)
def get_form_kwargs(self):
"""Pass additional keyword args for instantiating the form."""
kwargs = super().get_form_kwargs()
included_apps = self._get_included_apps()
installed_apps = backups_api.get_all_apps_for_backup()
kwargs['apps'] = [
app for app in self.installed_apps if app[0] in self.included_apps
app for app in installed_apps if app[0] in included_apps
]
return kwargs