add_picture_to_db.py 文件源码

def add_image(dataset_pk, image):
    try:
        dataset = db.Dataset.get(db.Dataset.dataset == dataset_pk)
    except peewee.DoesNotExist:
        print("Can't find any dataset with id {}".format(dataset_pk))
        sys.exit(1)

    with open(image, 'rb') as f:
        data = f.read()

    try:
        mimetype = infer_mimetype( image )
    except NoMimetypeException:
        print("Can't find mime type for <{}>".format(image))
        sys.exit(2)

    print(len(data))
    with db.database.atomic():
        db.DatasetLogo.create(
            dataset  = dataset,
            mimetype = mimetype,
            data     = data
        )