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

atomicshop/networks.py

@@ -3,6 +3,7 @@ import time
 from typing import Union
 import os
 import psutil
+import ctypes
 
 from icmplib import ping
 from icmplib.models import Host
@@ -20,9 +21,12 @@ MICROSOFT_LOOPBACK_DEVICE_HARDWARE_ID = "*MSLOOP"
 GUID_DEVCLASS_NET: str = '{4d36e972-e325-11ce-bfc1-08002be10318}'
 
 
-def is_ip_alive(ip_address: str, timeout: int = 1) -> bool:
+def is_ip_in_use_ping(ip_address: str, timeout: int = 1) -> bool:
     """
-    Returns True if icmplib.models.Host.is_alive returns True.
+    Returns True if the IP address is pingable, False otherwise.
+    :param ip_address: string, IP address to check.
+    :param timeout: int, timeout in seconds. Default is 1 second.
+    :return: bool, True if the IP address is pingable, False otherwise.
     """
 
     host_object: Host = ping(ip_address, count=1, timeout=timeout)
@@ -30,15 +34,72 @@ def is_ip_alive(ip_address: str, timeout: int = 1) -> bool:
     return host_object.is_alive
 
 
-def get_default_internet_ipv4() -> str:
+def is_ip_in_use_arp(
+        ipv4: str,
+        gateway_ip: str = None
+) -> tuple[
+        Union[str, None],
+        Union[bool, None]
+]:
     """
+    Windows only.
+    Check if an IPv4 address is in use on the local network using ARP.
+    :param ipv4: string, IPv4 address to check.
+    :param gateway_ip: string, IPv4 address of the default gateway.
+        How it works: If you provide the gateway_ip, the function will get yje MAC of the gateway,
+        then it will get the MAC of the target IP address. If the MACs are the same, it means that the target IP's
+        ARP reply is an ARP proxy reply from the gateway.
+    :return: tuple (mac_address: str | None, via_gateway: bool | None)
+        If the IP address is in use, mac_address will be the MAC address of the device using the IP address,
+        else None. If gateway_ip is provided, via_gateway will be True if the MAC address is the same as the gateway's MAC address,
+        False if it's different, and None if gateway_ip is not provided.
+    """
+
+    iphlpapi = ctypes.windll.iphlpapi
+    ws2_32 = ctypes.windll.ws2_32
+
+    def _send_arp(ip: str) -> str | None:
+        """Return MAC string like 'aa:bb:cc:dd:ee:ff' if IP is claimed on the LAN, else None."""
+        # inet_addr returns DWORD in network byte order
+        dest_ip = ws2_32.inet_addr(ip.encode('ascii'))
+        if dest_ip == 0xFFFFFFFF:  # INVALID
+            raise ValueError(f"Bad IPv4 address: {ip}")
+
+        mac_buf = ctypes.c_uint64(0)  # storage for up to 8 bytes
+        mac_len = ctypes.c_ulong(ctypes.sizeof(mac_buf))  # in/out len
+        # SrcIP=0 lets Windows pick the right interface
+        rc = iphlpapi.SendARP(dest_ip, 0, ctypes.byref(mac_buf), ctypes.byref(mac_len))
+        if rc != 0:  # Non-zero means no ARP reply / not on-link / other error
+            return None
+
+        # Extract the first 6 bytes from little-endian integer
+        mac_int = mac_buf.value
+        mac_bytes = mac_int.to_bytes(8, 'little')[:6]
+        return ':'.join(f'{b:02x}' for b in mac_bytes)
+
+    mac = _send_arp(ipv4)
+    if mac is None:
+        return None, None
+    via_gateway = None
+    if gateway_ip:
+        gw_mac = _send_arp(gateway_ip)
+        via_gateway = (gw_mac is not None and gw_mac.lower() == mac.lower())
+    return mac, via_gateway
+
+
+def __get_default_internet_ipv4() -> str:
+    """
+    FOR REFERENCE ONLY, DO NOT USE.
+    DOESN'T WORK UNDER ALL CIRCUMSTANCES, CAN'T PINPOINT THE REASON.
+
     Get the default IPv4 address of the interface that is being used for internet.
     :return: string, default IPv4 address.
     """
 
     return socket.gethostbyname(socket.gethostname())
 
-def get_default_internet_ipv4_by_connect(target: str = "8.8.8.8") -> str:
+
+def get_default_internet_ipv4(target: str = "8.8.8.8") -> str:
     with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
         s.connect((target, 80))  # no packet sent; OS just chooses a route
         return s.getsockname()[0]  # local address of that route
@@ -296,9 +357,9 @@ def generate_unused_ipv4_addresses_by_vlan(
     for i in range(number_of_ips):
         # Create the IP address.
         while True:
-            ip_address = f"{vlan}.{counter}"
+            ip_address: str = f"{vlan}.{counter}"
             counter += 1
-            is_ip_in_use: bool = is_ip_alive(ip_address)
+            is_ip_in_use, _ = is_ip_in_use_arp(ip_address)
             if not is_ip_in_use and not ip_address in skip_ips:
                 # print("[+] Found IP to assign: ", ip_address)
                 generated_ips.append(ip_address)
@@ -506,7 +567,7 @@ def add_virtual_ips_to_default_adapter_by_current_setting(
                 dns_gateways = default_adapter_info['dns_gateways']
 
             # We will get the default IP address of the machine.
-            default_ip_address: str = socket.gethostbyname(socket.gethostname())
+            default_ip_address: str = get_default_internet_ipv4()
             # So we can make it the first IP in the list, but first remove it from the list.
             _ = ips.pop(ips.index(default_ip_address))
             # At this point we will copy the list of IPs that we will set the SkipAsSource flag for.

{atomicshop-3.3.5.dist-info → atomicshop-3.3.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomicshop
-Version: 3.3.5
+Version: 3.3.6
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
 License-Expression: MIT

{atomicshop-3.3.5.dist-info → atomicshop-3.3.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=lCOFyywOc7EF6Esxu6aQIR9lx9HA8beRu1ut7lEQ3oU,122
+atomicshop/__init__.py,sha256=PKxHKogNO1jVrjMDxPohtYhd5093pVEONPVfmujNgfw,122
 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
@@ -23,7 +23,7 @@ atomicshop/http_parse.py,sha256=1Tna9YbOM0rE3t6i_M-klBlwd1KNSA9skA_BqKGXDFc,1186
 atomicshop/inspect_wrapper.py,sha256=sGRVQhrJovNygHTydqJj0hxES-aB2Eg9KbIk3G31apw,11429
 atomicshop/ip_addresses.py,sha256=penRFeJ1-LDVTko4Q0EwK4JiN5cU-KzCBR2VXg9qbUY,1238
 atomicshop/keyboard_press.py,sha256=1W5kRtOB75fulVx-uF2yarBhW0_IzdI1k73AnvXstk0,452
-atomicshop/networks.py,sha256=dTh6T9vAnjYOEPdfpL6W6wN091HwRbVLeNscXWZX_DI,22151
+atomicshop/networks.py,sha256=fwOMMHwn-XMq0WgYCFbo34NXW-Xu-4nDfWl3T5EsE9U,24807
 atomicshop/on_exit.py,sha256=9XlOnzoAG8zlI8wBF4AB8hyrC6Q1b84gkhqpAhhdN9g,6977
 atomicshop/pbtkmultifile_argparse.py,sha256=aEk8nhvoQVu-xyfZosK3ma17CwIgOjzO1erXXdjwtS4,4574
 atomicshop/print_api.py,sha256=SJNQIMqSLlYaPtjHnALySAI-jQYuYHOCGgfP7oe96fU,10957
@@ -58,7 +58,6 @@ atomicshop/a_installs/ubuntu/pycharm.py,sha256=Ld7YQBwPxrjuZcTG1K4kZqjbBdt8aooCV
 atomicshop/a_installs/win/fibratus.py,sha256=TU4e9gdZ_zI73C40uueJ59pD3qmN-UFGdX5GFoVf6cM,179
 atomicshop/a_installs/win/mongodb.py,sha256=AqyItXu19aaoe49pppDxtEkXey6PMy0PoT2Y_RmPpPE,179
 atomicshop/a_installs/win/pycharm.py,sha256=j_RSd7aDOyC3yDd-_GUTMLlQTmDrqtVFG--oUfGLiZk,140
-atomicshop/a_installs/win/robocorp.py,sha256=Ob8X9Czwd-OR3pudFIFft1dcUjDM0Pw0X03yfPaf3yw,3280
 atomicshop/a_installs/win/wsl_ubuntu_lts.py,sha256=dZbPRLNKFeMd6MotjkE6UDY9cOiIaaclIdR1kGYWI50,139
 atomicshop/a_mains/dns_gateway_setting.py,sha256=ncc2rFQCChxlNP59UshwmTonLqC6MWblrVAzbbz-13M,149
 atomicshop/a_mains/github_wrapper.py,sha256=F-PoZknVCxWPN0PTO6l7ZNiaYvo7OVFKFI_zlPt56ps,169
@@ -331,9 +330,9 @@ atomicshop/wrappers/socketw/statistics_csv.py,sha256=_gA8bMX6Sw_UCXKi2y9wNAwlqif
 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=ih0BVNwByLvf9F_Lac4EdmDYYJA3PzMvmG0PieDZrsE,9905
-atomicshop-3.3.5.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.3.5.dist-info/METADATA,sha256=QAiSikiYm60q_UBleSyptTY9jvGM6s_7fk2yrncVrec,9311
-atomicshop-3.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-atomicshop-3.3.5.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
-atomicshop-3.3.5.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.3.5.dist-info/RECORD,,
+atomicshop-3.3.6.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.3.6.dist-info/METADATA,sha256=IL5-xRKSd3KMQq1-z8F1hdeVTz9jNmhxtEdbXTAeKtM,9311
+atomicshop-3.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+atomicshop-3.3.6.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
+atomicshop-3.3.6.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.3.6.dist-info/RECORD,,

atomicshop/a_installs/win/robocorp.py

@@ -1,82 +0,0 @@
-import sys
-import subprocess
-import tempfile
-
-from atomicshop.print_api import print_api
-from atomicshop.wrappers import githubw
-from atomicshop.permissions import permissions
-from atomicshop.wrappers.nodejsw import install_nodejs_windows
-
-
-WINDOWS_TESSERACT_DEFAULT_INSTALLATION_DIRECTORY: str = r"C:\Program Files\Tesseract-OCR"
-
-
-def main():
-    if not permissions.is_admin():
-        print_api("Please run this script as an Administrator.", color="red")
-        return 1
-
-    if not install_nodejs_windows.is_nodejs_installed():
-        install_nodejs_windows.install_nodejs_windows()
-    install_nodejs_windows.add_nodejs_to_path()
-    if not install_nodejs_windows.is_nodejs_installed():
-        print_api("Node.js installation failed.", color="red")
-        return 1
-
-    print_api("PIP Installing Robocorp.", color="blue")
-    subprocess.check_call(["pip", "install", "--upgrade", "rpaframework"])
-
-    print_api("PIP Installing Robocorp-Browser.", color="blue")
-    subprocess.check_call(["pip", "install", "--upgrade", "robotframework-browser"])
-
-    print_api("PIP Installing Robocorp-Recognition.", color="blue")
-    subprocess.check_call(["pip", "install", "--upgrade", "rpaframework-recognition"])
-
-    print_api("PIP Installing pynput.", color="blue")
-    subprocess.check_call(["pip", "install", "--upgrade", "pynput"])
-
-    print_api("Installing Playwright browsers.", color="blue")
-    subprocess.check_call(["playwright", "install"])
-
-    print_api("Initializing Robocorp Browser.", color="blue")
-    subprocess.check_call(["rfbrowser", "init"])
-
-    print_api("Installing Additional modules.", color="blue")
-    subprocess.check_call(["pip", "install", "--upgrade", "matplotlib", "imagehash"])
-
-    print_api("Installing Tesseract OCR.", color="blue")
-    github_wrapper = githubw.GitHubWrapper(
-        user_name="tesseract-ocr",
-        repo_name="tesseract",
-        branch="main")
-    github_wrapper.build_links_from_user_and_repo()
-    temp_file_path: str = tempfile.gettempdir()
-    tesseract_installer = github_wrapper.download_latest_release(
-        target_directory=temp_file_path,
-        string_pattern="*tesseract*exe")
-
-    # The Admin needed to install Tesseract.
-    subprocess.check_call([tesseract_installer, "/S"])
-
-    # Add Tesseract to the PATH.
-    subprocess.check_call(["setx", "PATH", f"%PATH%;{WINDOWS_TESSERACT_DEFAULT_INSTALLATION_DIRECTORY}"])
-
-    # Patch robocorp: Remove mouse to the center of the screen on control command.
-    # Import the library to find its path.
-    print_api("Patching: .\RPA\Windows\keywords\window.py", color="blue")
-    import RPA.Windows.keywords.window as window
-    window_file_path = window.__file__
-
-    # Patch the file.
-    with open(window_file_path, "r") as file:
-        file_content = file.read()
-    file_content = file_content.replace(
-        "window.item.MoveCursorToMyCenter(simulateMove=self.ctx.simulate_move)",
-        "# window.item.MoveCursorToMyCenter(simulateMove=self.ctx.simulate_move)    # Patched to remove center placement during foreground window control."
-    )
-    with open(window_file_path, "w") as file:
-        file.write(file_content)
-
-
-if __name__ == '__main__':
-    sys.exit(main())