satisfy flake8 linter
This commit is contained in:
+14
-6
@@ -55,10 +55,11 @@ with open(userDir + '/BirdNET-Pi/scripts/thisrun.txt', 'r') as f:
|
|||||||
try:
|
try:
|
||||||
model = str(str(str([i for i in this_run if i.startswith('MODEL')]).split('=')[1]).split('\\')[0])
|
model = str(str(str([i for i in this_run if i.startswith('MODEL')]).split('=')[1]).split('\\')[0])
|
||||||
sf_thresh = str(str(str([i for i in this_run if i.startswith('SF_THRESH')]).split('=')[1]).split('\\')[0])
|
sf_thresh = str(str(str([i for i in this_run if i.startswith('SF_THRESH')]).split('=')[1]).split('\\')[0])
|
||||||
except Exception as e:
|
except Exception:
|
||||||
model = "BirdNET_6K_GLOBAL_MODEL"
|
model = "BirdNET_6K_GLOBAL_MODEL"
|
||||||
sf_thresh = 0.03
|
sf_thresh = 0.03
|
||||||
|
|
||||||
|
|
||||||
def loadModel():
|
def loadModel():
|
||||||
|
|
||||||
global INPUT_LAYER_INDEX
|
global INPUT_LAYER_INDEX
|
||||||
@@ -95,6 +96,7 @@ def loadModel():
|
|||||||
|
|
||||||
return myinterpreter
|
return myinterpreter
|
||||||
|
|
||||||
|
|
||||||
def loadMetaModel():
|
def loadMetaModel():
|
||||||
|
|
||||||
global M_INTERPRETER
|
global M_INTERPRETER
|
||||||
@@ -115,15 +117,16 @@ def loadMetaModel():
|
|||||||
|
|
||||||
print("loaded META model")
|
print("loaded META model")
|
||||||
|
|
||||||
|
|
||||||
def predictFilter(lat, lon, week):
|
def predictFilter(lat, lon, week):
|
||||||
|
|
||||||
global M_INTERPRETER
|
global M_INTERPRETER
|
||||||
|
|
||||||
# Does interpreter exist?
|
# Does interpreter exist?
|
||||||
try:
|
try:
|
||||||
if M_INTERPRETER == None:
|
if M_INTERPRETER is None:
|
||||||
loadMetaModel()
|
loadMetaModel()
|
||||||
except Exception as e:
|
except Exception:
|
||||||
loadMetaModel()
|
loadMetaModel()
|
||||||
|
|
||||||
# Prepare mdata as sample
|
# Prepare mdata as sample
|
||||||
@@ -135,6 +138,7 @@ def predictFilter(lat, lon, week):
|
|||||||
|
|
||||||
return M_INTERPRETER.get_tensor(M_OUTPUT_LAYER_INDEX)[0]
|
return M_INTERPRETER.get_tensor(M_OUTPUT_LAYER_INDEX)[0]
|
||||||
|
|
||||||
|
|
||||||
def explore(lat, lon, week):
|
def explore(lat, lon, week):
|
||||||
|
|
||||||
# Make filter prediction
|
# Make filter prediction
|
||||||
@@ -151,6 +155,7 @@ def explore(lat, lon, week):
|
|||||||
|
|
||||||
return l_filter
|
return l_filter
|
||||||
|
|
||||||
|
|
||||||
def predictSpeciesList(lat, lon, week):
|
def predictSpeciesList(lat, lon, week):
|
||||||
|
|
||||||
l_filter = explore(lat, lon, week)
|
l_filter = explore(lat, lon, week)
|
||||||
@@ -160,6 +165,7 @@ def predictSpeciesList(lat, lon, week):
|
|||||||
if (len(INCLUDE_LIST) == 0):
|
if (len(INCLUDE_LIST) == 0):
|
||||||
PREDICTED_SPECIES_LIST.append(s[1])
|
PREDICTED_SPECIES_LIST.append(s[1])
|
||||||
|
|
||||||
|
|
||||||
def loadCustomSpeciesList(path):
|
def loadCustomSpeciesList(path):
|
||||||
|
|
||||||
slist = []
|
slist = []
|
||||||
@@ -274,7 +280,6 @@ def analyzeAudioData(chunks, lat, lon, week, sensitivity, overlap,):
|
|||||||
if len(PREDICTED_SPECIES_LIST) == 0 or len(INCLUDE_LIST) != 0:
|
if len(PREDICTED_SPECIES_LIST) == 0 or len(INCLUDE_LIST) != 0:
|
||||||
predictSpeciesList(lat, lon, week)
|
predictSpeciesList(lat, lon, week)
|
||||||
|
|
||||||
|
|
||||||
# Convert and prepare metadata
|
# Convert and prepare metadata
|
||||||
mdata = convertMetadata(np.array([lat, lon, week]))
|
mdata = convertMetadata(np.array([lat, lon, week]))
|
||||||
mdata = np.expand_dims(mdata, 0)
|
mdata = np.expand_dims(mdata, 0)
|
||||||
@@ -320,7 +325,9 @@ def writeResultsToFile(detections, min_conf, path):
|
|||||||
rfile.write('Start (s);End (s);Scientific name;Common name;Confidence\n')
|
rfile.write('Start (s);End (s);Scientific name;Common name;Confidence\n')
|
||||||
for d in detections:
|
for d in detections:
|
||||||
for entry in detections[d]:
|
for entry in detections[d]:
|
||||||
if entry[1] >= min_conf and ((entry[0] in INCLUDE_LIST or len(INCLUDE_LIST) == 0) and (entry[0] not in EXCLUDE_LIST or len(EXCLUDE_LIST) == 0) and (entry[0] in PREDICTED_SPECIES_LIST or len(PREDICTED_SPECIES_LIST) == 0) ):
|
if entry[1] >= min_conf and ((entry[0] in INCLUDE_LIST or len(INCLUDE_LIST) == 0)
|
||||||
|
and (entry[0] not in EXCLUDE_LIST or len(EXCLUDE_LIST) == 0)
|
||||||
|
and (entry[0] in PREDICTED_SPECIES_LIST or len(PREDICTED_SPECIES_LIST) == 0)):
|
||||||
rfile.write(d + ';' + entry[0].replace('_', ';').split("/")[0] + ';' + str(entry[1]) + '\n')
|
rfile.write(d + ';' + entry[0].replace('_', ';').split("/")[0] + ';' + str(entry[1]) + '\n')
|
||||||
rcnt += 1
|
rcnt += 1
|
||||||
print('DONE! WROTE', rcnt, 'RESULTS.')
|
print('DONE! WROTE', rcnt, 'RESULTS.')
|
||||||
@@ -442,7 +449,8 @@ def handle_client(conn, addr):
|
|||||||
species_apprised_this_run = []
|
species_apprised_this_run = []
|
||||||
for entry in detections[d]:
|
for entry in detections[d]:
|
||||||
if entry[1] >= min_conf and ((entry[0] in INCLUDE_LIST or len(INCLUDE_LIST) == 0)
|
if entry[1] >= min_conf and ((entry[0] in INCLUDE_LIST or len(INCLUDE_LIST) == 0)
|
||||||
and (entry[0] not in EXCLUDE_LIST or len(EXCLUDE_LIST) == 0) and (entry[0] in PREDICTED_SPECIES_LIST or len(PREDICTED_SPECIES_LIST) == 0) ):
|
and (entry[0] not in EXCLUDE_LIST or len(EXCLUDE_LIST) == 0)
|
||||||
|
and (entry[0] in PREDICTED_SPECIES_LIST or len(PREDICTED_SPECIES_LIST) == 0)):
|
||||||
# Write to text file.
|
# Write to text file.
|
||||||
rfile.write(str(current_date) + ';' + str(current_time) + ';' + entry[0].replace('_', ';').split("/")[0] + ';'
|
rfile.write(str(current_date) + ';' + str(current_time) + ';' + entry[0].replace('_', ';').split("/")[0] + ';'
|
||||||
+ str(entry[1]) + ";" + str(args.lat) + ';' + str(args.lon) + ';' + str(min_conf) + ';' + str(week) + ';'
|
+ str(entry[1]) + ";" + str(args.lat) + ';' + str(args.lon) + ';' + str(min_conf) + ';' + str(week) + ';'
|
||||||
|
|||||||
@@ -24,6 +24,7 @@ config = apprise.AppriseConfig()
|
|||||||
config.add(APPRISE_CONFIG)
|
config.add(APPRISE_CONFIG)
|
||||||
apobj.add(config)
|
apobj.add(config)
|
||||||
|
|
||||||
|
|
||||||
def notify(body, title, attached=""):
|
def notify(body, title, attached=""):
|
||||||
if attached != "":
|
if attached != "":
|
||||||
apobj.notify(
|
apobj.notify(
|
||||||
@@ -38,7 +39,9 @@ def notify(body, title, attached=""):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def sendAppriseNotifications(species, confidence, confidencepct, path, date, time, week, latitude, longitude, cutoff, sens, overlap, settings_dict, db_path=DB_PATH):
|
def sendAppriseNotifications(species, confidence, confidencepct, path,
|
||||||
|
date, time, week, latitude, longitude, cutoff,
|
||||||
|
sens, overlap, settings_dict, db_path=DB_PATH):
|
||||||
# print(sendAppriseNotifications)
|
# print(sendAppriseNotifications)
|
||||||
# print(settings_dict)
|
# print(settings_dict)
|
||||||
if os.path.exists(APPRISE_CONFIG) and os.path.getsize(APPRISE_CONFIG) > 0:
|
if os.path.exists(APPRISE_CONFIG) and os.path.getsize(APPRISE_CONFIG) > 0:
|
||||||
@@ -69,10 +72,11 @@ def sendAppriseNotifications(species, confidence, confidencepct, path, date, tim
|
|||||||
image_url = ""
|
image_url = ""
|
||||||
|
|
||||||
if len(settings_dict.get('FLICKR_API_KEY')) > 0 and "$flickrimage" in body:
|
if len(settings_dict.get('FLICKR_API_KEY')) > 0 and "$flickrimage" in body:
|
||||||
if not comName in flickr_images:
|
if comName not in flickr_images:
|
||||||
try:
|
try:
|
||||||
# TODO: Make this work with non-english comnames. Implement the "// convert sci name to English name" logic from overview.php here
|
# TODO: Make this work with non-english comnames. Implement the "// convert sci name to English name" logic from overview.php here
|
||||||
url = 'https://www.flickr.com/services/rest/?method=flickr.photos.search&api_key='+str(settings_dict.get('FLICKR_API_KEY'))+'&text='+str(comName)+' bird&sort=relevance&per_page=5&media=photos&format=json&license=2%2C3%2C4%2C5%2C6%2C9&nojsoncallback=1'
|
url = 'https://www.flickr.com/services/rest/?method=flickr.photos.search&api_key='+str(settings_dict.get('FLICKR_API_KEY'))+'&text='+str(
|
||||||
|
comName)+' bird&sort=relevance&per_page=5&media=photos&format=json&license=2%2C3%2C4%2C5%2C6%2C9&nojsoncallback=1'
|
||||||
resp = requests.get(url=url)
|
resp = requests.get(url=url)
|
||||||
resp.encoding = "utf-8"
|
resp.encoding = "utf-8"
|
||||||
data = resp.json()["photos"]["photo"][0]
|
data = resp.json()["photos"]["photo"][0]
|
||||||
@@ -85,7 +89,6 @@ def sendAppriseNotifications(species, confidence, confidencepct, path, date, tim
|
|||||||
else:
|
else:
|
||||||
image_url = flickr_images[comName]
|
image_url = flickr_images[comName]
|
||||||
|
|
||||||
|
|
||||||
if settings_dict.get('APPRISE_NOTIFY_EACH_DETECTION') == "1":
|
if settings_dict.get('APPRISE_NOTIFY_EACH_DETECTION') == "1":
|
||||||
notify_body = body.replace("$sciname", sciName)\
|
notify_body = body.replace("$sciname", sciName)\
|
||||||
.replace("$comname", comName)\
|
.replace("$comname", comName)\
|
||||||
|
|||||||
@@ -113,7 +113,8 @@ def test_notifications(mocker):
|
|||||||
notify_call.call_args_list[0][0][0] == "A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence of 91 (first time today)"
|
notify_call.call_args_list[0][0][0] == "A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence of 91 (first time today)"
|
||||||
)
|
)
|
||||||
assert (
|
assert (
|
||||||
notify_call.call_args_list[1][0][0] == "A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence of 91 (only seen 1 times in last 7d)"
|
notify_call.call_args_list[1][0][0] == "A Great Crested Flycatcher (Myiarchus crinitus) was just detected with a confidence \
|
||||||
|
of 91 (only seen 1 times in last 7d)"
|
||||||
)
|
)
|
||||||
|
|
||||||
# Add each species notification.
|
# Add each species notification.
|
||||||
|
|||||||
Reference in New Issue
Block a user