atomicshop 2.19.14__py3-none-any.whl → 2.19.16__py3-none-any.whl

atomicshop/__init__.py

@@ -1,4 +1,4 @@
 """Atomic Basic functions and classes to make developer life easier"""
 
 __author__ = "Den Kras"
-__version__ = '2.19.14'
+__version__ = '2.19.16'

atomicshop/a_mains/update_config_toml.py

@@ -0,0 +1,38 @@
+import os.path
+import sys
+import argparse
+
+from atomicshop.file_io import tomls
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(
+        description='Update the config.toml file.\n'
+                    'This script will update the current config.toml file with the keys from the target config.toml file.\n'
+                    'If the key is not present in the current config.toml file, it will be added.\n'
+                    'If the key is present in the current config.toml file, its value will be left unchanged.\n')
+    parser.add_argument('current_config_file', type=str, help="Path to the current config.toml file that will be updated.")
+    parser.add_argument('target_config_file', type=str, help="Path to the target config.toml file that we'll get the updates from.")
+    parser.add_argument('-n', '--new_file_path', type=str, help="(OPTIONAL) Path to the new config.toml file that will be created with the updates.", default=None)
+    return parser.parse_args()
+
+
+def main():
+    args = parse_args()
+
+    if os.path.isfile(args.current_config_file) is False:
+        print(f"Error: The current config file '{args.current_config_file}' does not exist.")
+        return 1
+    if os.path.isfile(args.target_config_file) is False:
+        print(f"Error: The target config file '{args.target_config_file}' does not exist.")
+        return 1
+
+    tomls.update_toml_file_with_new_config(
+        main_config_file_path=args.current_config_file,
+        changes_config_file_path=args.target_config_file,
+        new_config_file_path=args.new_file_path)
+    return 0
+
+
+if __name__ == '__main__':
+    sys.exit(main())

atomicshop/file_io/tomls.py

@@ -5,6 +5,7 @@ try:
     import tomllib
 except ModuleNotFoundError:
     # This is library from pypi.
+    # noinspection PyPackageRequirements
     import tomli as tomllib
 
 from . import file_io
@@ -14,6 +15,7 @@ class TomlValueNotImplementedError(Exception):
     pass
 
 
