python - pymongo - 如何通过 GridFS 中的索引创建/查询

Question

就我而言，我需要通过 SHA1 确保文件的唯一性(存储为文件名)

db = pymongo.MongoClient('localhost', 27017).test
gfs = gridfs.GridFS(db)

# How may I create a unique index in GridFS?
gfs.files.create_index([('filename', 1)], unique=True)

如果文件已经存储，则通过SHA1查找该文件。

sha1 = hashlib.sha1(file_content).hexdigest()
try:
    return gfs.put(file_content, filename=sha1)
except pymongo.errors.DuplicateKeyError:

    # How may I find files via criterion?
    return gfs.find( { 'filename': sha1 } )['_id']

有人能告诉我如何做这些事情吗？提前致谢。

Answer 1

您可以手动为具有自身哈希值的文件提供 _id 键，而不是创建索引。

import pymongo 
db = pymongo.MongoClient('localhost', 27017).test
gfs = gridfs.GridFS(db)

def hash(file):
   #some code to extract hash of a file from its content..

file_hash = hash(file)
if gfs.exists(_id=file_hash):
    #file exists!
else:
    #file does not exist in the database.
    gfs.put(file, _id=file_hash) #or do something else..

http://api.mongodb.org/python/current/api/gridfs/