Dockerfile.api

FROM python:3.9-slim as base

ARG INSTALL_SLURM
ARG SLURM_VERSION

ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1
EXPOSE 8080

SHELL ["/bin/bash", "-c", "-l"]

RUN useradd -ms /bin/bash fastapi

USER root
RUN apt update
RUN apt install -y git
RUN pip install --upgrade pip

WORKDIR /tmp
RUN if [ "$INSTALL_SLURM" = "yes" ] ; \
    then \
        apt install -y munge && \
        wget "https://download.schedmd.com/slurm/slurm-${SLURM_VERSION}.tar.bz2" && \
        tar --bzip -x -f "slurm-${SLURM_VERSION}.tar.bz2" && \
        cd "slurm-${SLURM_VERSION}" && \
        ./configure && \
        make && \
        make install ; \
     else \
        echo 'Slurm will not be installed.' ; \
    fi

WORKDIR /home/fastapi/fastapi-celery
COPY . ./

########################## production #############################
FROM base as production

ARG WRAPPER_DEPS_INSTALL
ARG WRAPPER_DEPS_EXTRA

RUN chmod -R 0777 .
RUN chown -R fastapi:fastapi .

USER fastapi
RUN pip install osp-core ${WRAPPER_DEPS_INSTALL}
RUN pip install .[wrappers] ${WRAPPER_DEPS_EXTRA}

ENTRYPOINT ["./docker_entrypoint.sh"]

########################## development #############################
FROM base AS develop

ARG WRAPPER_DEPS_INSTALL
ARG WRAPPER_DEPS_EXTRA

RUN pip install osp-core ${WRAPPER_DEPS_INSTALL}
RUN pip install .[wrappers,tests,pre_commit] ${WRAPPER_DEPS_EXTRA}

WORKDIR /app