def index(self, proxy=None):
if proxy is None:
proxy=aq_inner(self.context)
if not self.approve(proxy):
return
if IReflectoDirectory.providedBy(proxy):
self.unindex(proxy)
for (id, entry) in proxy.iteritems():
if IReflectoDirectory.providedBy(entry):
self.index(entry)
elif IReflectoFile.providedBy(entry):
entry.indexObject()
proxy.indexObject()
def index(self, proxy=None):
if proxy is None:
proxy = aq_inner(self.context)
if not self.approve(proxy):
return
if IReflectoDirectory.providedBy(proxy):
self.unindex(proxy)
for (id, entry) in proxy.iteritems():
if IReflectoDirectory.providedBy(entry):
self.index(entry)
elif IReflectoFile.providedBy(entry):
entry.indexObject()
proxy.indexObject()
def testUnrestrictedTraverse(self):
self.assertTrue(IReflectoFile.providedBy(
self.reflecto.unrestrictedTraverse("reflecto.txt")))
self.assertTrue(IReflectoDirectory.providedBy(
self.reflecto.unrestrictedTraverse("subdir")))
self.assertTrue(IReflectoFile.providedBy(
self.reflecto.unrestrictedTraverse("subdir/emptyfile.txt")))
def testUnrestrictedTraverse(self):
self.assertTrue(
IReflectoFile.providedBy(
self.reflecto.unrestrictedTraverse("reflecto.txt")))
self.assertTrue(
IReflectoDirectory.providedBy(
self.reflecto.unrestrictedTraverse("subdir")))
self.assertTrue(
IReflectoFile.providedBy(
self.reflecto.unrestrictedTraverse("subdir/emptyfile.txt")))
def _addToZip(self, zip, entry, path=[]):
newpath=path + [entry.__name__]
if IReflectoDirectory.providedBy(entry):
for key in entry.keys():
if not self.acceptableName(key):
continue
value=entry[key]
if not self.acceptableEntry(value):
continue
self._addToZip(zip, entry[key], newpath)
elif IReflectoFile.providedBy(entry):
zip.write(entry.getFilesystemPath(), os.path.join(*newpath))
def _addToZip(self, zip, entry, path=[]):
newpath = path + [entry.__name__]
if IReflectoDirectory.providedBy(entry):
for key in entry.keys():
if not self.acceptableName(key):
continue
value = entry[key]
if not self.acceptableEntry(value):
continue
self._addToZip(zip, entry[key], newpath)
elif IReflectoFile.providedBy(entry):
zip.write(entry.getFilesystemPath(), os.path.join(*newpath))
def testDirectory(self):
self.assertTrue(IReflectoDirectory.providedBy(self.reflecto['subdir']))
def COPY(self, REQUEST, RESPONSE):
"""Create a duplicate of the source resource. This is only allowed
within the same reflecto directory."""
self.dav__init(REQUEST, RESPONSE)
if not hasattr(aq_base(self), 'cb_isCopyable') or \
not self.cb_isCopyable():
raise MethodNotAllowed, 'This object may not be copied.'
depth=REQUEST.get_header('Depth', 'infinity')
if not depth in ('0', 'infinity'):
raise BadRequest, 'Invalid Depth header.'
dest=REQUEST.get_header('Destination', '')
while dest and dest[-1]=='/':
dest=dest[:-1]
if not dest:
raise BadRequest, 'Invalid Destination header.'
try: path = REQUEST.physicalPathFromURL(dest)
except ValueError:
raise BadRequest, 'Invalid Destination header'
name = path.pop()
oflag=REQUEST.get_header('Overwrite', 'F').upper()
if not oflag in ('T', 'F'):
raise BadRequest, 'Invalid Overwrite header.'
try: parent=self.restrictedTraverse(path)
except ValueError:
raise Conflict, 'Attempt to copy to an unknown namespace.'
except NotFound:
raise Conflict, 'Object ancestors must already exist.'
except:
t, v, tb=sys.exc_info()
raise t, v
if hasattr(parent, '__null_resource__'):
raise Conflict, 'Object ancestors must already exist.'
existing=hasattr(aq_base(parent), name)
if existing and oflag=='F':
raise PreconditionFailed, 'Destination resource exists.'
try:
parent._checkId(name, allow_dup=1)
except:
raise Forbidden, sys.exc_info()[1]
try:
parent._verifyObjectPaste(self)
except Unauthorized:
raise
except:
raise Forbidden, sys.exc_info()[1]
# Now check locks. The If header on a copy only cares about the
# lock on the destination, so we need to check out the destinations
# lock status.
ifhdr = REQUEST.get_header('If', '')
if existing:
# The destination itself exists, so we need to check its locks
destob = aq_base(parent)._getOb(name)
if IWriteLock.providedBy(destob) and destob.wl_isLocked():
if ifhdr:
itrue = destob.dav__simpleifhandler(
REQUEST, RESPONSE, 'COPY', refresh=1)
if not itrue:
raise PreconditionFailed
else:
raise Locked, 'Destination is locked.'
elif IWriteLock.providedBy(parent) and parent.wl_isLocked():
if ifhdr:
parent.dav__simpleifhandler(REQUEST, RESPONSE, 'COPY',
refresh=1)
else:
raise Locked, 'Destination is locked.'
self._notifyOfCopyTo(parent, op=0)
#### This part is reflecto specific
if existing:
object=getattr(parent, name)
self.dav__validate(object, 'DELETE', REQUEST)
parent.manage_delObjects([name])
oldpath = self.getFilesystemPath()
newpath = os.path.join(parent.getFilesystemPath(), name)
if IReflectoDirectory.providedBy(self):
if depth=='0':
os.mkdir(newpath, 0775)
else:
shutil.copytree(oldpath, newpath)
else:
shutil.copy2(oldpath, newpath)
ob = parent[name]
ob.indexObject()
notify(ObjectCopiedEvent(ob, self))
notify(ObjectClonedEvent(ob))
notifyContainerModified(parent)
####
# We remove any locks from the copied object because webdav clients
# don't track the lock status and the lock token for copied resources
ob.wl_clearLocks()
RESPONSE.setStatus(existing and 204 or 201)
if not existing:
RESPONSE.setHeader('Location', dest)
RESPONSE.setBody('')
return RESPONSE
def manage_pasteObjects(self, cp):
"""Paste previously copied objects into the current object."""
op, mdatas = _cb_decode(cp)
COPY = op == 0 # Otherwise its a paste operation
# Copy or paste always fails without write permission here
sman = getSecurityManager()
if not sman.checkPermission(AddFilesystemObject, self):
raise CopyError, 'Insufficient Privileges'
oblist = []
app = self.getPhysicalRoot()
# Loading and security checks
for mdata in mdatas:
m = loadMoniker(mdata)
try:
ob = m.bind(app)
except ConflictError:
raise
except:
raise CopyError, 'Item not found'
if not IReflectoProxy.providedBy(ob) or IReflector.providedBy(ob):
raise CopyError, 'Cannot paste into this object'
parent = aq_parent(aq_inner(ob))
if not sman.validate(None, parent, None, ob):
raise Unauthorized(ob.getId())
if not COPY:
if not sman.checkPermission(DeleteObjects, parent):
raise Unauthorized('Delete not allowed')
prefix = os.path.commonprefix((ob.getFilesystemPath(),
self.getFilesystemPath()))
if prefix == ob.getFilesystemPath():
raise CopyError, "This object cannot be pasted into itself"
oblist.append(ob)
result = []
problem_ids = []
path = self.getFilesystemPath()
for ob in oblist:
old = ob.getId()
new = self._generate_copy_id(old)
result.append(dict(id=old, new_id=new))
oldpath = ob.getFilesystemPath()
newpath = os.path.join(path, new)
if COPY:
try:
if IReflectoDirectory.providedBy(ob):
shutil.copytree(oldpath, newpath)
else:
shutil.copy2(oldpath, newpath)
notify(ObjectCopiedEvent(self[new], ob))
notify(ObjectClonedEvent(self[new]))
except EnvironmentError:
problem_ids.append(result.pop())
else:
# paste/move
oldparent = aq_parent(aq_inner(ob))
notify(ObjectWillBeMovedEvent(ob, oldparent, old, self, new))
if aq_base(self) is aq_base(oldparent):
# No need to move from self to self
result[-1]['new_id'] = old
# Original CopyContainer does emit events for this case
notify(ObjectMovedEvent(ob, self, old, self, old))
continue
try:
shutil.move(oldpath, newpath)
indexview = oldparent.unrestrictedTraverse('@@index')
indexview.index()
notify(ObjectMovedEvent(self[new], oldparent, old, self,
new))
notifyContainerModified(oldparent)
except EnvironmentError:
if os.path.exists(newpath):
# partial move
try:
if os.path.isdir(newpath):
shutil.rmtree(newpath)
else:
os.unlink(newpath)
except EnvironmentError:
pass # Really weird access issues, time to bail
problem_ids.append(result.pop())
if problem_ids:
sm = IStatusMessage(getattr(self, 'REQUEST', None), None)
if sm is not None:
sm.addStatusMessage(
'Failed to copy or paste some files: %s' % problem_ids,
'stop')
if result:
indexview = self.unrestrictedTraverse('@@index')
indexview.index()
notifyContainerModified(self)
return result
def COPY(self, REQUEST, RESPONSE):
"""Create a duplicate of the source resource. This is only allowed
within the same reflecto directory."""
self.dav__init(REQUEST, RESPONSE)
if not hasattr(aq_base(self), 'cb_isCopyable') or \
not self.cb_isCopyable():
raise MethodNotAllowed, 'This object may not be copied.'
depth = REQUEST.get_header('Depth', 'infinity')
if not depth in ('0', 'infinity'):
raise BadRequest, 'Invalid Depth header.'
dest = REQUEST.get_header('Destination', '')
while dest and dest[-1] == '/':
dest = dest[:-1]
if not dest:
raise BadRequest, 'Invalid Destination header.'
try:
path = REQUEST.physicalPathFromURL(dest)
except ValueError:
raise BadRequest, 'Invalid Destination header'
name = path.pop()
oflag = REQUEST.get_header('Overwrite', 'F').upper()
if not oflag in ('T', 'F'):
raise BadRequest, 'Invalid Overwrite header.'
try:
parent = self.restrictedTraverse(path)
except ValueError:
raise Conflict, 'Attempt to copy to an unknown namespace.'
except NotFound:
raise Conflict, 'Object ancestors must already exist.'
except:
t, v, tb = sys.exc_info()
raise t, v
if hasattr(parent, '__null_resource__'):
raise Conflict, 'Object ancestors must already exist.'
existing = hasattr(aq_base(parent), name)
if existing and oflag == 'F':
raise PreconditionFailed, 'Destination resource exists.'
try:
parent._checkId(name, allow_dup=1)
except:
raise Forbidden, sys.exc_info()[1]
try:
parent._verifyObjectPaste(self)
except Unauthorized:
raise
except:
raise Forbidden, sys.exc_info()[1]
# Now check locks. The If header on a copy only cares about the
# lock on the destination, so we need to check out the destinations
# lock status.
ifhdr = REQUEST.get_header('If', '')
if existing:
# The destination itself exists, so we need to check its locks
destob = aq_base(parent)._getOb(name)
if IWriteLock.providedBy(destob) and destob.wl_isLocked():
if ifhdr:
itrue = destob.dav__simpleifhandler(REQUEST,
RESPONSE,
'COPY',
refresh=1)
if not itrue:
raise PreconditionFailed
else:
raise Locked, 'Destination is locked.'
elif IWriteLock.providedBy(parent) and parent.wl_isLocked():
if ifhdr:
parent.dav__simpleifhandler(REQUEST,
RESPONSE,
'COPY',
refresh=1)
else:
raise Locked, 'Destination is locked.'
self._notifyOfCopyTo(parent, op=0)
#### This part is reflecto specific
if existing:
object = getattr(parent, name)
self.dav__validate(object, 'DELETE', REQUEST)
parent.manage_delObjects([name])
oldpath = self.getFilesystemPath()
newpath = os.path.join(parent.getFilesystemPath(), name)
if IReflectoDirectory.providedBy(self):
if depth == '0':
os.mkdir(newpath, 0775)
else:
shutil.copytree(oldpath, newpath)
else:
shutil.copy2(oldpath, newpath)
ob = parent[name]
ob.indexObject()
notify(ObjectCopiedEvent(ob, self))
notify(ObjectClonedEvent(ob))
notifyContainerModified(parent)
####
# We remove any locks from the copied object because webdav clients
# don't track the lock status and the lock token for copied resources
ob.wl_clearLocks()
RESPONSE.setStatus(existing and 204 or 201)
if not existing:
RESPONSE.setHeader('Location', dest)
RESPONSE.setBody('')
return RESPONSE
def testDirectory(self):
self.assertTrue(IReflectoDirectory.providedBy(self.reflecto['subdir']))
def manage_pasteObjects(self, cp):
"""Paste previously copied objects into the current object."""
op, mdatas = _cb_decode(cp)
COPY = op == 0 # Otherwise its a paste operation
# Copy or paste always fails without write permission here
sman = getSecurityManager()
if not sman.checkPermission(AddFilesystemObject, self):
raise CopyError, 'Insufficient Privileges'
oblist = []
app = self.getPhysicalRoot()
# Loading and security checks
for mdata in mdatas:
m = loadMoniker(mdata)
try:
ob = m.bind(app)
except ConflictError:
raise
except:
raise CopyError, 'Item not found'
if not IReflectoProxy.providedBy(ob) or IReflector.providedBy(ob):
raise CopyError, 'Cannot paste into this object'
parent = aq_parent(aq_inner(ob))
if not sman.validate(None, parent, None, ob):
raise Unauthorized(ob.getId())
if not COPY:
if not sman.checkPermission(DeleteObjects, parent):
raise Unauthorized('Delete not allowed')
prefix = os.path.commonprefix(
(ob.getFilesystemPath(), self.getFilesystemPath()))
if prefix == ob.getFilesystemPath():
raise CopyError, "This object cannot be pasted into itself"
oblist.append(ob)
result = []
problem_ids = []
path = self.getFilesystemPath()
for ob in oblist:
old = ob.getId()
new = self._generate_copy_id(old)
result.append(dict(id=old, new_id=new))
oldpath = ob.getFilesystemPath()
newpath = os.path.join(path, new)
if COPY:
try:
if IReflectoDirectory.providedBy(ob):
shutil.copytree(oldpath, newpath)
else:
shutil.copy2(oldpath, newpath)
notify(ObjectCopiedEvent(self[new], ob))
notify(ObjectClonedEvent(self[new]))
except EnvironmentError:
problem_ids.append(result.pop())
else:
# paste/move
oldparent = aq_parent(aq_inner(ob))
notify(ObjectWillBeMovedEvent(ob, oldparent, old, self, new))
if aq_base(self) is aq_base(oldparent):
# No need to move from self to self
result[-1]['new_id'] = old
# Original CopyContainer does emit events for this case
notify(ObjectMovedEvent(ob, self, old, self, old))
continue
try:
shutil.move(oldpath, newpath)
indexview = oldparent.unrestrictedTraverse('@@index')
indexview.index()
notify(
ObjectMovedEvent(self[new], oldparent, old, self, new))
notifyContainerModified(oldparent)
except EnvironmentError:
if os.path.exists(newpath):
# partial move
try:
if os.path.isdir(newpath):
shutil.rmtree(newpath)
else:
os.unlink(newpath)
except EnvironmentError:
pass # Really weird access issues, time to bail
problem_ids.append(result.pop())
if problem_ids:
sm = IStatusMessage(getattr(self, 'REQUEST', None), None)
if sm is not None:
sm.addStatusMessage(
'Failed to copy or paste some files: %s' % problem_ids,
'stop')
if result:
indexview = self.unrestrictedTraverse('@@index')
indexview.index()
notifyContainerModified(self)
return result