+# noinspection PyUnusedLocal
 @file_io.read_file_decorator
 def read_toml_file(file_path: str,
                    file_mode: str = 'rb',
@@ -35,6 +37,7 @@ def read_toml_file(file_path: str,
     return tomllib.load(file_object)
 
 
+# noinspection PyUnusedLocal
 @file_io.write_file_decorator
 def write_toml_file(
         toml_content: dict,
@@ -112,7 +115,11 @@ def update_toml_file_with_new_config(
         If not, the changes will be written to the 'main_config_file_path'.
     """
 
-    # Sync the config files.
+    if not changes_config_file_path and not changes_dict:
+        raise ValueError("You must provide either 'changes_config_file_path' or 'changes_dict'.")
+    if changes_config_file_path and changes_dict:
+        raise ValueError("You can't provide both 'changes_config_file_path' and 'changes_dict'.")
+
     with open(main_config_file_path, 'r') as file:
         main_config_file_text_lines: list = file.readlines()
 
@@ -121,30 +128,29 @@ def update_toml_file_with_new_config(
     # Read the new config file.
     main_config_file_dict: dict = read_toml_file(main_config_file_path)
 
-    if changes_dict:
-        changes_config_file_dict = changes_dict
-    else:
-        changes_config_file_dict: dict = read_toml_file(changes_config_file_path)
+    if not changes_dict:
+        changes_dict: dict = read_toml_file(changes_config_file_path)
 
     # Update the config text lines.
     for category, settings in main_config_file_dict.items():
-        if category not in changes_config_file_dict:
+        if category not in changes_dict:
             continue
 
         for key, value in settings.items():
             # If the key is in the old config file, use the old value.
-            if key not in changes_config_file_dict[category]:
+            if key not in changes_dict[category]:
                 continue
 
-            if main_config_file_dict[category][key] != changes_config_file_dict[category][key]:
+            if main_config_file_dict[category][key] != changes_dict[category][key]:
                 # Get the line of the current category line.
                 current_category_line_index_in_text = None
                 for current_category_line_index_in_text, line in enumerate(main_config_file_text_lines):
                     if f"[{category}]" in line:
                         break
 
-                # current_category_line_index_in_text = main_config_file_text_lines.index(f"[{category}]\n")
-                current_category_index_in_main_config_dict = list(main_config_file_dict.keys()).index(category)
+                # Get the index inside the main config file dictionary of the current category.
+                main_config_list_of_keys: list = list(main_config_file_dict.keys())
+                current_category_index_in_main_config_dict = main_config_list_of_keys.index(category)
 
                 try:
                     next_category_name = list(
@@ -191,20 +197,20 @@ def update_toml_file_with_new_config(
                     # noinspection PyUnboundLocalVariable
                     comment = main_config_file_text_lines[line_index].replace(string_to_check, '')
 
-                    object_type = type(changes_config_file_dict[category][key])
+                    object_type = type(changes_dict[category][key])
                     if object_type == bool:
-                        value_string_to_set = str(changes_config_file_dict[category][key]).lower()
+                        value_string_to_set = str(changes_dict[category][key]).lower()
                     elif object_type == str:
-                        value_string_to_set = f"'{changes_config_file_dict[category][key]}'"
+                        value_string_to_set = f"'{changes_dict[category][key]}'"
                     elif object_type == int:
-                        value_string_to_set = str(changes_config_file_dict[category][key])
+                        value_string_to_set = str(changes_dict[category][key])
 
                     # noinspection PyUnboundLocalVariable
                     line_to_set = f"{key} = {value_string_to_set}{comment}"
                     # Replace the line with the old value.
                     main_config_file_text_lines[line_index] = line_to_set
 
-                    main_config_file_dict[category][key] = changes_config_file_dict[category][key]
+                    main_config_file_dict[category][key] = changes_dict[category][key]
 
     if new_config_file_path:
         file_path_to_write = new_config_file_path

{atomicshop-2.19.14.dist-info → atomicshop-2.19.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atomicshop
-Version: 2.19.14
+Version: 2.19.16
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
 License: MIT License

{atomicshop-2.19.14.dist-info → atomicshop-2.19.16.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=jnzkSiVKr7rNaRPe51te8qpwDBNj2V37gzYMQVqPkoQ,124
+atomicshop/__init__.py,sha256=iSWSvU0vNJjZowFs0TPAYg5UgE0d7XERSmYHkUprfXg,124
 atomicshop/_basics_temp.py,sha256=6cu2dd6r2dLrd1BRNcVDKTHlsHs_26Gpw8QS6v32lQ0,3699
 atomicshop/_create_pdf_demo.py,sha256=Yi-PGZuMg0RKvQmLqVeLIZYadqEZwUm-4A9JxBl_vYA,3713
 atomicshop/_patch_import.py,sha256=ENp55sKVJ0e6-4lBvZnpz9PQCt3Otbur7F6aXDlyje4,6334
@@ -64,6 +64,7 @@ atomicshop/a_mains/dns_gateway_setting.py,sha256=ncc2rFQCChxlNP59UshwmTonLqC6MWb
 atomicshop/a_mains/github_wrapper.py,sha256=F-PoZknVCxWPN0PTO6l7ZNiaYvo7OVFKFI_zlPt56ps,169
 atomicshop/a_mains/msi_unpacker.py,sha256=5hrkqETYt9HIqR_3PMf32_q06kCrIcsdm_RJV9oY438,188
 atomicshop/a_mains/search_for_hyperlinks_in_docx.py,sha256=HkIdo_Sz9nPbbbJf1mwfwFkyI7vkvpH8qiIkuYopN4w,529
+atomicshop/a_mains/update_config_toml.py,sha256=s-x_vmWCbB7jljNhR7rviM0IM1_CyxvybMpAwNjkBto,1664
 atomicshop/a_mains/FACT/factw_fact_extractor_docker_image_main_sudo.py,sha256=DDKX3Wp2SmzMCEtCIEOUbEKMob2ZQ7VEQGLEf9uYXrs,320
 atomicshop/a_mains/FACT/update_extract.py,sha256=atNIBKDeGAM1XNymDjnEHq7Ke3i_rXlAfrT-1pDdz_Q,544
 atomicshop/addons/PlayWrightCodegen.cmd,sha256=Z5cnllsyXD4F1W2h-WLEnyFkg5nZy0-hTGHRWXVOuW4,173
@@ -128,7 +129,7 @@ atomicshop/file_io/csvs.py,sha256=zv0kKjRT-ZWRi0WpMIUQ_FKyP9Dt0f5Bc98Qsj6ClPU,94
 atomicshop/file_io/docxs.py,sha256=Nyt3hSpzwqUKZEP5p5efqNpjFs9XqkK40Kp7BbbPo7E,6245
 atomicshop/file_io/file_io.py,sha256=8pKH08ILN91gfeNC0hrtLCx7l9ZgmcPAYIcLydDtZ-k,7198
 atomicshop/file_io/jsons.py,sha256=GJyVlGTcdhjZ-andfz7_dYVk1O74f-23Mc9PDmdjtPI,7424
-atomicshop/file_io/tomls.py,sha256=ol8EvQPf9sryTmZUf1v55BYSUQ6ml7HVVBHpNKbsIlA,9768
+atomicshop/file_io/tomls.py,sha256=vZ_Wng5alLf8z6HSEZj7PS0XKDA-Iies9ihVWOkTcKo,10048
 atomicshop/file_io/xlsxs.py,sha256=v_dyg9GD4LqgWi6wA1QuWRZ8zG4ZwB6Dz52ytdcmmmI,2184
 atomicshop/file_io/xmls.py,sha256=zh3SuK-dNaFq2NDNhx6ivcf4GYCfGM8M10PcEwDSpxk,2104
 atomicshop/mitm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -329,8 +330,8 @@ atomicshop/wrappers/socketw/statistics_csv.py,sha256=fgMzDXI0cybwUEqAxprRmY3lqbh
 atomicshop/wrappers/winregw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/winregw/winreg_installed_software.py,sha256=Qzmyktvob1qp6Tjk2DjLfAqr_yXV0sgWzdMW_9kwNjY,2345
 atomicshop/wrappers/winregw/winreg_network.py,sha256=AENV88H1qDidrcpyM9OwEZxX5svfi-Jb4N6FkS1xtqA,8851
-atomicshop-2.19.14.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-2.19.14.dist-info/METADATA,sha256=1ZQp-Wt5osHkFGnEavFkPEC41-jmfqJsSsktj0T7aBk,10631
-atomicshop-2.19.14.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-atomicshop-2.19.14.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-2.19.14.dist-info/RECORD,,
+atomicshop-2.19.16.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-2.19.16.dist-info/METADATA,sha256=uDnHTYzOlT54oM3QgJckEUYc5iDwE6wBipRH9S1aUB8,10631
+atomicshop-2.19.16.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+atomicshop-2.19.16.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-2.19.16.dist-info/RECORD,,