feat: Admin-panel strategy and logrotate (#2)

Author: electricalgorithm

.gitignore

@@ -1,3 +1,4 @@
+.config.json
 __pycache__/
 .pytest_cache/
 .tox/

core/observers/observer/hss_observer.py

@@ -1,7 +1,7 @@
 """
 The observer for Home Security System.
 """
-import logging
+from core.utils.logger import get_logger
 from core.observers.observer.base_observer import BaseObserver
 from core.observers.subject.base_subject import BaseSubject
 from core.observers.subject.wifi_subject import WiFiSubject
@@ -11,7 +11,7 @@
 from core.strategies.notifier.base_notifier_strategy import BaseNotifierStrategy
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class HomeSecuritySystemObserver(BaseObserver):

core/observers/subject/eye_subject.py

@@ -11,13 +11,14 @@
 
 import cv2
 
+from core.utils.logger import get_logger
 from core.utils.datatypes import EyeStates, EyeStrategyResult
 from core.utils.fileio_adaptor import upload_to_fileio
 from core.observers.subject.base_subject import BaseSubject
 from core.strategies.eye.base_eye_strategy import BaseEyeStrategy
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class EyeSubject(BaseSubject):
@@ -26,7 +27,7 @@ class EyeSubject(BaseSubject):
     Concretes a subject for Eye/Camera features.
     """
     DEFAULT_IMAGE_LOCATIONS: str = "~/.home-security-system/images"
-    DEFAULT_SLEEP_INTERVAL = 10
+    DEFAULT_SLEEP_INTERVAL = 30
     SLEEP_INTERVAL_DETECTED = 5
 
     def __init__(self, image_path: str = DEFAULT_IMAGE_LOCATIONS):

core/observers/subject/wifi_subject.py

@@ -2,7 +2,7 @@
 This class inherits from IBaseSubject.
 Concretes a subject WiFi features.
 """
-import logging
+from core.utils.logger import get_logger
 from threading import Thread, Lock
 from time import sleep
 from typing import Optional
@@ -12,7 +12,7 @@
 from core.strategies.wifi.base_wifi_strategy import BaseWiFiStrategy
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class WiFiSubject(BaseSubject):

core/strategies/eye/picamera_strategy.py

@@ -1,17 +1,16 @@
 """
 The Camera strategy for eye strategies.
 """
-import logging
+from core.utils.logger import get_logger
 
-import cv2
 import numpy
 from picamera2 import Picamera2
 
 from core.strategies.detectors.base_detector_strategy import BaseDetectorStrategy
 from core.strategies.eye.base_eye_strategy import BaseEyeStrategy
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class PiCameraStrategy(BaseEyeStrategy):

core/strategies/eye/usbcamera_strategy.py

@@ -1,7 +1,7 @@
 """
 The Camera strategy for eye strategies.
 """
-import logging
+from core.utils.logger import get_logger
 
 import cv2
 import numpy
@@ -10,7 +10,7 @@
 from core.strategies.eye.base_eye_strategy import BaseEyeStrategy
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class UsbCameraStrategy(BaseEyeStrategy):

core/strategies/notifier/whatsapp_strategy.py

@@ -1,15 +1,15 @@
 """
 This module contains the Whatsapp notifier strategy.
 """
-import logging
+from core.utils.logger import get_logger
 
 import requests
 
 from core.utils.datatypes import WhatsappReciever
 from core.strategies.notifier.base_notifier_strategy import BaseNotifierStrategy
 
 # Get the logger instance.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class WhatsappStrategy(BaseNotifierStrategy):

core/strategies/wifi/admin_panel_strategy.py

@@ -0,0 +1,66 @@
+"""
+The strategy which searches for MAC addresses using Admin Panel.
+"""
+from core.utils.logger import get_logger
+from bs4 import BeautifulSoup
+import requests
+from typing import Any
+from core.strategies.wifi.base_wifi_strategy import BaseWiFiStrategy
+from core.utils.datatypes import WiFiStrategyResult, ConnectedDeviceResult
+
+# Add logging support.
+logger = get_logger(__name__)
+
+
+class AdminPanelStrategy(BaseWiFiStrategy):
+    """
+    The strategy which searches for MAC addresses using Admin Panel.
+    """
+    
+    def __init__(self, login_data: dict[str, Any]) -> None:
+        """Constructor for AdminPanelStrategy."""
+        super().__init__()
+        self._login_data: dict[str, Any] = login_data
+
+    def check_protectors(self) -> WiFiStrategyResult:
+        """This method checks if there are any protectors around."""
+        for protector in self.protectors:
+            # Check if the protector is connected to the network.
+            if protector.address in [device.address for device in self._get_all_connected()]:
+                logger.debug("Protector found: " + protector.name)
+                return WiFiStrategyResult(protector, True)
+        logger.debug("No protectors found.")
+        return WiFiStrategyResult(None, False)
+    
+    # Internal methods
+    def _get_all_connected(self) -> list[ConnectedDeviceResult]:
+        """This method returns a list of addresses of the clients connected to the network."""
+        # Create a session to store cookies.
+        session = requests.Session()
+        session.get("http://192.168.1.95/login_security.html")
+        session.post(
+            "http://192.168.1.95/Forms/login_security_1",
+            headers={
+                "Content-Type": "application/x-www-form-urlencoded",
+                "Referer": "http://192.168.1.95/login_security.html",
+            },
+            data=self._login_data
+        )
+        
+        # Get the response for the page with MAC address list.
+        response = session.get("http://192.168.1.95/status/status_deviceinfo.htm")
+
+        # Parse the response.
+        soup = BeautifulSoup(response.text, "html.parser")
+        tabdata = soup.find_all("td", class_="tabdata")
+
+        # Get the MAC addresses.
+        mac_addrs: list[str] = [
+            element.text 
+            for element in tabdata
+            if len(element.text) == 17
+        ]
+        session.close()
+
+        logger.debug("Connected devices: " + str(mac_addrs))
+        return [ConnectedDeviceResult(mac_addr.upper()) for mac_addr in mac_addrs]

core/strategies/wifi/ipaddress_strategy.py

@@ -1,13 +1,13 @@
 """
 The strategy which searches for IP addresses.
 """
-import logging
+from core.utils.logger import get_logger
 from core.utils.program_launcher import run_program, PingCommands
 from core.strategies.wifi.base_wifi_strategy import BaseWiFiStrategy
 from core.utils.datatypes import WiFiStrategyResult
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class IpAddressStrategy(BaseWiFiStrategy):

core/strategies/wifi/macaddress_strategy.py

@@ -1,14 +1,14 @@
 """
 The strategy which searches for MAC addresses.
 """
-import logging
+from core.utils.logger import get_logger
 
 from core.utils.program_launcher import run_program, ArpScanCommands
 from core.strategies.wifi.base_wifi_strategy import BaseWiFiStrategy
 from core.utils.datatypes import WiFiStrategyResult, ConnectedDeviceResult
 
 # Add logging support.
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class MacAddressStrategy(BaseWiFiStrategy):