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

atomicshop/networks.py

@@ -169,6 +169,59 @@ def get_interface_ips(
     return ips
 
 
+def get_host_ips(
+        localhost: bool = True,
+        ipv4: bool = True,
+        ipv6: bool = True
+) -> list[str]:
+    """
+    Yield (ifname, family, ip) for all UP interfaces that have bindable addresses.
+
+    Args:
+        localhost: include 127.0.0.0/8 and ::1 if True.
+        ipv4: include IPv4 addresses if True.
+        ipv6: include IPv6 addresses if True.
+    """
+    stats = psutil.net_if_stats()
+
+    ip_list: list[str] = []
+    for ifname, addrs in psutil.net_if_addrs().items():
+        st = stats.get(ifname)
+        if not st or not st.isup:
+            continue  # interface is down or unknown
+
+        for a in addrs:
+            fam = a.family
+            if fam not in (socket.AF_INET, socket.AF_INET6):
+                continue
+
+            # Family filters
+            if fam == socket.AF_INET and not ipv4:
+                continue
+            if fam == socket.AF_INET6 and not ipv6:
+                continue
+
+            ip = a.address
+
+            # Skip placeholders/wildcards
+            if fam == socket.AF_INET and ip == "0.0.0.0":
+                continue
+            if fam == socket.AF_INET6 and ip in ("::",):
+                continue
+
+            # Optionally skip loopback
+            if not localhost:
+                if fam == socket.AF_INET and ip.startswith("127."):
+                    continue
+                if fam == socket.AF_INET6 and (ip == "::1" or ip.startswith("::1%")):
+                    continue
+
+            # yield ifname, fam, ip
+            ip_list.append(ip)
+
+    return ip_list
+
+
 def get_microsoft_loopback_device_network_configuration(
         wmi_instance: CDispatch = None,
         timeout: int = 1,

atomicshop/wrappers/socketw/dns_server.py

@@ -13,6 +13,7 @@ from ..loggingw import loggingw
 from ..psutilw import psutil_networks
 from ...basics import booleans, tracebacks
 from ...file_io import csvs
+from ... import networks
 
 # noinspection PyPackageRequirements
 import dnslib
@@ -302,6 +303,15 @@ class DnsServer:
             time.sleep(1)
             raise DnsConfigurationValuesError(e)
 
+        host_ips: list[str] = networks.get_host_ips(ipv6=False)
+        if self.listening_interface not in host_ips:
+            message = (f"Listening interface [{self.listening_interface}] is not assigned to any of the host "
+                       f"network interfaces. Current host IPv4 addresses: {host_ips}")
+            print_api(f'DnsConfigurationValuesError: {str(message)}', error_type=True, color="red", logger=self.logger)
+            # Wait for the message to be printed and saved to file.
+            time.sleep(1)
+            raise DnsConfigurationValuesError(message)
+
         ips_ports: list[str] = [f'{self.listening_interface}:{self.listening_port}']
         port_in_use = psutil_networks.get_processes_using_port_list(ips_ports)
         if port_in_use:

atomicshop/wrappers/socketw/socket_wrapper.py

@@ -370,7 +370,12 @@ class SocketWrapper:
 
             # If someone removed the CA certificate file manually, and now it was created, we also need to
             # clear the cached certificates.
-            shutil.rmtree(self.sni_server_certificates_cache_directory)
+            try:
+                shutil.rmtree(self.sni_server_certificates_cache_directory)
+            # If the directory doesn't exist it will throw an exception, which is OK.
+            except FileNotFoundError:
+                pass
+
             os.makedirs(self.sni_server_certificates_cache_directory, exist_ok=True)
             print_api("Removed cached server certificates.", logger=self.logger)
 

{atomicshop-3.3.7.dist-info → atomicshop-3.3.9.dist-info}/METADATA

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

{atomicshop-3.3.7.dist-info → atomicshop-3.3.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=MIwc3VJrXSpw6OLpv-QypjAnF5OWyBPBQvLHLdy2SVA,122
+atomicshop/__init__.py,sha256=E3S7H0jQWGqjILSoky2n8u09FS0ToNPC9B6NTFZ6i2s,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
@@ -24,7 +24,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=fwOMMHwn-XMq0WgYCFbo34NXW-Xu-4nDfWl3T5EsE9U,24807
+atomicshop/networks.py,sha256=IfhEgI0rJHPgkbIDEC6lsPVZm-fyylpacKSo5HrIEic,26403
 atomicshop/on_exit.py,sha256=9XlOnzoAG8zlI8wBF4AB8hyrC6Q1b84gkhqpAhhdN9g,6977
 atomicshop/pbtkmultifile_argparse.py,sha256=aEk8nhvoQVu-xyfZosK3ma17CwIgOjzO1erXXdjwtS4,4574
 atomicshop/print_api.py,sha256=SJNQIMqSLlYaPtjHnALySAI-jQYuYHOCGgfP7oe96fU,10957
@@ -311,7 +311,7 @@ atomicshop/wrappers/socketw/accepter.py,sha256=4I9ORugRDvwaqSzm_gWSjZnRwQGY8hDTl
 atomicshop/wrappers/socketw/base.py,sha256=EcosGkD8VzgBY3GeIHDSG29ThQfXwg3-GQPmBTAqTdw,3048
 atomicshop/wrappers/socketw/certificator.py,sha256=mtWPJ_ew3OSwt0-1W4jaoco1VIY4NRCrMv3mDUxb_Cc,12418
 atomicshop/wrappers/socketw/creator.py,sha256=LGI4gcgJ47thx6f96rjwjPz3CsTAIv6VxWFY4EyUF2E,13667
-atomicshop/wrappers/socketw/dns_server.py,sha256=iQym3LmrfUfBwrFmuSHdp_DgeUOueNpEKa_TPYQv0r4,55147
+atomicshop/wrappers/socketw/dns_server.py,sha256=SYpjiBWHfYnTpHoekKe5UvTucrdtuQcjyW-CIT2AglU,55767
 atomicshop/wrappers/socketw/exception_wrapper.py,sha256=qW_1CKyPgGlsIt7_jusKkMV4A4hih4bX324u0PLnoO8,7382
 atomicshop/wrappers/socketw/get_process.py,sha256=aJC-_qFUv3NgWCSUzDI72E4z8_-VTZE9NVZ0CwUoNlM,5698
 atomicshop/wrappers/socketw/receiver.py,sha256=9B3MvcDqr4C3x2fsnjG5SQognd1wRqsBgikxZa0wXG8,8243
@@ -319,15 +319,15 @@ atomicshop/wrappers/socketw/sender.py,sha256=aX_K8l_rHjd5AWb8bi5mt8-YTkMYVRDB6Dn
 atomicshop/wrappers/socketw/sni.py,sha256=YlKavbExcPFfHFLYAJ3i3W6QorY7o4mbQp39g-DnDKA,17911
 atomicshop/wrappers/socketw/socket_client.py,sha256=McBd3DeCy787oDGCEMUEP2awWy3vdkPqr9w-aFh2fBM,22502
 atomicshop/wrappers/socketw/socket_server_tester.py,sha256=Qobmh4XV8ZxLUaw-eW4ESKAbeSLecCKn2OWFzMhadk0,6420
-atomicshop/wrappers/socketw/socket_wrapper.py,sha256=u_v0pjMMrgsdtI0iPPddiLe2wXnBoqTgNM9Y3zjGD4U,41013
+atomicshop/wrappers/socketw/socket_wrapper.py,sha256=VZe27EQhExaiLQ0FEW4ePJhNSwPMyPzgcl6oljMSbGg,41185
 atomicshop/wrappers/socketw/ssl_base.py,sha256=62-hPm7zla1rh3m_WvDnXqKH-sDUTdiRptD8STCkgdk,2313
 atomicshop/wrappers/socketw/statistics_csv.py,sha256=_gA8bMX6Sw_UCXKi2y9wNAwlqifgExgDGfQIa9pFxQA,5543
 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.7.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.3.7.dist-info/METADATA,sha256=YtwfcbICVVpiUrkEndWY-sfDjaT6_bHn8b2JOi3N8QI,9311
-atomicshop-3.3.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-atomicshop-3.3.7.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
-atomicshop-3.3.7.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.3.7.dist-info/RECORD,,
+atomicshop-3.3.9.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.3.9.dist-info/METADATA,sha256=tAksy0C2kBQBoyBa3TkozMBkD0KlKEBZ09b2oJYY4pQ,9311
+atomicshop-3.3.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+atomicshop-3.3.9.dist-info/entry_points.txt,sha256=SJEgEP0KoFtfxuGwe5tOzKfXkjR9Dv6YYug33KNYxyY,69
+atomicshop-3.3.9.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.3.9.dist-info/RECORD,,