atomicshop 2.18.27__py3-none-any.whl → 2.18.29__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.18.27'
+__version__ = '2.18.29'

atomicshop/basics/multiprocesses.py

@@ -125,8 +125,8 @@ class MultiProcessorRecursive:
                         result = async_result.get()
                         # Assuming process_function returns a list, extend new_input_list
                         new_input_list.extend(result)
-                    except Exception as e:
-                        print(f"An error occurred: {e}")
+                    except Exception:
+                        raise
 
                 # Update the input_list for the next iteration
                 self.input_list = new_input_list

atomicshop/file_io/file_io.py

@@ -1,5 +1,6 @@
 from typing import Union
 import functools
+import os
 
 from .. import print_api
 from .. import inspect_wrapper
@@ -23,7 +24,7 @@ def write_file_decorator(function_name):
         print_api.print_api(message=f"Writing file: {kwargs['file_path']}", **kwargs)
 
         enable_long_file_path = kwargs.get('enable_long_file_path', False)
-        if enable_long_file_path:
+        if enable_long_file_path and os.name == 'nt':
             # A simpler string method would be to add '\\?\' to the beginning of the file path.
             # kwargs['file_path'] = rf"\\?\{kwargs['file_path']}"
 

atomicshop/file_io/jsons.py

@@ -133,3 +133,52 @@ def is_dict_json_serializable(
             raise e
         else:
             return False, str(e)
+
+
+def append_to_json(
+        dict_or_list: Union[dict, list],
+        json_file_path: str,
+        indent=None,
+        use_default_indent=False,
+        enable_long_file_path=False,
+        print_kwargs: dict = None
+) -> None:
+    """
+    Append dictionary or list of dictionaries to json file.
+
+    :param dict_or_list: dictionary or list of dictionaries to append.
+    :param json_file_path: full file path to the json file.
+    :param indent: integer number of spaces for indentation.
+        If 'ident=0' new lines still will be created. The most compact is 'indent=None' (from documentation)
+        So, using default as 'None' and not something else.
+    :param use_default_indent: boolean. Default indent for 'json' format in many places is '2'. So, if you don't want
+        to set 'indent=2', just set this to 'True'.
+    :param enable_long_file_path: Boolean, by default Windows has a limit of 260 characters for file path. If True,
+        the long file path will be enabled, and the limit will be 32,767 characters.
+    :param print_kwargs: dict, the print_api arguments.
+    :return:
+    """
+
+    # Read existing data from the file
+    try:
+        with open(json_file_path, 'r') as f:
+            current_json_file = json.load(f)
+    except FileNotFoundError:
+        current_json_file: list = []
+
+    # Append the new message to the existing data
+    final_json_list_of_dicts: list[dict] = []
+    if isinstance(current_json_file, list):
+        current_json_file.append(dict_or_list)
+        final_json_list_of_dicts = current_json_file
+    elif isinstance(current_json_file, dict):
+        final_json_list_of_dicts.append(current_json_file)
+        final_json_list_of_dicts.append(dict_or_list)
+    else:
+        error_message = "The current file is neither a list nor a dictionary."
+        raise TypeError(error_message)
+
+    # Write the data back to the file
+    write_json_file(
+        final_json_list_of_dicts, json_file_path, indent=indent, use_default_indent=use_default_indent,
+        enable_long_file_path=enable_long_file_path, **print_kwargs)

atomicshop/filesystem.py

@@ -329,6 +329,30 @@ def remove_directory(directory_path: str, force_readonly: bool = False, print_kw
         return False
 
 
+def clear_directory(directory: str) -> tuple[list[str], list[str]]:
+    """
+    The function will clear the directory of all files and subdirectories.
+    :param directory:
+    :return: tuple of lists of removed file paths and removed directory paths.
+    """
+
+    file_paths: list = []
+    directory_paths: list = []
+    # Iterate through all files and subdirectories in the directory
+    for item in os.listdir(directory):
+        item_path = os.path.join(directory, item)
+        # If it's a file, remove it
+        if os.path.isfile(item_path) or os.path.islink(item_path):  # Handle symbolic links too
+            os.remove(item_path)
+            file_paths.append(item_path)
+        # If it's a directory, remove it and its contents
+        elif os.path.isdir(item_path):
+            shutil.rmtree(item_path)
+            directory_paths.append(item_path)
+
+    return file_paths, directory_paths
+
+
 def remove_empty_directories(directory_path: str) -> list[str]:
     """
     Recursively removes empty directories in the given path, including the given path if it is empty.
@@ -675,12 +699,12 @@ def move_folder_contents_to_folder(
 
         if os.path.isdir(s):
             if os.path.exists(d) and not overwrite:
-                print(f"Directory {d} already exists. Skipping due to overwrite=False.")
+                raise FileExistsError(f"Directory already exists: {d}. Skipping due to overwrite=False.")
             else:
                 shutil.move(s, d)
         else:
             if os.path.exists(d) and not overwrite:
-                print(f"File {d} already exists. Skipping due to overwrite=False.")
+                raise FileExistsError(f"File {d} already exists. Skipping due to overwrite=False.")
             else:
                 shutil.move(s, d)
 
@@ -1675,3 +1699,88 @@ def find_file(file_name: str, directory_path: str):
             if filename == file_name:
                 return os.path.join(dir_path, filename)
     return None
+
+
+def create_ubuntu_desktop_shortcut(
+        file_path: str = None,
+        shortcut_name: str = None,
+        command: str = None,
+        working_directory: str = None,
+        icon_path: str = None,
+        terminal: bool = False,
+        comment: str = "Shortcut to execute the Python script",
+        categories: str = "Utility",
+        set_executable: bool = False,
+        set_xfce_exe_checksum: bool = False
+):
+    """
+    Create a desktop shortcut on Ubuntu.
+
+    Either file_path or command must be specified.
+    If command is specified, working_directory must be specified.
+
+    :param file_path: string, The file_path to execute when the shortcut is clicked.
+        Example2: '/path/to/script.sh'
+    :param shortcut_name: string, The name of the shortcut.
+        Example: 'My Python Script'
+        Result: 'My Python Script.desktop'
+    :param command: string, The command to execute when the shortcut is clicked.
+        Example: 'python3 /path/to/script.py'
+    :param working_directory: string, The working directory for the command.
+        If None, the command will be executed in the same script's directory.
+    :param icon_path: string, The path to the icon file.
+    :param terminal: boolean, If True, the command will be executed in a terminal.
+    :param comment: string, A comment to describe the shortcut.
+    :param categories: string, The categories of the shortcut.
+    :param set_executable: boolean, If True, the shortcut will be made executable.
+    :param set_xfce_exe_checksum: boolean, If True, the shortcut will be made safe executable for XFCE.
+
+    :return: None
+    """
+
+    if not file_path and not command:
+        raise ValueError("Either 'file_path' or 'command' must be specified.")
+    if command and file_path:
+        raise ValueError("Only one of 'file_path' or 'command' can be specified.")
+    if command and not working_directory:
+        raise ValueError("Working directory must be specified if 'command' is specified.")
+
+    from .permissions import ubuntu_permissions
+
+    # Get the user's directory.
+    desktop_dir = os.path.expanduser("~/Desktop")
+
+    # Full path to the .desktop file
+    shortcut_path = os.path.join(desktop_dir, f"{shortcut_name}.desktop")
+
+    if not working_directory:
+        working_directory = os.path.dirname(file_path)
+
+    if not shortcut_name:
+        shortcut_name: str = Path(file_path).stem
+
+    # Generate the content for the .desktop file
+    desktop_entry = [
+        "[Desktop Entry]",
+        "Version=1.0",
+        "Type=Application",
+        f"Name={shortcut_name}",
+        f"Exec={file_path}",
+        f"Path={working_directory}",
+        f"Icon={icon_path}" if icon_path else "",
+        f"Terminal={'true' if terminal else 'false'}",
+        f"Comment={comment}",
+        f"Categories={categories};",
+    ]
+
+    # Write the .desktop file
+    with open(shortcut_path, "w") as shortcut_file:
+        shortcut_file.write("\n".join(line for line in desktop_entry if line))  # Skip empty lines
+
+    # Make the .desktop file executable
+    if set_executable:
+        ubuntu_permissions.set_executable(shortcut_path)
+
+    # Make the .desktop file safe executable for XFCE
+    if set_xfce_exe_checksum:
+        ubuntu_permissions.set_xfce_exe_checksum(shortcut_path)

atomicshop/mitm/engines/__parent/recorder___parent.py

@@ -113,30 +113,14 @@ def save_message_worker(
         if record_message_dict is None:
             break
 
-        # Read existing data from the file
         try:
-            with open(record_file_path, 'r') as f:
-                current_json_file = json.load(f)
-        except FileNotFoundError:
-            current_json_file: list = []
-
-        # Append the new message to the existing data
-        final_json_list_of_dicts: list[dict] = []
-        if isinstance(current_json_file, list):
-            current_json_file.append(record_message_dict)
-            final_json_list_of_dicts = current_json_file
-        elif isinstance(current_json_file, dict):
-            final_json_list_of_dicts.append(current_json_file)
-            final_json_list_of_dicts.append(record_message_dict)
-        else:
-            error_message = "The current file is neither a list nor a dictionary."
-            print_api(error_message, logger_method="critical", logger=logger)
-            raise TypeError(error_message)
-
-        # Write the data back to the file
-        jsons.write_json_file(
-            final_json_list_of_dicts, record_file_path, indent=2,
-            enable_long_file_path=True, **{'logger': logger})
+            jsons.append_to_json(
+                record_message_dict, record_file_path, indent=2,
+                enable_long_file_path=True, print_kwargs={'logger': logger}
+            )
+        except TypeError as e:
+            print_api(str(e), logger_method="critical", logger=logger)
+            raise e
 
         logger.info(f"Recorded to file: {record_file_path}")
 

atomicshop/permissions/ubuntu_permissions.py

@@ -33,6 +33,48 @@ def set_executable(file_path: str):
     os.chmod(file_path, os.stat(file_path).st_mode | stat.S_IXUSR)
 
 
+def set_trusted_executable(file_path: str):
+    """
+    Function sets the executable permission on a file and marks it as trusted.
+    :param file_path: str, path to the file.
+    :return:
+    """
+
+    # Check if the file exists
+    if not os.path.exists(file_path):
+        raise FileNotFoundError(f"The file does not exist: {file_path} ")
+
+    # Execute the `gio set` command
+    subprocess.run(
+        ["gio", "set", file_path, "metadata::trusted", "true"],
+        check=True
+    )
+
+
+def set_xfce_exe_checksum(desktop_file_path):
+    # Expand `~` to the full home directory path
+    desktop_file_path = os.path.expanduser(desktop_file_path)
+
+    # Ensure the file exists
+    if not os.path.exists(desktop_file_path):
+        raise FileNotFoundError(f"The file does not exist: {desktop_file_path} ")
+
+    # Calculate the SHA256 checksum of the file
+    result = subprocess.run(
+        ["sha256sum", desktop_file_path],
+        stdout=subprocess.PIPE,
+        check=True,
+        text=True
+    )
+    checksum = result.stdout.split()[0]
+
+    # Set the metadata::xfce-exe-checksum attribute using `gio`
+    subprocess.run(
+        ["gio", "set", "-t", "string", desktop_file_path, "metadata::xfce-exe-checksum", checksum],
+        check=True
+    )
+
+
 def change_file_owner(file_path: str, username: str):
     """
     Function changes the owner of the file to the specified user.

atomicshop/tempfiles.py

@@ -1,4 +1,3 @@
-# v1.0.3 - 01.04.2021 - 15:30
 import os
 import datetime
 import tempfile
@@ -59,4 +58,4 @@ class TempFile:
         self.file_path: str = self.directory + os.sep + self.file_name
 
     def remove(self):
-        os.remove(self.file_path)
+        os.remove(self.file_path)

{atomicshop-2.18.27.dist-info → atomicshop-2.18.29.dist-info}/METADATA

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

{atomicshop-2.18.27.dist-info → atomicshop-2.18.29.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=cgFc16qqVGnOgJgV1Zh4BmuMSDReWteBDmPf5OF4EiI,124
+atomicshop/__init__.py,sha256=vD4D8D_jAMvDG5p9XvZfPSYd4zd_0MaNWcj-D9lkvE8,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
@@ -14,7 +14,7 @@ atomicshop/dns.py,sha256=5Gimq_WY2arqg7BeGmR7P--fGfnH0Dsh8lrOt_H0jRY,6817
 atomicshop/domains.py,sha256=Rxu6JhhMqFZRcoFs69IoEd1PtYca0lMCG6F1AomP7z4,3197
 atomicshop/emails.py,sha256=I0KyODQpIMEsNRi9YWSOL8EUPBiWyon3HRdIuSj3AEU,1410
 atomicshop/file_types.py,sha256=-0jzQMRlmU1AP9DARjk-HJm1tVE22E6ngP2mRblyEjY,763
-atomicshop/filesystem.py,sha256=StVeiGCjlpvMOmB-4SW-s7eR_qxYofW9jsfh1ewpaG4,63263
+atomicshop/filesystem.py,sha256=jSFHv34KhR4r2dt65pUYNy3bCqBtnOiqqxOo_vdrmdA,67573
 atomicshop/functions.py,sha256=pK8hoCE9z61PtWCxQJsda7YAphrLH1wxU5x-1QJP-sY,499
 atomicshop/get_process_list.py,sha256=8cxb7gKe9sl4R6H2yMi8J6oe-RkonTvCdKjRFqi-Fs4,6075
 atomicshop/get_process_name_cmd_dll.py,sha256=CtaSp3mgxxJKCCVW8BLx6BJNx4giCklU_T7USiCEwfc,5162
@@ -39,7 +39,7 @@ atomicshop/ssh_remote.py,sha256=Mxixqs2-xGy1bhbcP0LKqjxKTNPz1Gmzz8PzO8aLB4c,1734
 atomicshop/sys_functions.py,sha256=MTBxRve5bh58SPvhX3gMiGqHlSBuI_rdNN1NnnBBWqI,906
 atomicshop/system_resource_monitor.py,sha256=yHdBU4mAVqoVS0Nn_SM_H42i4PgsgXDaXaMxfnL5CgA,14588
 atomicshop/system_resources.py,sha256=iKUvVSaXR47inmr3cTYsgNfclT38dRia2oupnlhIpK4,9290
-atomicshop/tempfiles.py,sha256=uq1ve2WlWehZ3NOTXJnpBBMt6HyCdBufqedF0HyzA6k,2517
+atomicshop/tempfiles.py,sha256=cI8IlINHAcqomBtLOQIedYDE--1J5GcUsFuOQ6fDn_Y,2484
 atomicshop/timer.py,sha256=7Zw1KRV0acHCRATMnanyX2MLBb63Hc-6us3rCZ9dNlY,2345
 atomicshop/urls.py,sha256=aJ0NGS9qqaKeqjkkWBs80jaBBg6MYBiPuLIyPGxscVc,1557
 atomicshop/uuids.py,sha256=JSQdm3ZTJiwPQ1gYe6kU0TKS_7suwVrHc8JZDGYlydM,2214
@@ -100,7 +100,7 @@ atomicshop/basics/isinstancing.py,sha256=fQ35xfqbguQz2BUn-3a4KVGskhTcIn8JjRtxV2r
 atomicshop/basics/list_of_classes.py,sha256=PJoE1VJdhhQ4gSFr88zW7IApXd4Ez7xLz-7vAM-7gug,978
 atomicshop/basics/list_of_dicts.py,sha256=tj0LNPf1ljNI_qpoO-PiOT4Ulmk1M-UpTGyn9twVcw8,8039
 atomicshop/basics/lists.py,sha256=I0C62vrDrNwCTNl0EjUZNa1Jsd8l0rTkp28GEx9QoEI,4258
-atomicshop/basics/multiprocesses.py,sha256=nSskxJSlEdalPM_Uf8cc9kAYYlVwYM1GonBLAhCL2mM,18831
+atomicshop/basics/multiprocesses.py,sha256=oU6LjcLLGBtPIGJzZBpDWoLU3HRmMoanITEOE2luAYw,18799
 atomicshop/basics/numbers.py,sha256=ESX0z_7o_ok3sOmCKAUBoZinATklgMy2v-4RndqXlVM,1837
 atomicshop/basics/package_module.py,sha256=fBd0uVgFce25ZCVtLq83iyowRlbwdWYFj_t4Ml7LU14,391
 atomicshop/basics/randoms.py,sha256=DmYLtnIhDK29tAQrGP1Nt-A-v8WC7WIEB8Edi-nk3N4,282
@@ -119,8 +119,8 @@ atomicshop/etws/traces/trace_sysmon_process_creation.py,sha256=OM-bkK38uYMwWLZKN
 atomicshop/file_io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/file_io/csvs.py,sha256=zv0kKjRT-ZWRi0WpMIUQ_FKyP9Dt0f5Bc98Qsj6ClPU,9495
 atomicshop/file_io/docxs.py,sha256=Nyt3hSpzwqUKZEP5p5efqNpjFs9XqkK40Kp7BbbPo7E,6245
-atomicshop/file_io/file_io.py,sha256=5Kl0P6vF4GQVdwew1lzHLb-db9qiMvDjTgccbi5P-zk,7167
-atomicshop/file_io/jsons.py,sha256=q9ZU8slBKnHLrtn3TnbK1qxrRpj5ZvCm6AlsFzoANjo,5303
+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/xlsxs.py,sha256=v_dyg9GD4LqgWi6wA1QuWRZ8zG4ZwB6Dz52ytdcmmmI,2184
 atomicshop/file_io/xmls.py,sha256=zh3SuK-dNaFq2NDNhx6ivcf4GYCfGM8M10PcEwDSpxk,2104
@@ -140,7 +140,7 @@ atomicshop/mitm/engines/create_module_template.py,sha256=TAzsA4eLD2wYr7auuL4Nf_7
 atomicshop/mitm/engines/create_module_template_example.py,sha256=X5xhvbV6-g9jU_bQVhf_crZmaH50LRWz3bS-faQ18ds,489
 atomicshop/mitm/engines/__parent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/mitm/engines/__parent/parser___parent.py,sha256=HHaCXhScl3OlPjz6eUxsDpJaZyk6BNuDMc9xCkeo2Ws,661
-atomicshop/mitm/engines/__parent/recorder___parent.py,sha256=exfElkgOU57hupV6opRCeYPHw91GIkIZL6UY3f2OClM,5635
+atomicshop/mitm/engines/__parent/recorder___parent.py,sha256=vG2h46CojK4jIinXmME2VwJFSWKygh4ayMwZyejwwI0,4846
 atomicshop/mitm/engines/__parent/responder___parent.py,sha256=yEgR33CkAd1b9jyH5E8I4E16upqjlcK3WLo3BFHdjyk,9157
 atomicshop/mitm/engines/__reference_general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/mitm/engines/__reference_general/parser___reference_general.py,sha256=57MEPZMAjTO6xBDZ-yt6lgGJyqRrP0Do5Gk_cgCiPns,2998
@@ -159,7 +159,7 @@ atomicshop/monitor/checks/process_running.py,sha256=x66wd6-l466r8sbRQaIli0yswyGt
 atomicshop/monitor/checks/url.py,sha256=1PvKt_d7wFg7rDMFpUejAQhj0mqWsmlmrNfjNAV2G4g,4123
 atomicshop/permissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/permissions/permissions.py,sha256=CYTDVOI0jh9ks0ZLnnOuPzppgCszFEc9-92DTkVTYi4,522
-atomicshop/permissions/ubuntu_permissions.py,sha256=xmLfrBh-uhJqmZzx754Yt2ECBsZI5uwWECg6hCsTtek,4649
+atomicshop/permissions/ubuntu_permissions.py,sha256=n8z1vcIXDts4zLVue33dtJiTopjgEAYPg4dqFn0qwwc,5943
 atomicshop/permissions/win_permissions.py,sha256=eDQm1jfK9x_hkbLqIJjFTwfqinAWQ0iSr0kW3XrF1BE,1272
 atomicshop/process_poller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/process_poller/process_pool.py,sha256=4Qs427qd7OcBxu5PMFU5PTmyuxRy0vgj2GLsRt0IoEw,9565
@@ -321,8 +321,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.18.27.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-2.18.27.dist-info/METADATA,sha256=P-gMfXaiasKYKaXDUsQbwZ-YtcwJ-nFJNVBIW9UbTB4,10631
-atomicshop-2.18.27.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-atomicshop-2.18.27.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-2.18.27.dist-info/RECORD,,
+atomicshop-2.18.29.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-2.18.29.dist-info/METADATA,sha256=qUnmPIO-_pwUQ74EV5jl84JWBrlUigxl-lavHMHKEHQ,10631
+atomicshop-2.18.29.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+atomicshop-2.18.29.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-2.18.29.dist-info/RECORD,,