Compare commits

...
9 Commits
Author SHA1 Message Date
brb c19fc58a25 deleted unused code 2023-09-04 14:43:19 +02:00
brb bd5dc8fbc4 changed function to execute commands from runner script 2023-07-07 09:46:12 +02:00
brb f4a81b4b9b added and commented out code for passing through gpu 2023-07-07 09:45:45 +02:00
brb 477389d353 updated script 2023-07-06 17:34:08 +02:00
brb 90a44e8b6e updated running method 2023-07-06 17:32:41 +02:00
brb 36d2233567 updated exception handling when running commands from runner script 2023-07-06 17:30:40 +02:00
brb 2fc979a845 updated to run as user instead of root 2023-07-06 17:28:38 +02:00
brb 7218004f2f updated to run actual function 2023-07-06 17:22:21 +02:00
brb 75fd57391d added pip install requirements 2023-07-06 17:22:09 +02:00
4 changed files with 39 additions and 11 deletions
+13 -4
View File
@@ -24,10 +24,12 @@ RUN pip wheel --no-cache-dir --no-deps --wheel-dir /usr/src/app/wheels -r requir
# pull official base image # pull official base image
FROM dvcorg/cml:0-dvc2-base1-gpu FROM dvcorg/cml:0-dvc2-base1-gpu
# create app home # create home directory and app user
ENV APP_HOME = /home/app RUN mkdir -p /home/app && \
RUN mkdir -p $APP_HOME addgroup --system app && \
WORKDIR $APP_HOME adduser --system --group app
# add folder to store repo in
RUN mkdir -p /home/app/repo RUN mkdir -p /home/app/repo
# install packages from builder # install packages from builder
@@ -36,7 +38,14 @@ COPY --from=builder /usr/src/app/requirements.txt requirements.txt
RUN pip install --no-cache /wheels/* RUN pip install --no-cache /wheels/*
# copy in files # copy in files
WORKDIR /home/app
COPY ./code . COPY ./code .
# change ownership to app user
RUN chown -R app:app /home/app
# change to the app user
USER app
# start execution # start execution
ENTRYPOINT [ "python", "main.py" ] ENTRYPOINT [ "python", "main.py" ]
+17 -4
View File
@@ -62,7 +62,7 @@ def handle_event(request: GiteaRequest):
runner_script = load_runner_script(path=runner_script_path) runner_script = load_runner_script(path=runner_script_path)
# determine pipeline to run # determine pipeline to run
if branch_name not in runner_script.pipelines.keys(): if branch_name not in runner_script.pipelines.keys():
# run default # get default pipeline
if 'default' not in runner_script.pipelines.keys(): if 'default' not in runner_script.pipelines.keys():
raise HTTPException( raise HTTPException(
status_code=404, status_code=404,
@@ -70,15 +70,28 @@ def handle_event(request: GiteaRequest):
) )
pipeline = runner_script.pipelines['default'] pipeline = runner_script.pipelines['default']
else: else:
# run custom # get custom pipeline
pipeline = runner_script.pipelines[branch_name] pipeline = runner_script.pipelines[branch_name]
print(pipeline) logging.info(f'running pipeline {pipeline.name}')
# TODO: create virtualenv?
# install requirements
assert pipeline.requirements is not None
requirements_path = local_repo_dir / pipeline.requirements
os.system(f'pip install -r {requirements_path}')
for cmd in pipeline.script:
try:
os.system(cmd)
except Exception as e:
raise HTTPException(
status_code=503,
detail=f'failed running {cmd}: {e}'
)
# send response # send response
raise HTTPException( raise HTTPException(
status_code=202, status_code=202,
detail='received task' detail='received task'
) )
# consider to include link to page that show progress
if __name__ == '__main__': if __name__ == '__main__':
load_dotenv('dev.env') load_dotenv('dev.env')
+7
View File
@@ -11,3 +11,10 @@ services:
- LISTEN_IP=0.0.0.0 - LISTEN_IP=0.0.0.0
- LISTEN_PORT=8000 - LISTEN_PORT=8000
- REPO_DIR=/home/app/repo/ - REPO_DIR=/home/app/repo/
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# device_ids: ['0']
# capabilities: [gpu]
+1 -2
View File
@@ -4,8 +4,7 @@ definitions:
name: test name: test
requirements: req-prod.txt requirements: req-prod.txt
script: script:
- command arg1 arg2 - nvidia-smi
- command2 arg1 arg2
pipelines: pipelines:
default: default: