added the possibility for build args

Author: P-T-I

docker-compose.yml

@@ -1,11 +1,14 @@
-version: "3.5"
+version: "3.8"
 services:
 
   cve_search:
     image: cve_search
     build:
       context: .
       dockerfile: docker/images/cve_search/dockerfile-cve_search
+      args:
+        - REPO=cve-search/cve-search
+        - BRANCH=master
     hostname: cve_search
     depends_on:
       - redis

docker/images/cve_search/dockerfile-cve_search

@@ -10,7 +10,12 @@ ENV APP /app
 RUN mkdir $APP
 WORKDIR $APP
 
-RUN git clone https://github.com/cve-search/cve-search.git ./
+ARG REPO
+ARG BRANCH
+
+RUN echo "Using repo - branch: $REPO - $BRANCH"
+
+RUN git clone -b $BRANCH --single-branch https://github.com/$REPO.git ./
 
 # Install the dependencies
 RUN pip install -r requirements.txt