avm/hassio-addons
1
0
Fork 0
mirror of https://github.com/alexbelgium/hassio-addons.git synced 2026-06-08 08:35:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update addition_processed.py

Browse Source
This commit is contained in:
Alexandre
2024-05-31 21:37:06 +02:00
committed by GitHub
parent 89c5e99987
commit 7718fc532e
1 changed files with 13 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
birdnet-pi/rootfs/helpers/addition_processed.py
View File

@@ -1,5 +1,5 @@
import os # Code to modify the birdnet_analysis.py and restore the Processed folder
import sys # The last 15 wav files are stored in a dynamic manner in the Processed folder
def read_file(file_path): def read_file(file_path):
with open(file_path, 'r') as file: with open(file_path, 'r') as file:
@@ -9,7 +9,7 @@ def write_file(file_path, content):
with open(file_path, 'w') as file: with open(file_path, 'w') as file:
file.writelines(content) file.writelines(content)
def process_code(lines, max_files): def process_code(lines):
processed_lines = [] processed_lines = []
for line in lines: for line in lines:
# Add the import statement for shutil if it's not already there # Add the import statement for shutil if it's not already there
@@ -33,14 +33,14 @@ def process_code(lines, max_files):
processed_lines.append(f"{indent}shutil.move(file.file_name, processed_dir)\n") processed_lines.append(f"{indent}shutil.move(file.file_name, processed_dir)\n")
processed_lines.append(f"{indent}\n") processed_lines.append(f"{indent}\n")
processed_lines.append(f"{indent}# Maintain the file count in the 'Processed' folder\n") processed_lines.append(f"{indent}# Maintain the file count in the 'Processed' folder\n")
processed_lines.append(f"{indent}maintain_file_count(processed_dir, max_files)\n") processed_lines.append(f"{indent}maintain_file_count(processed_dir)\n")
processed_lines.append(f"{indent}\n") processed_lines.append(f"{indent}\n")
continue # Skip the original line that removes the file continue # Skip the original line that removes the file
# Add the new maintain_file_count function at the end of the file # Add the new maintain_file_count function at the end of the file
if 'if __name__' in line and not any('def maintain_file_count' in l for l in processed_lines): if 'if __name__' in line and not any('def maintain_file_count' in l for l in processed_lines):
processed_lines.append('\n') processed_lines.append('\n')
processed_lines.append('def maintain_file_count(directory, max_files):\n') processed_lines.append('def maintain_file_count(directory, max_files=15):\n')
processed_lines.append(' files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith(\'.wav\')]\n') processed_lines.append(' files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith(\'.wav\')]\n')
processed_lines.append(' files.sort(key=lambda x: os.path.getmtime(x))\n') processed_lines.append(' files.sort(key=lambda x: os.path.getmtime(x))\n')
processed_lines.append('\n') processed_lines.append('\n')
@@ -50,30 +50,17 @@ def process_code(lines, max_files):
return processed_lines return processed_lines
# Hardcoded path to the birdnet_analysis.py file # Paths to the original and new code files
file_path = os.path.expanduser('~/BirdNET-Pi/scripts/birdnet_analysis.py') original_code_path = '/home/pi/BirdNET-Pi/scripts/birdnet_analysis.py'
new_code_path = '/home/pi/BirdNET-Pi/scripts/birdnet_analysis.py'
# Get the maximum number of processed files from the environment variable
max_files = int(os.environ.get('Processed_Files', 15))
# If the environment variable is set to 0, exit the script
if max_files == 0:
print("The 'Processed_Files' environment variable is set to 0. Exiting without making changes.")
sys.exit(0)
# Read the original code # Read the original code
original_lines = read_file(file_path) original_lines = read_file(original_code_path)
# Process the code # Process the code
modified_lines = process_code(original_lines, max_files) modified_lines = process_code(original_lines)
# Write the modified code back to the same file # Write the modified code to a new file
write_file(file_path, modified_lines) write_file(new_code_path, modified_lines)
# Ensure the 'Processed' directory exists and is owned by the user print(f"The code has been modified and saved to {new_code_path}")
processed_dir = os.path.join(os.path.dirname(file_path), 'Processed')
if not os.path.exists(processed_dir):
os.makedirs(processed_dir)
os.chown(processed_dir, os.getuid(), os.getgid())
print(f"The code has been modified and saved to {file_path}")