Как использовать почтальон для загрузки данных DBRef в json? - PullRequest
Новая
       36

Как использовать почтальон для загрузки данных DBRef в json?

0 голосов
/ 15 марта 2020

Я пытаюсь отправить данные в mongodb с помощью почтальона, но я не знаю правильного соглашения для загрузки ссылки на файл изображения в корзину fs.files. По сути, файл уже находится в базе данных, я просто пытаюсь опубликовать нового пользователя со ссылкой на изображение.

Вот моя модель: 

class Users(db.Document):
    _id = db.StringField()
    name = db.StringField()
    picture = db.FileField()
    email = db.StringField()
    password = db.StringField()
    meta = {'collection': 'Users'}

В почтальоне, Я пытаюсь опубликовать данные следующим образом: 

{
"_id" : "1",
"name" : "John Doe",
"picture": [{"$id": "5e6a...f9q102"}], #This is the reference id for the image already in the database, in fs.files
"password" : "<hashed pw>",
"email" : "example@example.com"
}

Я использую flask restful api, поэтому в скрипте python функция post определяется следующим образом: 

def post(self):
    body = request.get_json()
    print (body)
    user = Users()
    user = Users(**body).save()
    return 'Successful Upload', 200

Я получаю сообщение об ошибке при попытке использовать вышеуказанное соглашение: 

mongoengine.errors.ValidationError: ValidationError (Users:None) ('list' object has no attribute
    'grid_id': ['picture'])

Как мне опубликовать нового пользователя в почтальоне? Ваша помощь приветствуется

Ответы [ 2 ]

0 голосов
/ 16 марта 2020

Это была ошибка проверки. База данных принимала JSON в определенном формате, чем я публиковал. И способ обработки почтового запроса также был неверным. Это ожидаемый формат: 

{
    ...,
    "picture" = {"$ref": "fs.files", 
                 "$id": ObjectId("5e6a...f9q102")},
    ...
}

Почтальон не может принять вышеуказанный формат, вместо этого он принял это: 

{
    "_id" : "1",
    "name" : "John Doe",
    "picture": {"$ref": "fs.files", "$id": {"$oid": "5e6a...f9q102"}}, 
    "password" : "<hashed pw>",
    "email" : "example@example.com"
}

Чтобы сделать эту работу, я изменил модель, чтобы она выглядела так в моем flask Приложении: 

class Users(db.Document):
    _id = db.StringField()
    name = db.StringField()
    picture = db.ReferenceField('fs.files') #I changed this to a reference field because it holds the reference for the file and not the actual file in the database
    upload_picture = db.FileField() #I added this field so I can still upload pics via flask and via this document
    email = db.StringField()
    password = db.StringField()
    meta = {'collection': 'Users'}

Тогда я должен был добавить этот импорт и изменить код так, что он будет читать ввод как JSON и передать опорное значение изображений на ObjectId(id) так что он соответствует формату, который ожидала база данных. 

from bson.json_util import loads

def post(self):
        body = str(request.get_json())
        x = body.replace("'", '"') #replace single quotes as double quotes to match JSON format
        data = loads(x)
        officer = Officers(**data).save()
        return 'Successful Upload', 200

Тогда вуаля, все работает!

0 голосов
/ 15 марта 2020

Вам нужно немного изменить свой код

Добавить эти импорта: 

from mongoengine.fields import ImageGridFsProxy
from mongoengine import ReferenceField, DynamicDocument
from bson.dbref import DBRef
from bson import ObjectId

Изменить ваш класс picture Определение поля + добавить дополнительный класс fs

class Fs(DynamicDocument):
    #Add 'db_alias':'default' to meta
    meta = {'collection': 'fs.files'}

class Users(Document):
    ...
    picture = ReferenceField('Fs', dbref=True)
    ...

Теперь вам нужно создать новый экземпляр для DBRef следующим образом: 

def post(self):
    body = request.get_json()
    body["picture"] = DBRef('fs.files', ObjectId(body["picture"]))

    #mongoengine assumes `ObjectId('xxx')` already exists in `fs.files`. 
    #If you want to check, run below code:
    #if Fs.objects(_id=body["picture"].id).first() is None:
    #    return 'Picture ' + str(body["picture"].id) + ' not found', 400

    user = Users(**body).save()
    return 'Successful Upload', 200

В конце, если вам нужно прочитать содержимое изображения: 

image = ImageGridFsProxy(grid_id=ObjectId('xxx'))
f = open("image.png", "wb")
f.write(image.read())
f.close()
...