Nikolas Nyby 5439084c84
Add flake8 to gitlib-ci - closes #58
And fix the rest of the flake8 errors.

Reviewed-by: Sunil Mohan Adapa <sunil@medhas.org>
2019-07-25 11:07:06 -07:00

153 lines
4.6 KiB
Python

#
# This file is part of FreedomBox.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
"""
FreedomBox app to manage filesystem snapshots.
"""
import json
import augeas
from django.utils.translation import ugettext_lazy as _
from plinth import actions
from plinth import app as app_module
from plinth import menu
from plinth.modules import storage
from .manifest import backup # noqa, pylint: disable=unused-import
version = 4
managed_packages = ['snapper']
name = _('Storage Snapshots')
description = [
_('Snapshots allows creating and managing btrfs file system snapshots. '
'These can be used to roll back the system to a previously known '
'good state in case of unwanted changes to the system.'),
# Translators: xgettext:no-python-format
_('Snapshots are taken periodically (called timeline snapshots) and also '
'before and after a software installation. Older snapshots will be '
'automatically cleaned up according to the settings below.'),
_('Snapshots currently work on btrfs file systems only and on the root '
'partition only. Snapshots are not a replacement for '
'<a href="/plinth/sys/backups">backups</a> since '
'they can only be stored on the same partition. ')
]
manual_page = 'Snapshots'
DEFAULT_FILE = '/etc/default/snapper'
fs_types_supported = ['btrfs']
app = None
class SnapshotApp(app_module.App):
"""FreedomBox app for snapshots."""
app_id = 'snapshot'
def __init__(self):
"""Create components for the app."""
super().__init__()
menu_item = menu.Menu('menu-snapshot', name, None, 'fa-film',
'snapshot:index', parent_url_name='system')
self.add(menu_item)
def init():
"""Initialize the module."""
global app
app = SnapshotApp()
setup_helper = globals()['setup_helper']
if setup_helper.get_state() != 'needs-setup' and app.is_enabled():
app.set_enabled(True)
def is_supported():
"""Return whether snapshots are support on current setup."""
fs_type = storage.get_filesystem_type()
return fs_type in fs_types_supported
def setup(helper, old_version=None):
"""Install and configure the module."""
helper.install(managed_packages)
if is_supported():
helper.call('post', actions.superuser_run, 'snapshot',
['setup', '--old-version',
str(old_version)])
helper.call('post', app.enable)
def load_augeas():
"""Initialize Augeas."""
aug = augeas.Augeas(
flags=augeas.Augeas.NO_LOAD + augeas.Augeas.NO_MODL_AUTOLOAD)
# shell-script config file lens
aug.set('/augeas/load/Shellvars/lens', 'Shellvars.lns')
aug.set('/augeas/load/Shellvars/incl[last() + 1]', DEFAULT_FILE)
aug.load()
return aug
def is_apt_snapshots_enabled(aug):
"""Return whether APT snapshots is enabled."""
value = aug.get('/files' + DEFAULT_FILE + '/DISABLE_APT_SNAPSHOT')
return value != 'yes'
def get_configuration():
aug = load_augeas()
output = actions.superuser_run('snapshot', ['get-config'])
output = json.loads(output)
def get_boolean_choice(status):
return ('yes', 'Enabled') if status else ('no', 'Disabled')
def get_max_from_range(key):
return output[key].split('-')[-1]
return {
'enable_timeline_snapshots':
get_boolean_choice(output['TIMELINE_CREATE'] == 'yes'),
'enable_software_snapshots':
get_boolean_choice(is_apt_snapshots_enabled(aug)),
'hourly_limit':
get_max_from_range('TIMELINE_LIMIT_HOURLY'),
'daily_limit':
get_max_from_range('TIMELINE_LIMIT_DAILY'),
'weekly_limit':
get_max_from_range('TIMELINE_LIMIT_WEEKLY'),
'monthly_limit':
get_max_from_range('TIMELINE_LIMIT_MONTHLY'),
'yearly_limit':
get_max_from_range('TIMELINE_LIMIT_YEARLY'),
'free_space':
output['FREE_LIMIT'],
}
def restore_post(packet):
"""Run after restore."""
actions.superuser_run('snapshot', ['kill-daemon'])