broadcasting func - get conn infos from server

boswatch/network/broadcast.py

@@ -0,0 +1,87 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+"""!
+    ____  ____  ______       __      __       __       _____
+   / __ )/ __ \/ ___/ |     / /___ _/ /______/ /_     |__  /
+  / __  / / / /\__ \| | /| / / __ `/ __/ ___/ __ \     /_ <
+ / /_/ / /_/ /___/ /| |/ |/ / /_/ / /_/ /__/ / / /   ___/ /
+/_____/\____//____/ |__/|__/\__,_/\__/\___/_/ /_/   /____/
+                German BOS Information Script
+                     by Bastian Schroll
+
+@file:        broadcast.py
+@date:        21.09.2018
+@author:      Bastian Schroll
+@description: UDP broadcast server and client class
+"""
+import logging
+import socket
+
+logging.debug("- %s loaded", __name__)
+
+
+class BroadcastClient:
+
+    def __init__(self, port=5000):
+        self._broadcastPort = port
+
+        self._serverIP = ""
+        self._serverPort = 0
+
+        self._socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        self._socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        self._socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+        self._socket.settimeout(3)
+
+    def sendBroadcast(self):
+        while True:
+            try:
+
+                logging.debug("send magic <BW3-Request>")
+                self._socket.sendto("<BW3-Request>".encode(), ('255.255.255.255', self._broadcastPort))
+                payload, address = self._socket.recvfrom(1024)
+                payload = str(payload, "UTF-8")
+
+                if payload.startswith("<BW3-Result>"):
+                    logging.debug("received magic <BW3-Result>")
+                    self._serverIP = address[0]
+                    self._serverPort = int(payload.split(";")[1])
+                    logging.info("got connection info: %s:%d", self._serverIP, self._serverPort)
+                    return True
+            except socket.timeout:
+                logging.warning("retry sending magic")
+                continue
+            except:
+                logging.exception("error on getting connection info")
+                return False
+
+    @property
+    def serverIP(self):
+        return self._serverIP
+
+    @property
+    def serverPort(self):
+        return self._serverPort
+
+
+class BroadcastServer:
+    """!General class comment"""
+
+    def __init__(self, port=5000):
+        """!init comment"""
+        self._socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        self._socket.bind(('', port))
+
+    def listen(self):
+        try:
+            logging.debug("start listening for magic")
+            while True:
+                payload, address = self._socket.recvfrom(1024)
+                payload = str(payload, "UTF-8")
+                if payload == "<BW3-Request>":
+                    logging.debug("received magic <BW3-Request> from: %s", address[0])
+                    logging.debug("send connection info in magic <BW3-Result>")
+                    self._socket.sendto("<BW3-Result>;8080".encode(), address)  # todo give the TCPServer port
+                    return True
+        except:
+            logging.exception("error while listening for clients")