added a build arg to container files

Author: btr1975

cookiecutter.json

@@ -14,7 +14,7 @@
     "include_webpages": ["n", "y"],
     "container_runtime": ["podman", "docker"],
     "__template_repo": "https://github.com/btr1975/cookiecutter-python-fastapi-openapi",
-    "__template_version": "1.0.6",
+    "__template_version": "1.0.7",
     "_new_lines":"\n",
     "_copy_without_render": [
         "{{cookiecutter.__app_name}}/templates",

{{cookiecutter.git_repo_name}}/Makefile

@@ -24,12 +24,12 @@ build:
 
 {% if cookiecutter.container_runtime == "podman" %}
 build-container:
-	@cd containers && podman build --ssh=default -t {{ cookiecutter.git_repo_name }}:latest -f Containerfile
+	@cd containers && podman build --ssh=default –build-arg=build_branch=main -t {{ cookiecutter.git_repo_name }}:latest -f Containerfile
 {% endif %}
 
 {% if cookiecutter.container_runtime == "docker" %}
 build-container:
-	@cd containers && docker build --ssh=default -t {{ cookiecutter.git_repo_name }}:latest -f Dockerfile
+	@cd containers && docker build --ssh=default -t –build-arg=build_branch=main {{ cookiecutter.git_repo_name }}:latest -f Dockerfile
 {% endif %}
 
 coverage:

{{cookiecutter.git_repo_name}}/containers/Containerfile

@@ -9,8 +9,12 @@ RUN ${PIP_INSTALL_COMMAND} --upgrade pip setuptools wheel \
 
 FROM base AS final
 
+ARG build_branch=main
+
+ENV BUILD_BRANCH=${build_branch}
+
 RUN --mount=type=ssh,required=true  \
-    ${PIP_INSTALL_COMMAND} git+ssh://git@github.com/{{ cookiecutter.git_username }}/{{ cookiecutter.git_repo_name }}
+    ${PIP_INSTALL_COMMAND} git+ssh://git@github.com/{{ cookiecutter.git_username }}/{{ cookiecutter.git_repo_name }}@${BUILD_BRANCH}
 
 EXPOSE 8080/tcp
 

{{cookiecutter.git_repo_name}}/containers/Dockerfile

@@ -9,8 +9,12 @@ RUN ${PIP_INSTALL_COMMAND} --upgrade pip setuptools wheel \
 
 FROM base AS final
 
+ARG build_branch=main
+
+ENV BUILD_BRANCH=${build_branch}
+
 RUN --mount=type=ssh,required=true  \
-    ${PIP_INSTALL_COMMAND} git+ssh://git@github.com/{{ cookiecutter.git_username }}/{{ cookiecutter.git_repo_name }}
+    ${PIP_INSTALL_COMMAND} git+ssh://git@github.com/{{ cookiecutter.git_username }}/{{ cookiecutter.git_repo_name }}@${BUILD_BRANCH}
 
 EXPOSE 8080/tcp