Merge pull request #49 from ai-cfia/48-as-a-devops-id-like-for-the-cu…

…stom-postgres-image-to-also-seed-initial-data

Issue #48: Adds step for fertiscan initial data insertion

dockerfiles/postgres/README.md

@@ -20,6 +20,8 @@ are required:
   schema.
 - `BB_DATABASE_ID` – The ID of the specific database schema to be fetched from
   Bytebase.
+- `PROJECT_NAME` – The name of the project to which the database belongs. This
+  is used to insert initial data into the database of FeritScan.
 
 ## Usage example
 

dockerfiles/postgres/entrypoint.sh

@@ -6,6 +6,12 @@ pg_ctl -D "/var/lib/postgresql/data" -o "-c listen_addresses=''" -w start
 
 psql -U "$POSTGRES_USER" -d "$POSTGRES_DB" -f /app/schema.sql
 
+if [ "$PROJECT_NAME" = "fertiscan" ]; then
+    curl -o /app/init_data.sql https://raw.githubusercontent.com/ai-cfia/ailab-datastore/main/datastore/db/bytebase/FertiScan/init_data.sql
+
+    psql -U "$POSTGRES_USER" -d "$POSTGRES_DB" -f /app/init_data.sql
+fi
+
 pg_ctl -D "/var/lib/postgresql/data" -m fast -w stop
 
 exec docker-entrypoint.sh "$@"

dockerfiles/postgres/fetch_schema.py

@@ -26,7 +26,7 @@ def generate_token():
         print(f"Failed to generate token: {response.status_code}, {response.text}")
         return None
 
-def fetch_fertiscan_dev_schema():
+def fetch_schema():
     token = generate_token()
     if token:
         schema_url = f"{BB_URL}/v1/instances/{INSTANCE_ID}/databases/{DATABASE_ID}/schema"
@@ -46,4 +46,4 @@ def fetch_fertiscan_dev_schema():
         print("No token generated, cannot fetch schema.")
 
 if __name__ == "__main__":
-    fetch_fertiscan_dev_schema()
+    fetch_schema()