atomicshop 3.1.4__py3-none-any.whl → 3.1.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.1.4'
+__version__ = '3.1.6'

atomicshop/networks.py

@@ -432,15 +432,26 @@ def add_virtual_ips_to_default_adapter_by_current_setting(
             if dns_gateways is None:
                 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())
+            # 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.
+            ips_for_skip_as_source = ips.copy()
+            # Add it back to the beginning of the list.
+            ips.insert(0, default_ip_address)
+
             win32_networkadapterconfiguration.set_static_ips(
                 default_network_adapter_config, ips=ips, masks=masks,
                 gateways=gateways, dns_gateways=dns_gateways,
                 availability_wait_seconds=availability_wait_seconds)
 
+            # If there were already virtual IPs assigned to the adapter and already were set SkipAsSource,
+            # we need to set SkipAsSource for them once again as well as for the new IPs.
             if set_virtual_ips_skip_as_source:
                 wmi_standard_cimv2_instance, _ = wmi_helpers.get_wmi_instance(
                     namespace='root\\StandardCimv2', wmi_instance=wmi_civ2_instance, locator=locator)
-                msft_netipaddress.set_skip_as_source(ips_to_assign, enable=True, wmi_instance=wmi_standard_cimv2_instance)
+                msft_netipaddress.set_skip_as_source(ips_for_skip_as_source, enable=True, wmi_instance=wmi_standard_cimv2_instance)
         else:
             # print("[!] No new IPs to assign.")
             pass

atomicshop/web.py

@@ -157,6 +157,7 @@ def download(
         target_directory: str = None,
         file_name: str = None,
         headers: dict = None,
+        use_certifi_ca_repository: bool = False,
         **kwargs
 ) -> str:
     """
@@ -168,6 +169,8 @@ def download(
     :param file_name: string, file name (example: file.zip) that you want the downloaded file to be saved as.
         If not specified, the default filename from 'file_url' will be used.
     :param headers: dictionary, HTTP headers to use when downloading the file.
+    :param use_certifi_ca_repository: boolean, if True, the certifi CA store will be used for SSL context
+        instead of the system's default CA store.
     :return: string, full file path of downloaded file. If download failed, 'None' will be returned.
     """
 
@@ -196,8 +199,15 @@ def download(
     print_api.print_api(f'Downloading: {file_url}', **kwargs)
     print_api.print_api(f'To: {file_path}', **kwargs)
 
-    # Open the URL for data gathering with SSL context with default CA store of the system.
-    ssl_context = ssl.create_default_context()
+    # Open the URL for data gathering with SSL context.
+    if not use_certifi_ca_repository:
+        # Create a default SSL context using the system's CA store.
+        ssl_context = ssl.create_default_context()
+    else:
+        # Create a default SSL context using the certifi CA store.
+        # This is useful for environments where the system's CA store is not available or not trusted.
+        # 'certifi.where()' returns the path to the certifi CA bundle.
+        ssl_context = ssl.create_default_context(cafile=certifi.where())
 
     # In order to use 'urllib.request', it is not enough to 'import urllib', you need to 'import urllib.request'.
     # Build a Request object with headers if provided.

atomicshop/wrappers/pycharmw/win.py

@@ -62,7 +62,7 @@ def download_install_main():
         print_api("Starting the download...")
         file_name = "pycharm-latest.exe"
         # download_file(download_url, file_name)
-        installer_path = web.download(file_url=download_url, file_name=file_name)
+        installer_path = web.download(file_url=download_url, file_name=file_name, use_certifi_ca_repository=True)
         print_api(f"Downloaded the latest version of PyCharm to {file_name}", color='green')
     except Exception as e:
         print_api(f"An error occurred: {e}")

atomicshop/wrappers/pywin32w/wmis/msft_netipaddress.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from win32com.client import CDispatch
 
 from . import wmi_helpers
@@ -73,4 +75,39 @@ def set_skip_as_source(
 
             obj.SkipAsSource = enable
             obj.Put_()                          # commit the change
-            print(f"[+] {ip}: SkipAsSource set to {enable}")
+            print(f"[+] {ip}: SkipAsSource set to {enable}")
+
+
+def is_skip_as_source(
+        ip_address: str,
+        wmi_instance: CDispatch = None
+) -> Optional[bool]:
+    """
+    Check whether *ip_address* currently has SkipAsSource set.
+
+    Returns
+    -------
+    True        – the flag is enabled
+    False       – the flag is disabled
+    None        – no MSFT_NetIPAddress object matches the IP (not present)
+
+    Notes
+    -----
+    * Works for both IPv4/IPv6.
+    * Uses the same Win32 CIM class (root\\StandardCimv2 › MSFT_NetIPAddress).
+    * You can pass an existing `wmi_instance` to avoid reconnecting in tight loops.
+    """
+    # Get a WMI connection if the caller didn’t hand us one
+    if not wmi_instance:
+        wmi_instance, _ = wmi_helpers.get_wmi_instance(namespace='root\\StandardCimv2')
+
+    query = f"SELECT SkipAsSource FROM MSFT_NetIPAddress WHERE IPAddress='{ip_address}'"
+    matches = wmi_instance.ExecQuery(query)
+
+    if not matches:                       # address not configured on this host/NIC
+        return None
+
+    # There should be only one entry per literal IP, but handle duplicates sanely
+    # Return True if *any* matching record has SkipAsSource = True,
+    # otherwise False (all are False).
+    return any(bool(obj.SkipAsSource) for obj in matches)

{atomicshop-3.1.4.dist-info → atomicshop-3.1.6.dist-info}/METADATA

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

{atomicshop-3.1.4.dist-info → atomicshop-3.1.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=9-xFtR4CZMUytv4s7xSLB71ElNOa9hz00EgvneY_IqE,122
+atomicshop/__init__.py,sha256=lgJ4Y2cZjapTRcmtqh6h2LaQj-h7MwBqzq-fHO09skc,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=XiQYQEPcBRX_HwNNHa1c8jN1OnRj6E8azcR24oFCkng,17922
+atomicshop/networks.py,sha256=OSFaEUu8otxCkNq9oMbYQpj5SLzHie2r5uJDiyLR19U,18685
 atomicshop/on_exit.py,sha256=9XlOnzoAG8zlI8wBF4AB8hyrC6Q1b84gkhqpAhhdN9g,6977
 atomicshop/pbtkmultifile_argparse.py,sha256=aEk8nhvoQVu-xyfZosK3ma17CwIgOjzO1erXXdjwtS4,4574
 atomicshop/print_api.py,sha256=SJNQIMqSLlYaPtjHnALySAI-jQYuYHOCGgfP7oe96fU,10957
@@ -47,7 +47,7 @@ atomicshop/uuids.py,sha256=JSQdm3ZTJiwPQ1gYe6kU0TKS_7suwVrHc8JZDGYlydM,2214
 atomicshop/venvs.py,sha256=D9lwOoObkYoRx-weuoAmbvN-RdSHhVm4DE9TVl-utAs,903
 atomicshop/versioning.py,sha256=e5W6m9AF3__M5nntqI9CqNAeHqkwY9JhlnpYeZ1CEus,970
 atomicshop/virtualization.py,sha256=LPP4vjE0Vr10R6DA4lqhfX_WaNdDGRAZUW0Am6VeGco,494
-atomicshop/web.py,sha256=gqlDdv7_uYdsbhUdwmrgz_l_S_8_N2fCMdxmfpukHyE,12411
+atomicshop/web.py,sha256=hkAS0MeMW_mGgxhFCX1xckT3yslCJ2cMP2nx9ADpe3o,13024
 atomicshop/websocket_parse.py,sha256=aLHWyKqaYqEn_MRBWm2L6rIl6QPmqbVrjEXE_rBzwCw,16711
 atomicshop/a_installs/ubuntu/docker_rootless.py,sha256=9IPNtGZYjfy1_n6ZRt7gWz9KZgR6XCgevjqq02xk-o0,281
 atomicshop/a_installs/ubuntu/docker_sudo.py,sha256=JzayxeyKDtiuT4Icp2L2LyFRbx4wvpyN_bHLfZ-yX5E,281
@@ -300,7 +300,7 @@ atomicshop/wrappers/psutilw/psutilw.py,sha256=q3EwgprqyrR4zLCjl4l5DHFOQoukEvQMIP
 atomicshop/wrappers/psycopgw/psycopgw.py,sha256=XJvVf0oAUjCHkrYfKeFuGCpfn0Oxj3u4SbKMKA1508E,7118
 atomicshop/wrappers/pycharmw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/pycharmw/ubuntu.py,sha256=m9MpgqvIYygulhPxo9g2zlGGXrihBpiY3GNLNyT-B7U,1290
-atomicshop/wrappers/pycharmw/win.py,sha256=jdnTkUqZX_BrMW8AmW-xGtxdV-wmmNr_NMA2jB6JHsQ,2725
+atomicshop/wrappers/pycharmw/win.py,sha256=hNP-d95z1zhcCpYqhHE5HZVYxaAlt8JJCNXh65jZsHc,2757
 atomicshop/wrappers/pywin32w/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/pywin32w/cert_store.py,sha256=dV1XyoKTFKZ-HCIVqU2Nd6CTZ8HANqjAXv26rsNzO6s,4365
 atomicshop/wrappers/pywin32w/console.py,sha256=LstHajPLgXp9qQxFNR44QfH10nOnNp3bCJquxaTquns,1175
@@ -312,7 +312,7 @@ atomicshop/wrappers/pywin32w/win_event_log/subscribes/process_create.py,sha256=I
 atomicshop/wrappers/pywin32w/win_event_log/subscribes/process_terminate.py,sha256=OJFWywGGGkBHq1N0MKGtHSFFQMFQSDVU6FXCRIdssg8,3761
 atomicshop/wrappers/pywin32w/win_event_log/subscribes/schannel_logging.py,sha256=8nxIcNcbeEuvoBwhujgh7-oIpL9A6J-gg1NM8hOGAVA,3442
 atomicshop/wrappers/pywin32w/wmis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/wrappers/pywin32w/wmis/msft_netipaddress.py,sha256=jCfLS81byOgBwtZWTRPzDWm2IUkcODy1REYjdN80fJo,3482
+atomicshop/wrappers/pywin32w/wmis/msft_netipaddress.py,sha256=Kn2gQsyEquM5QQkdF2Vtr1MqK5D7uo6EFi-cxt52Q0g,4800
 atomicshop/wrappers/pywin32w/wmis/win32_networkadapterconfiguration.py,sha256=Iz-p--cE2iP6wyD9ceSF3scMMvdXC5fF-VKP1v-iIDo,10813
 atomicshop/wrappers/pywin32w/wmis/win32networkadapter.py,sha256=iao4OK6u097uRacw6vQuya1kOtJzLA1IM4cgo6UCWLM,3941
 atomicshop/wrappers/pywin32w/wmis/win32process.py,sha256=qMzXtJ5hBZ5ydAyqpDbSx0nO2RJQL95HdmV5SzNKMhk,6826
@@ -336,8 +336,8 @@ atomicshop/wrappers/socketw/statistics_csv.py,sha256=WcNyaqEZ82S5-f3kzqi1nllNT2N
 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=3Ts1sVqSUiCDsHRHwJCbiZ9EYvv2ELGxF0Y_pibGU4k,9596
-atomicshop-3.1.4.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.1.4.dist-info/METADATA,sha256=OzJZl-P3897NdQPGhj4UKBbYKCQITS87eJIAt6WWa7E,10653
-atomicshop-3.1.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-atomicshop-3.1.4.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.1.4.dist-info/RECORD,,
+atomicshop-3.1.6.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.1.6.dist-info/METADATA,sha256=nOFaTyw74lY191rFSACO5nChQHK6jRcTq4yPfbvOj2A,10653
+atomicshop-3.1.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+atomicshop-3.1.6.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.1.6.dist-info/RECORD,,