mirror of
https://gitlab.com/dslackw/slpkg.git
synced 2024-11-17 07:48:18 +01:00
294 lines
11 KiB
Python
Executable file
294 lines
11 KiB
Python
Executable file
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import sys
|
|
import shutil
|
|
import difflib
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
|
|
class NewConfig:
|
|
|
|
def __init__(self, flags: list):
|
|
self.flags: list = flags
|
|
self.etc_path: str = '/etc/slpkg'
|
|
self.slpkg_config = Path(self.etc_path, 'slpkg.toml')
|
|
self.repositories_config = Path(self.etc_path, 'repositories.toml')
|
|
self.blacklist_config = Path(self.etc_path, 'blacklist.toml')
|
|
self.slpkg_config_new = Path(self.etc_path, 'slpkg.toml.new')
|
|
self.repositories_config_new = Path(self.etc_path, 'repositories.toml.new')
|
|
self.blacklist_config_new = Path(self.etc_path, 'blacklist.toml.new')
|
|
|
|
self.color: dict = {}
|
|
self.colors()
|
|
|
|
self.bold: str = self.color['bold']
|
|
self.red: str = self.color['red']
|
|
self.green: str = self.color['green']
|
|
self.yellow: str = self.color['yellow']
|
|
self.bgreen: str = f'{self.color["bold"]}{self.color["green"]}'
|
|
self.byellow: str = f'{self.color["bold"]}{self.color["yellow"]}'
|
|
self.endc: str = self.color['endc']
|
|
self.choice = None
|
|
|
|
def colors(self):
|
|
if '--no-colors' in self.flags:
|
|
self.color = {
|
|
'bold': '',
|
|
'red': '',
|
|
'green': '',
|
|
'yellow': '',
|
|
'cyan': '',
|
|
'blue': '',
|
|
'grey': '',
|
|
'violet': '',
|
|
'endc': ''
|
|
}
|
|
else:
|
|
self.color = {
|
|
'bold': '\033[1m',
|
|
'red': '\x1b[91m',
|
|
'green': '\x1b[32m',
|
|
'yellow': '\x1b[93m',
|
|
'cyan': '\x1b[96m',
|
|
'blue': '\x1b[94m',
|
|
'grey': '\x1b[38;5;247m',
|
|
'violet': '\x1b[35m',
|
|
'endc': '\x1b[0m'
|
|
}
|
|
|
|
def check(self):
|
|
""" Checks for .new files. """
|
|
print('Checking for NEW configuration files...')
|
|
if (self.slpkg_config_new.is_file() or self.blacklist_config_new.is_file()
|
|
or self.repositories_config_new.is_file()):
|
|
print('\nThere are NEW files:\n')
|
|
|
|
if self.slpkg_config_new.is_file():
|
|
print(f"{self.bgreen:>12}{self.slpkg_config_new}{self.endc}")
|
|
|
|
if self.repositories_config_new.is_file():
|
|
print(f"{self.bgreen:>12}{self.repositories_config_new}{self.endc}")
|
|
|
|
if self.blacklist_config_new.is_file():
|
|
print(f"{self.bgreen:>12}{self.blacklist_config_new}{self.endc}")
|
|
|
|
print(f'\nWhat would you like to do ({self.byellow}K{self.endc}/{self.byellow}O{self.endc}/'
|
|
f'{self.byellow}R{self.endc}/{self.byellow}P{self.endc})?\n')
|
|
|
|
print(f"{'':>2}({self.byellow}K{self.endc})eep the old files and consider '.new' files later.\n"
|
|
f"{'':>2}({self.byellow}O{self.endc})verwrite all old files with the new ones.\n"
|
|
f"{'':>5}The old files will be stored with the suffix '.orig'.\n"
|
|
f"{'':>2}({self.byellow}R{self.endc})emove all '.new' files.\n"
|
|
f"{'':>2}({self.byellow}P{self.endc})rompt K, O, R, D, V selection for every single file.\n")
|
|
|
|
self.menu()
|
|
|
|
elif (not self.slpkg_config_new.is_file() and not self.blacklist_config_new.is_file()
|
|
and not self.repositories_config_new.is_file()):
|
|
print(f"\n{'No .new files found.':>23}\n")
|
|
|
|
def menu(self):
|
|
""" Menu of choices. """
|
|
choice = input('Choice: ')
|
|
|
|
choice = choice.lower()
|
|
|
|
arguments: dict = {
|
|
'k': self.keep,
|
|
'o': self.overwrite,
|
|
'r': self.remove,
|
|
'p': self.prompt
|
|
}
|
|
|
|
try:
|
|
arguments[choice]()
|
|
except KeyError:
|
|
self.keep()
|
|
|
|
@staticmethod
|
|
def keep():
|
|
print("\nNo changes were made.\n")
|
|
|
|
def overwrite(self):
|
|
""" Copy tne .new files and rename the olds to .orig. """
|
|
if self.slpkg_config_new.is_file():
|
|
self.overwrite_config_file()
|
|
|
|
if self.repositories_config_new.is_file():
|
|
self.overwrite_repositories_file()
|
|
|
|
if self.blacklist_config_new.is_file():
|
|
self.overwrite_blacklist_file()
|
|
print() # new line
|
|
|
|
def overwrite_config_file(self):
|
|
""" Copy tne slpkg.toml.new file and rename the old to .orig. """
|
|
if self.slpkg_config.is_file():
|
|
shutil.copy(self.slpkg_config, f"{self.slpkg_config}.orig")
|
|
print(f"\ncp {self.green}{self.slpkg_config}{self.endc} -> {self.slpkg_config}.orig")
|
|
|
|
shutil.move(self.slpkg_config_new, self.slpkg_config)
|
|
print(f"mv {self.slpkg_config_new} -> {self.green}{self.slpkg_config}{self.endc}")
|
|
|
|
def overwrite_repositories_file(self):
|
|
""" Copy tne repositories.toml.new file and rename the old to .orig. """
|
|
if self.slpkg_config.is_file():
|
|
shutil.copy(self.repositories_config, f"{self.repositories_config}.orig")
|
|
print(f"\ncp {self.green}{self.repositories_config}{self.endc} -> {self.repositories_config}.orig")
|
|
|
|
shutil.move(self.repositories_config_new, self.repositories_config)
|
|
print(f"mv {self.repositories_config_new} -> {self.green}{self.repositories_config}{self.endc}")
|
|
|
|
def overwrite_blacklist_file(self):
|
|
""" Copy tne blacklist.toml.new file and rename the old to .orig. """
|
|
if self.blacklist_config.is_file():
|
|
shutil.copy(self.blacklist_config, f"{self.blacklist_config}.orig")
|
|
print(f"\ncp {self.green}{self.blacklist_config}{self.endc} -> {self.blacklist_config}.orig")
|
|
|
|
shutil.move(self.blacklist_config_new, self.blacklist_config)
|
|
print(f"mv {self.blacklist_config_new} -> {self.green}{self.blacklist_config}{self.endc}")
|
|
|
|
def remove(self):
|
|
""" Removes the .new files. """
|
|
print() # new line
|
|
self.remove_config_new_file()
|
|
self.remove_repositories_new_file()
|
|
self.remove_blacklist_new_file()
|
|
print() # new line
|
|
|
|
def remove_config_new_file(self):
|
|
""" Remove slpkg.toml.new file. """
|
|
if self.slpkg_config_new.is_file():
|
|
self.slpkg_config_new.unlink()
|
|
print(f"rm {self.red}{self.slpkg_config_new}{self.endc}")
|
|
|
|
def remove_repositories_new_file(self):
|
|
""" Remove repositories.toml.new file. """
|
|
if self.repositories_config_new.is_file():
|
|
self.repositories_config_new.unlink()
|
|
print(f"rm {self.red}{self.repositories_config_new}{self.endc}")
|
|
|
|
def remove_blacklist_new_file(self):
|
|
""" Remove blacklist.toml.new file. """
|
|
if self.blacklist_config_new.is_file():
|
|
self.blacklist_config_new.unlink()
|
|
print(f"rm {self.red}{self.blacklist_config_new}{self.endc}")
|
|
|
|
def prompt(self):
|
|
""" Prompt K, O, R selection for every single file. """
|
|
print(f"\n{'':>2}({self.byellow}K{self.endc})eep, ({self.byellow}O{self.endc})verwrite, "
|
|
f"({self.byellow}R{self.endc})emove, ({self.byellow}D{self.endc})iff, "
|
|
f"({self.byellow}V{self.endc})imdiff\n")
|
|
|
|
if self.slpkg_config_new.is_file():
|
|
make = input(f'{self.bgreen}{self.slpkg_config_new}{self.endc} - '
|
|
f'({self.byellow}K{self.endc}/{self.byellow}O{self.endc}/'
|
|
f'{self.byellow}R{self.endc}/{self.byellow}D{self.endc}/'
|
|
f'{self.byellow}V{self.endc}): ')
|
|
|
|
if make.lower() == 'k':
|
|
pass
|
|
if make.lower() == 'o':
|
|
self.overwrite_config_file()
|
|
print() # new line
|
|
if make.lower() == 'r':
|
|
print() # new line
|
|
self.remove_config_new_file()
|
|
print() # new line
|
|
if make.lower() == 'd':
|
|
self.diff_files(self.slpkg_config_new, self.slpkg_config)
|
|
if make.lower() == 'v':
|
|
self.vimdiff(self.slpkg_config_new, self.slpkg_config)
|
|
|
|
if self.repositories_config_new.is_file():
|
|
make = input(f'{self.bgreen}{self.repositories_config_new}{self.endc} - '
|
|
f'({self.byellow}K{self.endc}/{self.byellow}O{self.endc}/'
|
|
f'{self.byellow}R{self.endc}/{self.byellow}D{self.endc}/'
|
|
f'{self.byellow}V{self.endc}): ')
|
|
|
|
if make.lower() == 'k':
|
|
pass
|
|
if make.lower() == 'o':
|
|
self.overwrite_repositories_file()
|
|
print() # new line
|
|
if make.lower() == 'r':
|
|
print() # new line
|
|
self.remove_repositories_new_file()
|
|
print() # new line
|
|
if make.lower() == 'd':
|
|
self.diff_files(self.repositories_config_new, self.repositories_config)
|
|
if make.lower() == 'v':
|
|
self.vimdiff(self.repositories_config_new, self.repositories_config)
|
|
|
|
if self.blacklist_config_new.is_file():
|
|
make = input(f'{self.bgreen}{self.blacklist_config_new}{self.endc} - '
|
|
f'({self.byellow}K{self.endc}/{self.byellow}O{self.endc}/'
|
|
f'{self.byellow}R{self.endc}/{self.byellow}D{self.endc}/'
|
|
f'{self.byellow}V{self.endc}): ')
|
|
|
|
if make.lower() == 'k':
|
|
pass
|
|
if make.lower() == 'o':
|
|
self.overwrite_blacklist_file()
|
|
print() # new line
|
|
if make.lower() == 'r':
|
|
print() # new line
|
|
self.remove_blacklist_new_file()
|
|
print() # new line
|
|
if make.lower() == 'd':
|
|
self.diff_files(self.blacklist_config_new, self.blacklist_config)
|
|
if make.lower() == 'v':
|
|
self.vimdiff(self.blacklist_config_new, self.blacklist_config)
|
|
|
|
@staticmethod
|
|
def diff_files(file2, file1):
|
|
""" Diff the .new and the current file. """
|
|
with open(file1, 'r') as f1:
|
|
with open(file2, 'r') as f2:
|
|
diff = difflib.context_diff(
|
|
f1.readlines(),
|
|
f2.readlines(),
|
|
fromfile=str(file1),
|
|
tofile=str(file2)
|
|
)
|
|
for line in diff:
|
|
print(line, end='')
|
|
|
|
@staticmethod
|
|
def vimdiff(file1, file2):
|
|
output = subprocess.call(f'vimdiff {file1} {file2}', shell=True)
|
|
if output != 0:
|
|
raise SystemExit(output)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
args = sys.argv
|
|
args.pop(0)
|
|
|
|
options: list = [
|
|
'--no-colors',
|
|
'-h', '--help'
|
|
]
|
|
|
|
if len(args) == 1:
|
|
if options[1] in args or options[2] in args:
|
|
print('slpkg_new-configs [OPTIONS]\n'
|
|
'\n --no-colors Disable the output colors.\n'
|
|
' -h, --help Show this message and exit.\n')
|
|
sys.exit()
|
|
elif args[0] == options[0]:
|
|
pass
|
|
else:
|
|
print('\ntry: slpkg_new-configs --help\n')
|
|
sys.exit(1)
|
|
elif len(args) > 1:
|
|
print('\ntry: slpkg_new-configs --help\n')
|
|
sys.exit(1)
|
|
|
|
try:
|
|
config = NewConfig(args)
|
|
config.check()
|
|
except KeyboardInterrupt:
|
|
raise SystemExit(1)
|