New logic

birdnet-pi/rootfs/helpers/birdnet_to_mqtt.py

@@ -3,12 +3,10 @@
 
 import time
 import re
-import dateparser
 import datetime
 import json
 import logging
 import paho.mqtt.client as mqtt
-import subprocess
 import requests
 import sys
 import os
@@ -16,6 +14,9 @@ import os
 sys.path.append('/home/pi/BirdNET-Pi/scripts/utils')
 from helpers import get_settings
 
+# Setup logging
+log = logging.getLogger(__name__)
+
 # Used in flickrimage
 flickr_images = {}
 conf = get_settings()
@@ -25,7 +26,7 @@ settings_dict = dict(conf)
 mqtt_server = "%%mqtt_server%%"
 mqtt_user = "%%mqtt_user%%"
 mqtt_pass = "%%mqtt_pass%%"
-mqtt_port = %%mqtt_port%%
+mqtt_port = "%%mqtt_port%%"
 
 mqtt_topic_confident_birds = 'birdnet'
 bird_lookup_url_base = 'http://en.wikipedia.org/wiki/'
@@ -33,9 +34,9 @@ bird_lookup_url_base = 'http://en.wikipedia.org/wiki/'
 def on_connect(client, userdata, flags, rc, properties=None):
     """ Callback for when the client receives a CONNACK response from the server. """
     if rc == 0:
-        logging.info("Connected to MQTT Broker!")
+        log.info("Connected to MQTT Broker!")
     else:
-        logging.error(f"Failed to connect, return code {rc}\n")
+        log.error(f"Failed to connect, return code {rc}\n")
 
 def get_bird_code(scientific_name):
     with open('/home/pi/BirdNET-Pi/scripts/ebird.php', 'r') as file:
@@ -48,7 +49,44 @@ def get_bird_code(scientific_name):
 
     return bird_dict.get(scientific_name)
 
-ts_noise = 0.0
+def automatic_mqtt_publish(file, detections):
+    for detection in detections:
+        bird = {}
+        bird['Date'] = file.date
+        bird['Time'] = file.time
+        bird['ScientificName'] = detection.scientific_name.replace('_', ' ')
+        bird['CommonName'] = detection.common_name
+        bird['Confidence'] = detection.confidence
+        bird['SpeciesCode'] = get_bird_code(detection.scientific_name)
+        bird['ClipName'] = os.path.basename(detection.file_name_ext)
+        bird['url'] = bird_lookup_url_base + detection.scientific_name
+
+        image_url = ""
+        common_name = detection.common_name
+        if len(settings_dict.get('FLICKR_API_KEY', '')) > 0:
+            if common_name not in flickr_images:
+                try:
+                    headers = {'User-Agent': 'Python_Flickr/1.0'}
+                    url = ('https://www.flickr.com/services/rest/?method=flickr.photos.search&api_key=' +
+                           str(settings_dict.get('FLICKR_API_KEY')) +
+                           '&text=' + str(common_name) + ' bird&sort=relevance&per_page=5&media=photos&format=json&license=2%2C3%2C4%2C5%2C6%2C9&nojsoncallback=1')
+                    resp = requests.get(url=url, headers=headers, timeout=10)
+
+                    resp.encoding = "utf-8"
+                    data = resp.json()["photos"]["photo"][0]
+
+                    image_url = 'https://farm'+str(data["farm"])+'.static.flickr.com/'+str(data["server"])+'/'+str(data["id"])+'_'+str(data["secret"])+'_n.jpg'
+                    flickr_images[common_name] = image_url
+                except Exception as e:
+                    log.error("FLICKR API ERROR: " + str(e))
+                    image_url = ""
+            else:
+                image_url = flickr_images[common_name]
+            bird['Flickrimage'] = image_url
+
+        json_bird = json.dumps(bird)
+        mqttc.publish(mqtt_topic_confident_birds, json_bird, 1)
+        log.info("Posted to MQTT: ok")
 
 mqttc = mqtt.Client('birdnet_mqtt')
 mqttc.username_pw_set(mqtt_user, mqtt_pass)
@@ -58,47 +96,11 @@ try:
     mqttc.connect(mqtt_server, mqtt_port)
     mqttc.loop_start()
 
-    def automatic_mqtt_publish(file, detections):
-        for detection in detections:
-            bird = {}
-            bird['Date'] = file.date
-            bird['Time'] = file.time
-            bird['ScientificName'] = detection.scientific_name.replace('_', ' ')
-            bird['CommonName'] = detection.common_name
-            bird['Confidence'] = detection.confidence
-            bird['SpeciesCode'] = get_bird_code(detection.scientific_name)
-            bird['ClipName'] = os.path.basename(detection.file_name_ext)
-            bird['url'] = bird_lookup_url_base + detection.scientific_name
-
-            image_url = ""
-            common_name = detection.common_name
-            if len(settings_dict.get('FLICKR_API_KEY', '')) > 0:
-                if common_name not in flickr_images:
-                    try:
-                        headers = {'User-Agent': 'Python_Flickr/1.0'}
-                        url = ('https://www.flickr.com/services/rest/?method=flickr.photos.search&api_key=' +
-                               str(settings_dict.get('FLICKR_API_KEY')) +
-                               '&text=' + str(common_name) + ' bird&sort=relevance&per_page=5&media=photos&format=json&license=2%2C3%2C4%2C5%2C6%2C9&nojsoncallback=1')
-                        resp = requests.get(url=url, headers=headers, timeout=10)
-
-                        resp.encoding = "utf-8"
-                        data = resp.json()["photos"]["photo"][0]
-
-                        image_url = 'https://farm'+str(data["farm"])+'.static.flickr.com/'+str(data["server"])+'/'+str(data["id"])+'_'+str(data["secret"])+'_n.jpg'
-                        flickr_images[common_name] = image_url
-                    except Exception as e:
-                        logging.error("FLICKR API ERROR: " + str(e))
-                        image_url = ""
-                else:
-                    image_url = flickr_images[common_name]
-                bird['Flickrimage'] = image_url
-
-            json_bird = json.dumps(bird)
-            mqttc.publish(mqtt_topic_confident_birds, json_bird, 1)
-            logging.info('Posted to MQTT: ok')
+    # Assuming `file` and `detections` are provided from somewhere
+    # automatic_mqtt_publish(file, detections)
 
 except Exception as e:
-    logging.error(f"An error occurred: {e}")
+    log.error("Cannot post mqtt: %s", e)
 
 finally:
     mqttc.loop_stop()