itdominator
/
terminator
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Create config file if it doesn't exist 

Use pathlib.Path for config file and backup file
This commit is contained in:
planet36 2020-10-09 20:28:48 -04:00
parent ebae1e1440
commit 310f70ac5f
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
terminatorlib/config.py
View File

@ -71,6 +71,7 @@ KeyError: 'ConfigBase::get_item: unknown key algo'
""" """
import os import os
import pathlib
import shutil import shutil
from copy import copy from copy import copy
from configobj import ConfigObj, flatten_errors from configobj import ConfigObj, flatten_errors
@ -719,15 +720,19 @@ class ConfigBase(Borg):
config_dir = get_config_dir() config_dir = get_config_dir()
if not os.path.isdir(config_dir): if not os.path.isdir(config_dir):
os.makedirs(config_dir) os.makedirs(config_dir)
config_file = pathlib.Path(self.command_line_options.config)
if not config_file.is_file():
config_file.touch()
try: try:
backup_file = self.command_line_options.config + '~' backup_file = config_file.with_suffix('.bak')
shutil.copy2(self.command_line_options.config, backup_file) shutil.copy2(config_file, backup_file)
with open(self.command_line_options.config, 'wb') as fh: with open(config_file, 'wb') as fh:
parser.write(fh) parser.write(fh)
os.remove(backup_file) backup_file.unlink()
except Exception as ex: except Exception as ex:
err('ConfigBase::save: Unable to save config: %s' % ex) err('ConfigBase::save: Unable to save config: %s' % ex)