From 53c20a20e202a2428646e7b5a8df49224c6997d6 Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Fri, 31 May 2024 21:21:06 +0200
Subject: [PATCH] Update addition_processed.py

---
 .../rootfs/helpers/addition_processed.py      | 23 +++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/birdnet-pi/rootfs/helpers/addition_processed.py b/birdnet-pi/rootfs/helpers/addition_processed.py
index ed2786e11..5f4473538 100644
--- a/birdnet-pi/rootfs/helpers/addition_processed.py
+++ b/birdnet-pi/rootfs/helpers/addition_processed.py
@@ -1,4 +1,5 @@
 import os
+import sys
 
 def read_file(file_path):
     with open(file_path, 'r') as file:
@@ -8,7 +9,7 @@ def write_file(file_path, content):
     with open(file_path, 'w') as file:
         file.writelines(content)
 
-def process_code(lines):
+def process_code(lines, max_files):
     processed_lines = []
     for line in lines:
         # Add the import statement for shutil if it's not already there
@@ -32,14 +33,14 @@ def process_code(lines):
             processed_lines.append(f"{indent}shutil.move(file.file_name, processed_dir)\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_file_count(processed_dir)\n")
+            processed_lines.append(f"{indent}maintain_file_count(processed_dir, max_files)\n")
             processed_lines.append(f"{indent}\n")
             continue  # Skip the original line that removes 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):
             processed_lines.append('\n')
-            processed_lines.append('def maintain_file_count(directory, max_files=15):\n')
+            processed_lines.append('def maintain_file_count(directory, max_files):\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('\n')
@@ -52,13 +53,27 @@ def process_code(lines):
 # Hardcoded path to the birdnet_analysis.py file
 file_path = os.path.expanduser('~/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
 original_lines = read_file(file_path)
 
 # Process the code
-modified_lines = process_code(original_lines)
+modified_lines = process_code(original_lines, max_files)
 
 # Write the modified code back to the same file
 write_file(file_path, modified_lines)
 
+# Ensure the 'Processed' directory exists and is owned by the user
+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}")