make init better

Author: antonpirker

test-django-admin-query-perf/init.sh

@@ -10,13 +10,23 @@ fi
 
 # Start PostgreSQL database in Docker
 echo "Starting PostgreSQL database in Docker..."
-docker run -d \
-  --name postgres-demo \
-  -e POSTGRES_DB=demo \
-  -e POSTGRES_USER=admin \
-  -e POSTGRES_PASSWORD=admin \
-  -p 5439:5432 \
-  postgres:15
+
+# Check if container exists and is running
+if docker ps --filter "name=postgres-demo" --filter "status=running" --format "{{.Names}}" | grep -q "postgres-demo"; then
+    echo "PostgreSQL container is already running"
+elif docker ps -a --filter "name=postgres-demo" --format "{{.Names}}" | grep -q "postgres-demo"; then
+    echo "PostgreSQL container exists but is stopped, starting it..."
+    docker start postgres-demo
+else
+    echo "Creating new PostgreSQL container..."
+    docker run -d \
+      --name postgres-demo \
+      -e POSTGRES_DB=demo \
+      -e POSTGRES_USER=admin \
+      -e POSTGRES_PASSWORD=admin \
+      -p 5439:5432 \
+      postgres:15
+fi
 
 # Wait for PostgreSQL to be ready
 echo "Waiting for PostgreSQL to be ready..."
@@ -33,7 +43,14 @@ export DJANGO_SUPERUSER_USERNAME=admin
 export [email protected]
 export DJANGO_SUPERUSER_PASSWORD=admin
 
-uv run ./manage.py createsuperuser --noinput
+# Check if admin user already exists
+echo "Checking if admin user exists..."
+if uv run ./manage.py shell -c "from django.contrib.auth.models import User; exit(0 if User.objects.filter(username='admin').exists() else 1)"; then
+    echo "Admin user already exists, skipping creation"
+else
+    echo "Creating admin user..."
+    uv run ./manage.py createsuperuser --noinput
+fi
 echo "Username: admin"
 echo "Password: admin"