Add precommit hook

.gitignore

@@ -118,3 +118,5 @@ pyroengine/version.py
 *.onnx
 # Release
 conda-dist/
+
+docker-compose.yml.bak

Makefile

@@ -26,6 +26,7 @@ build:
 
 # Run the engine wrapper
 run:
+	bash setup-docker-compose.sh
 	docker build . -t pyronear/pyro-engine:latest
 	docker compose up -d
 

docker-compose.yml

@@ -12,6 +12,11 @@ services:
     volumes:
       - ./data:/usr/src/app/data
     restart: always
+    deploy:
+      resources:
+        limits:
+          cpus: "3"
+          memory:
     logging:
       driver: "json-file"
       options:

setup-docker-compose.sh

@@ -0,0 +1,33 @@
+#!/bin/bash
+
+# Define the percentage of host memory you want to allocate
+PERCENTAGE=70
+
+# Get the total memory of the host system in kilobytes
+TOTAL_MEM_KB=$(grep MemTotal /proc/meminfo | awk '{print $2}')
+
+# Calculate the memory limit in kilobytes
+LIMIT_MEM_KB=$((TOTAL_MEM_KB * PERCENTAGE / 100))
+
+# Convert the limit to a format Docker understands (e.g., "m" for megabytes)
+LIMIT_MEM_MB=$((LIMIT_MEM_KB / 1024))m
+
+# Define the Docker Compose file to modify
+DOCKER_COMPOSE_FILE="docker-compose.yml"
+
+# Backup the original Docker Compose file
+cp $DOCKER_COMPOSE_FILE "${DOCKER_COMPOSE_FILE}.bak"
+
+# Use awk to update the memory limits in the Docker Compose file, preserving indentation
+awk -v mem_limit="$LIMIT_MEM_MB" '
+/services:/ { in_services=1 }
+in_services && /deploy:/ { in_deploy=1 }
+in_deploy && /resources:/ { in_resources=1 }
+in_resources && /limits:/ { in_limits=1 }
+in_limits && /memory:/ {
+    $0 = gensub(/memory:.*/, "memory: " mem_limit, 1)
+}
+{ print }
+' "${DOCKER_COMPOSE_FILE}.bak" > $DOCKER_COMPOSE_FILE
+
+echo "Memory limits set to $LIMIT_MEM_MB in $DOCKER_COMPOSE_FILE"