Minor fixes

AddisonG · Dec 3, 2021 · 74a1c74 · 74a1c74
1 parent e79549f
commit 74a1c74
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/network_scanner.py b/network_scanner.py
@@ -2,6 +2,7 @@
 
 from scapy.all import arping
 
+
 def scan(ip_range: str):
     try:
         responses, _ = arping(ip_range, verbose=False)
@@ -21,7 +22,8 @@ def scan(ip_range: str):
 
     return devices
 
-def find_ip_by_mac(ip_range: str, mac: str):
+
+def find_ip_by_mac(ip_range: str, mac: str) -> str:
     for device in scan(ip_range):
         if device["mac"].upper() == mac.upper():
             return device["ip"]

diff --git a/timtamcam.py b/timtamcam.py
@@ -138,7 +138,7 @@ def record_gif(self, duration, fps):
                 # Save a single image
                 # cv2.imwrite("/tmp/timtam-thief.jpg", frame)
 
-                if self.mask:
+                if self.mask is not None:
                     # Add overlay
                     frame = cv2.subtract(frame, self.mask)
                     frame = cv2.addWeighted(frame, 1, self.border, 1, 0)