在运行docker image时将url作为参数传递到python中

Question

我想在运行docker镜像时传递一个GCP存储URL作为参数，这样它就可以从我的存储中拉出我的csv文件并打印数据集。

下面是我的dockerfile

# Use the official lightweight Python image.
# https://hub.docker.com/_/python
FROM continuumio/miniconda3


# Copy local code to the container image.
ENV APP_HOME /app
WORKDIR $APP_HOME
COPY . ./

# Install production dependencies.
RUN pip install Flask gunicorn
RUN pip install scikit-learn==0.20.2 firefly-python==0.1.15
RUN pip install --upgrade google-cloud-storage
ENTRYPOINT ["python"]
CMD ["pre.py"]

我试着用下面的命令运行docker镜像，得到下面的错误

docker run preprocess:v1 "https://storage.googleapis.com/MYBucket/app/Model/IrisClassifier.sav"

。

python: can't open file 'https://storage.googleapis.com/MYBucket/app/Model/IrisClassifier.sav': [Errno 2] No such file or directory

import os
import argparse
from google.cloud import storage
from sklearn.externals import joblib
from urllib.request import urlopen

def parse_arguments():
    print('entered parse arg')
    parser = argparse.ArgumentParser()
    parser.add_argument('data_dir', type=str, help='GCSpath')
    args = parser.parse_known_args()[0]
    print('Argument passed')
    print(os.getcwd())
    print('STARTING CLOUD RETRIVAL')
    print('*****client initialized')
    dataset_load = joblib.load(urlopen(args.dat_dir))
    print('*****loaded Dataset')
    print(dataset_load)


def main(_):
    print("Prior to entering arg")
    parse_arguments()

我想在运行docker镜像https://storage.googleapis.com/MYBucket/app/Model/IrisClassifier.sav时传递一个类似的GCP bucket

Answer 1

你需要先把你所有的CMD改成ENTRYPOINT：

FROM continuumio/miniconda3

ENV APP_HOME /app
WORKDIR $APP_HOME
COPY . ./

RUN pip install Flask gunicorn
RUN pip install scikit-learn==0.20.2 firefly-python==0.1.15
RUN pip install --upgrade google-cloud-storage
ENTRYPOINT ["python", "pre.py"]

然后你可以通过你的URL。

您的设置存在以下问题：

docker将启动入口点，即python，并使用您的命令覆盖CMD，这将为您提供：

python YOUR_URL

更新

我不知道你是否添加了if语句来运行main def，但是在这里你应该如何编辑脚本：

def main():
    print("Prior to entering arg")
    parse_arguments()


if __name__ == '__main__':
    main()