From b402b604542155769a00f609f61ab7fb1ca74722 Mon Sep 17 00:00:00 2001
From: Brian Brondel <BBrondel@lsst.org>
Date: Wed, 8 Jan 2025 16:49:01 -0300
Subject: [PATCH] Fix docker files

---
 Dockerfile.pqserver | 15 ++++++++++++++-
 Dockerfile.pytest   |  2 +-
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.pqserver b/Dockerfile.pqserver
index ba25b98c..e85b257f 100644
--- a/Dockerfile.pqserver
+++ b/Dockerfile.pqserver
@@ -1,7 +1,20 @@
 FROM python:3.11
 RUN pip install fastapi safir astropy uvicorn gunicorn sqlalchemy psycopg2
 WORKDIR /
-COPY python/lsst/consdb/__init__.py python/lsst/consdb/pqserver.py python/lsst/consdb/utils.py /consdb_pq/
+COPY \
+    python/lsst/consdb/__init__.py \
+    python/lsst/consdb/pqserver.py \
+    python/lsst/consdb/cdb_schema.py \
+    python/lsst/consdb/config.py \
+    python/lsst/consdb/dependencies.py \
+    python/lsst/consdb/exceptions.py \
+    python/lsst/consdb/models.py \
+    /consdb_pq/
+COPY \
+    python/lsst/consdb/handlers/external.py \
+    python/lsst/consdb/handlers/internal.py \
+    /consdb_pq/handlers/
+
 # Environment variables that must be set:
 # DB_HOST DB_PASS DB_USER DB_NAME or POSTGRES_URL
 
diff --git a/Dockerfile.pytest b/Dockerfile.pytest
index 20cd79ad..c54aa954 100644
--- a/Dockerfile.pytest
+++ b/Dockerfile.pytest
@@ -7,7 +7,7 @@ RUN yum install -y postgresql-server postgresql && rmdir /usr/local/bin && ln -s
 
 USER lsst
 RUN source loadLSST.bash && mamba install -y aiokafka httpx
-RUN source loadLSST.bash && pip install kafkit aiokafka httpx pytest-asyncio testing.postgresql lsst-felis
+RUN source loadLSST.bash && pip install kafkit aiokafka httpx pytest-asyncio testing.postgresql lsst-felis safir
 
 WORKDIR /home/lsst/