atomicshop 2.18.28__py3-none-any.whl → 2.18.30__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.28'
+__version__ = '2.18.30'

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/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,95 @@ 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_trusted: 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_trusted: boolean, If True, the shortcut will be marked as trusted.
+        This is needed for GNOME.
+    :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")
+
+    if not working_directory:
+        working_directory = os.path.dirname(file_path)
+
+    if not shortcut_name:
+        shortcut_name: str = Path(file_path).stem
+
+    # Full path to the .desktop file
+    shortcut_path = os.path.join(desktop_dir, f"{shortcut_name}.desktop")
+
+    # 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)
+
+    # Mark the .desktop file as trusted
+    if set_trusted:
+        ubuntu_permissions.set_trusted_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/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-2.18.28.dist-info → atomicshop-2.18.30.dist-info}/METADATA

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

{atomicshop-2.18.28.dist-info → atomicshop-2.18.30.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=NPXYaerttT8nwfrwAuSuQRk-VuX8kIyXrNdJ0gpi-e4,124
+atomicshop/__init__.py,sha256=6a8s52oEErBoMqHlOVc95TxLFuoK8ZGvWHPZKkVUgAA,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=vXyc73_8ltcsuvPrmFor8QtehZs0Dg3_FvflAoZKT00,67857
 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
@@ -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
@@ -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.28.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-2.18.28.dist-info/METADATA,sha256=SsZ4Ze8vESp5LjMbbM8_MghpZ9WhVYHs36BxPg63-CE,10631
-atomicshop-2.18.28.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-atomicshop-2.18.28.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-2.18.28.dist-info/RECORD,,
+atomicshop-2.18.30.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-2.18.30.dist-info/METADATA,sha256=c7ZY3AG2mw1LVujTZo1Svw7dQLemtJdNBBLUE_7obZc,10631
+atomicshop-2.18.30.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+atomicshop-2.18.30.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-2.18.30.dist-info/RECORD,,