class FSContainer(object):
def __init__(self, container_name, obj):
self._container_name = container_name
self._obj = obj
self._fs = GridFS(self._obj.db)
if Magic:
self._magic = Magic(mime=True)
def __getitem__(self, key):
f = self.open(key)
content = f.read()
f.close()
return content
def __setitem__(self, key, value):
content_type = None
if value and Magic:
content_type = self._magic.from_buffer(value)
f = self.open(key, 'w')
try:
f.content_type = content_type
f.write(value)
except TypeError:
raise TypeError("GridFS value mus be string not %s" % type(value))
finally:
f.close()
def __delitem__(self, key):
spec = {'metadata.doc_id':self._obj['_id'], 'metadata.container':self._container_name, 'metadata.name':key}
self._fs.remove(spec,collection=self._obj.collection.name)
def open(self, name, mode='r'):
search_spec = {'metadata.name':name, 'metadata.container': self._container_name, 'metadata.doc_id':self._obj['_id']}
if mode == 'r':
try:
return GridFile(search_spec, self._obj.db, 'r', self._obj.collection.name)
except IOError:
raise IOError('"%s" is not found in the database' % name)
else:
file = self._obj.collection.files.find_one(search_spec)
if file:
return GridFile({'_id':ObjectId(file['_id'])}, self._obj.db, 'w', self._obj.collection.name)
write_spec = {'metadata':{'name':name, 'container':self._container_name, 'doc_id':self._obj['_id']}}
return GridFile(write_spec, self._obj.db, 'w', self._obj.collection.name)
def __iter__(self):
for metafile in self._obj.collection.files.find(
{'metadata.container': self._container_name, 'metadata.doc_id': self._obj['_id']}):
yield metafile['metadata']['name']
def list(self):
return [i for i in self]
def __repr__(self):
return "<%s '%s'>" % (self.__class__.__name__, self._container_name)
class FS(object):
def __init__(self, gridfs, obj):
self._gridfs = gridfs
for container in self._gridfs.get('containers', []):
self.__dict__[container] = FSContainer(container, obj)
self._obj = obj
self._fs = GridFS(self._obj.db)
if Magic:
self._magic = Magic(mime=True)
def __getitem__(self, key):
f = self.open(key)
content = f.read()
f.close()
return content
def __setitem__(self, key, value):
content_type = None
if value and Magic:
content_type = self._magic.from_buffer(value)
f = self.open(key, 'w')
try:
f.content_type = content_type
f.write(value)
except TypeError:
raise TypeError("GridFS value mus be string not %s" % type(value))
finally:
f.close()
def __getattr__(self, key):
if key not in ['_gridfs', '_obj', '_fs', '_containers', '_magic']:
if key not in self._gridfs.get('containers', []) and key in self._gridfs.get('files', []):
return self[key]
return super(FS, self).__getattribute__(key)
def __setattr__(self, key, value):
if key not in ['_gridfs', '_obj', '_fs', '_containers', '_magic']:
if key not in self._gridfs.get('containers', []) and key in self._gridfs.get('files', []):
self[key] = value
else:
super(FS, self).__setattr__(key, value)
def __delitem__(self, key):
self._fs.remove({'metadata.doc_id':self._obj['_id'],
'metadata.name':key}, collection=self._obj.collection.name)
def __delattr__(self, key):
del self[key]
def open(self, name, mode='r'):
assert name in self._gridfs.get('files', []), "%s is not declared in gridfs" % name
search_spec = {'metadata.name':name, 'metadata.doc_id':self._obj['_id']}
if mode == 'r':
try:
return GridFile(search_spec, self._obj.db, 'r', self._obj.collection.name)
except IOError:
raise IOError('"%s" is not found in the database' % name)
else:
file = self._obj.collection.files.find_one(search_spec)
if file:
return GridFile({'_id':ObjectId(file['_id'])}, self._obj.db, 'w', self._obj.collection.name)
write_spec = {'metadata':{'name':name, 'doc_id':self._obj['_id']}}
return GridFile(write_spec, self._obj.db, 'w', self._obj.collection.name)
def __iter__(self):
for i in self._obj.collection.files.find({'metadata.doc_id': self._obj['_id']}):
container, name = i['metadata'].get('container'), i['metadata']['name']
if container:
name = "%s/%s" % (container, name)
yield name
def list(self):
return [i for i in self]
def __repr__(self):
return "<%s of object '%s'>" % (self.__class__.__name__, self._obj['_id'])
