Skip to content

Commit

Permalink
Add backup support for config files
Browse files Browse the repository at this point in the history
Move the backup functionality from `depends.py`

Signed-off-by: Gabor Mezei <gabor.mezei@arm.com>
  • Loading branch information
gabor-mezei-arm committed Sep 19, 2024
1 parent 8c488b1 commit ac4c133
Showing 1 changed file with 46 additions and 0 deletions.
46 changes: 46 additions & 0 deletions scripts/mbedtls_framework/config_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import argparse
import os
import re
import shutil
import sys

from abc import ABCMeta
Expand Down Expand Up @@ -194,6 +195,18 @@ def filename(self, name=None):

return self._get_configfile(name).filename

def backup(self, suffix='.bak'):
"""Back up the configuration file."""

for configfile in self.configfiles:
configfile.backup(suffix)

def restore(self):
"""Restore the configuration file."""

for configfile in self.configfiles:
configfile.restore()


class ConfigFile(metaclass=ABCMeta):
"""Representation of a configuration file."""
Expand All @@ -214,6 +227,8 @@ def __init__(self, default_path, name, filename=None):
self.current_section = None
self.inclusion_guard = None
self.modified = False
self._backupname = None
self._own_backup = False

_define_line_regexp = (r'(?P<indentation>\s*)' +
r'(?P<commented_out>(//\s*)?)' +
Expand Down Expand Up @@ -324,6 +339,37 @@ def write(self, settings, filename=None):
with open(filename, 'w', encoding='utf-8') as output:
self.write_to_stream(settings, output)

def backup(self, suffix='.bak'):
"""Back up the configuration file.
If the backup file already exists, it is presumed to be the desired backup,
so don't make another backup.
"""
if self._backupname:
return

self._backupname = self.filename + suffix
if os.path.exists(self._backupname):
self._own_backup = False
else:
self._own_backup = True
shutil.copy(self.filename, self._backupname)

def restore(self):
"""Restore the configuration file.
Only delete the backup file if it was created earlier.
"""
if not self._backupname:
return

if self._own_backup:
shutil.move(self._backupname, self.filename)
else:
shutil.copy(self._backupname, self.filename)

self._backupname = None


class ConfigTool(metaclass=ABCMeta):
"""Command line config manipulation tool.
Expand Down

0 comments on commit ac4c133

Please sign in to comment.