[UPD] script: format code

This commit is contained in:
Mathieu Benoit 2021-07-20 20:00:58 -04:00
parent 1fdad0ee31
commit 6bb4dafba6
3 changed files with 37 additions and 17 deletions

View file

@ -30,7 +30,11 @@ def get_config():
)
parser.add_argument("--csv_1", required=True, help="CSV file first")
parser.add_argument("--csv_2", required=True, help="CSV file second")
parser.add_argument("--compare_key", required=True, help="Key to search and compare with it.")
parser.add_argument(
"--compare_key",
required=True,
help="Key to search and compare with it.",
)
args = parser.parse_args()
return args
@ -38,8 +42,8 @@ def get_config():
def main():
config = get_config()
with open(config.csv_1, mode='r') as csv_file_1:
with open(config.csv_2, mode='r') as csv_file_2:
with open(config.csv_1, mode="r") as csv_file_1:
with open(config.csv_2, mode="r") as csv_file_2:
csv_reader_1 = csv.DictReader(csv_file_1)
csv_reader_2 = csv.DictReader(csv_file_2)
lst_csv_1 = [a.get(config.compare_key) for a in csv_reader_1]

View file

@ -5,7 +5,9 @@ import argparse
import logging
import yaml
new_path = os.path.normpath(os.path.join(os.path.dirname(__file__), "..", ".."))
new_path = os.path.normpath(
os.path.join(os.path.dirname(__file__), "..", "..")
)
sys.path.append(new_path)
from script.git_tool import GitTool

View file

@ -37,14 +37,22 @@ def get_config():
parser.add_argument("--backup_2", help="Backup name second")
args = parser.parse_args()
die(bool(args.backup_file_1) and bool(args.backup_1),
"Take only --backup_file_1 or --backup_1")
die(not bool(args.backup_file_1) and not bool(args.backup_1),
"Missing --backup_file_1 or --backup_1")
die(bool(args.backup_file_2) and bool(args.backup_2),
"Take only --backup_file_2 or --backup_2")
die(not bool(args.backup_file_2) and not bool(args.backup_2),
"Missing --backup_file_2 or --backup_2")
die(
bool(args.backup_file_1) and bool(args.backup_1),
"Take only --backup_file_1 or --backup_1",
)
die(
not bool(args.backup_file_1) and not bool(args.backup_1),
"Missing --backup_file_1 or --backup_1",
)
die(
bool(args.backup_file_2) and bool(args.backup_2),
"Take only --backup_file_2 or --backup_2",
)
die(
not bool(args.backup_file_2) and not bool(args.backup_2),
"Missing --backup_file_2 or --backup_2",
)
return args
@ -61,11 +69,11 @@ def main():
file_path_2 = config.backup_file_2
else:
file_path_2 = os.path.join("image_db", f"{config.backup_2}.zip")
with zipfile.ZipFile(file_path_1, 'r') as zip_ref:
with zipfile.ZipFile(file_path_1, "r") as zip_ref:
manifest_file_1 = zip_ref.open("manifest.json")
with zipfile.ZipFile(file_path_2, 'r') as zip_ref:
with zipfile.ZipFile(file_path_2, "r") as zip_ref:
manifest_file_2 = zip_ref.open("manifest.json")
json_manifest_file_1 = json.load(manifest_file_1)
@ -81,10 +89,16 @@ def main():
print(f"{len(same)} same")
if same:
print(same)
print(f"{Fore.BLUE}{len(difference_1)}{Style.RESET_ALL} difference manifest 1 to manifest 2")
print(
f"{Fore.BLUE}{len(difference_1)}{Style.RESET_ALL} difference manifest"
" 1 to manifest 2"
)
if difference_1:
print(difference_1)
print(f"{Fore.MAGENTA}{len(difference_2)}{Style.RESET_ALL} difference manifest 2 to manifest 1")
print(
f"{Fore.MAGENTA}{len(difference_2)}{Style.RESET_ALL} difference"
" manifest 2 to manifest 1"
)
if difference_2:
print(difference_2)