2013-04-07 176 views
6

Mongoengine将FileField和ImageField存储到GridFS。什么是最简单的方法来复制原始文件/图像领域的功能?编辑:Mongoengine FileField保存到磁盘?

所以这是我现在的班级。我可以加载文件并将其保存到磁盘,Mongo保存数据库中文件的路径。

我摔倒在“to_python”,因为我认为它需要创建一个proxy_class的对象,但我不明白如何,如果我所得到的是文件的路径(作为值传递在)。

import os 
import datetime 

from mongoengine.python_support import str_types 
from django.db.models.fields.files import FieldFile 
from django.core.files.base import File 
from django.core.files.storage import default_storage 
from mongoengine.base import BaseField 
from mongoengine.connection import get_db, DEFAULT_CONNECTION_NAME 
from django.utils.encoding import force_text 
#from django.utils.encoding import force_str 


class DJFileField(BaseField): 

    proxy_class = FieldFile 

    def __init__(self, 
       db_alias=DEFAULT_CONNECTION_NAME, 
       name=None, 
       upload_to='', 
       storage=None, 
       **kwargs): 

     self.db_alias = db_alias 
     self.storage = storage or default_storage 
     self.upload_to = upload_to 

     if callable(upload_to): 
      self.generate_filename = upload_to 

     super(DJFileField, self).__init__(**kwargs) 


    def __get__(self, instance, owner): 
     # Lots of information on whats going on here can be found 
     # on Django's FieldFile implementation, go over to GitHub to 
     # read it. 
     file = instance._data.get(self.name) 

     if isinstance(file, str_types) or file is None: 
      attr = self.proxy_class(instance, self, file) 
      instance._data[self.name] = attr 

     elif isinstance(file, File) and not isinstance(file, FieldFile): 
      file_copy = self.proxy_class(instance, self, file.name) 
      file_copy.file = file 
      file_copy._committed = False 
      instance._data[self.name] = file_copy 

     elif isinstance(file, FieldFile) and not hasattr(file, 'field'): 
      file.instance = instance 
      file.field = self 
      file.storage = self.storage 

     # That was fun, wasn't it? 
     return instance._data[self.name] 


    def __set__(self, instance, value): 
     instance._data[self.name] = value 

    # The 3 methods below get used by the FieldFile proxy_object 
    def get_directory_name(self): 
     return os.path.normpath(force_text(datetime.datetime.now().strftime(self.upload_to))) 

    def get_filename(self, filename): 
     return os.path.normpath(self.storage.get_valid_name(os.path.basename(filename))) 

    def generate_filename(self, instance, filename): 
     return os.path.join(self.get_directory_name(), self.get_filename(filename)) 


    def to_mongo(self, value): 
    # Store the path in MongoDB 
    # I also used this bit to actually save the file to disk. 
    # The value I'm getting here is a FileFiled and it all looks 
    # pretty good at this stage even though I'm not 100% sure 
    # of what's going on. 

     import ipdb; ipdb.set_trace() 

     if not value._committed and value is not None: 
      value.save(value.name, value) 
      return value.path 

     return value.path  


    def to_python(self, value): 
     # Now this is the real problem, value is the path that got saved 
     # in mongo. No idea how to return a FileField obj from here. 
     # self.instance and instance throw errors. 
+0

这是一个真正的问题吗?你能扩展你想要的领域吗? – Ross 2013-04-10 11:38:35

+0

罗斯确定了一个真正的问题。我的意思是:如果我想要一个新的Field,它基本上与Django中的原始FileField的行为方式相同,您会建议我做什么。使用存储对象和mongo仅用于路径信息等。 – holografix 2013-04-11 02:05:56

回答

2

我认为这将是一个很好的补充 - 或许称为LocalFileField,使之更加框架不可知,如果你提供的测试和文档将成为一位伟大的除了https://github.com/MongoEngine/extras-mongoengine

我的唯一原因没有在核心中销售它 - 如果您正在运行复制集,该文件仍然只能存储在一个节点上。

+1

感谢您的回复Ross。你会推荐我做什么样的测试?你能举几个例子吗?我尊敬地不同意,我认为它应该是核心的一部分,因为我看不到一种简单的方式来上传与可以在模板上使用的文档相关联的图像,我想很多开发人员都希望这样做。或者我错过了一些明显的东西,即:如何在模板上使用当前的MongoEngine.ImageField。文档可能会建议您不要使用本地文件系统,而应使用S3之类的东西。 – holografix 2013-04-20 01:58:21