Add config and update scripts

.pre-commit-config.yaml

@@ -0,0 +1,38 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.4.0
+    hooks:
+      - id: check-added-large-files
+      - id: check-case-conflict
+      - id: check-executables-have-shebangs
+      - id: check-merge-conflict
+      - id: check-symlinks
+      - id: check-yaml
+        exclude: (observability|testing/tavern)
+      - id: debug-statements
+        exclude: tests/
+      - id: destroyed-symlinks
+      - id: end-of-file-fixer
+        files: \.(py|sh|rst|yml|yaml)$
+      - id: mixed-line-ending
+      - id: trailing-whitespace
+        files: \.(py|sh|rst|yml|yaml)$
+  - repo: https://github.com/ambv/black
+    rev: 23.10.0
+    hooks:
+      - id: black
+  - repo: https://github.com/pycqa/flake8
+    rev: 6.0.0
+    hooks:
+      - id: flake8
+        args: [--config, src/setup.cfg]
+  - repo: https://github.com/pycqa/isort
+    rev: 5.12.0
+    hooks:
+      - id: isort
+        args: [
+          '--profile',
+          'black',
+          '--src-path',
+          'src',
+        ]

broadcast.py

@@ -1,22 +0,0 @@
-import socket
-import struct
-import time
-
-service_name = "windows.local"
-ip_address = "192.168.137.1"
-port = 5000
-message = f"{service_name}:{ip_address}:{port}"
-
-multicast_group = "224.0.0.251"
-multicast_port = 5353
-
-sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
-sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 2)
-
-try:
-    while True:
-        sock.sendto(message.encode("utf-8"), (multicast_group, multicast_port))
-        print(f"Broadcasting service info: {message}")
-        time.sleep(2)
-finally:
-    sock.close()

src/broadcast.py

@@ -0,0 +1,24 @@
+import socket
+import time
+
+from config import (
+    BROADCAST_INTERVAL,
+    MULTICAST_IP,
+    MULTICAST_PORT,
+    SERVICE_IP,
+    SERVICE_NAME,
+    SERVICE_PORT,
+)
+
+message = f"{SERVICE_NAME}:{SERVICE_IP}:{SERVICE_PORT}"
+
+sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
+sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 2)
+
+try:
+    while True:
+        sock.sendto(message.encode("utf-8"), (MULTICAST_IP, MULTICAST_PORT))
+        print(f"Broadcasting service info: {message}")
+        time.sleep(BROADCAST_INTERVAL)
+finally:
+    sock.close()

src/client.py

@@ -1,18 +1,17 @@
 import socket
 import struct
 
-multicast_address = '224.0.0.251'
+from config import MULTICAST_IP, MULTICAST_PORT
-multicast_port = 5353
 
 sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-sock.bind(('', multicast_port))
+sock.bind(("", MULTICAST_PORT))
 
-group = socket.inet_aton(multicast_address)
+group = socket.inet_aton(MULTICAST_IP)
-mreq = struct.pack('4sL', group, socket.INADDR_ANY)
+mreq = struct.pack("4sL", group, socket.INADDR_ANY)
 sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
 
-print(f"Listening for messages on {multicast_address}:{multicast_port}")
+print(f"Listening for messages on {MULTICAST_IP}:{MULTICAST_PORT}")
 
 while True:
     data, addr = sock.recvfrom(1024)

src/config.py

@@ -0,0 +1,8 @@
+MULTICAST_IP = "224.0.0.1"
+MULTICAST_PORT = 5353
+
+BROADCAST_INTERVAL = 5
+
+SERVICE_NAME = "windows.local"
+SERVICE_IP = "192.168.137.1"
+SERVICE_PORT = 5000

src/setup.cfg

@@ -0,0 +1,2 @@
+[flake8]
+max-line-length = 110