def test_attachments(self):
su = StatusUpdate("foo bar")
attachments = IAttachmentStorage(su)
f = ATFile("data.dat")
attachments.add(f)
self.assertEqual([k for k in attachments.keys()], [f.getId()])
attachments.remove(f.getId())
self.assertEqual(len(attachments.keys()), 0)
def test_attachments(self):
su = StatusUpdate('foo bar')
attachments = IAttachmentStorage(su)
f = ATFile('data.dat')
attachments.add(f)
self.assertEqual([k for k in attachments.keys()], [f.getId()])
attachments.remove(f.getId())
self.assertEqual(len(attachments.keys()), 0)
def clean_up_temporary_attachments(workspace, maxage=1):
temp_attachments = IAttachmentStorage(workspace)
for key in temp_attachments.keys():
keyparts = key.split('-')
datestr = keyparts[1]
try:
date = datetime.strptime(datestr, '%Y%m%d%H%M%S%f')
except ValueError:
date = datetime.min # No proper datestr, treat as old
if datetime.now() - date > timedelta(maxage):
temp_attachments.remove(key)
log.info('Cleaned up temporary attachment {0} from '
'{1}'.format(key, workspace.Title()))
def pop_temporary_attachment(workspace, file_field, token):
"""
Replace a temporary attachment on the workspace with
the uploaded data
"""
temp_attachments = IAttachmentStorage(workspace)
temp_id = '{0}-{1}'.format(token, file_field.filename)
if temp_id in temp_attachments.keys():
temp_att = aq_base(temp_attachments.get(temp_id))
temp_att.id = file_field.filename
temp_att.file = NamedBlobFile(
data=file_field.read(),
filename=file_field.filename.decode('utf-8'),
)
temp_attachments.remove(temp_id)
return temp_att
return None
def test_remove(self):
""" """
question = createObject('slc.underflow.question')
attachments = IAttachmentStorage(question)
self.assertEqual(len(attachments.keys()), 0)
self.assertEqual(len(attachments.values()), 0)
for fname in ['data1.dat', 'data2.dat']:
attachments.add(file.ATFile(fname))
self.assertEqual(len(attachments.keys()), 2)
self.assertEqual(len(attachments.values()), 2)
self.assertRaises(KeyError, attachments.remove, 'data3.dat')
attachments.remove('data1.dat')
self.assertEqual(len(attachments.keys()), 1)
self.assertTrue('data2.dat' in attachments.keys())
attachments.remove('data2.dat')
self.assertEqual(len(attachments.keys()), 0)
def pop_temporary_attachment(workspace, file_field, token):
"""
Replace a temporary attachment on the workspace with
the uploaded data
"""
temp_attachments = IAttachmentStorage(workspace)
temp_id = getUtility(IURLNormalizer).normalize(u'{0}-{1}'.format(
token, safe_unicode(file_field.filename)))
if temp_id in temp_attachments.keys():
temp_att = aq_base(temp_attachments.get(temp_id))
temp_att.id = file_field.filename
temp_att.file = NamedBlobFile(
data=file_field.read(),
filename=file_field.filename.decode('utf-8'),
)
temp_attachments.remove(temp_id)
return temp_att
return None
def test_remove(self):
""" """
question = createObject('Document')
alsoProvides(question, IAttachmentStoragable)
attachments = IAttachmentStorage(question)
self.assertEqual(len(attachments.keys()), 0)
self.assertEqual(len(attachments.values()), 0)
for fname in ['data1.dat', 'data2.dat']:
attachments.add(file.ATFile(fname))
self.assertEqual(len(attachments.keys()), 2)
self.assertEqual(len(attachments.values()), 2)
self.assertRaises(KeyError, attachments.remove, 'data3.dat')
attachments.remove('data1.dat')
self.assertEqual(len(attachments.keys()), 1)
self.assertTrue('data2.dat' in attachments.keys())
attachments.remove('data2.dat')
self.assertEqual(len(attachments.keys()), 0)
def clean_up_temporary_attachments(workspace, maxage=1):
"""Garbage collect temporary attachments on a workspace
-- these are used while creating a statusupdate but already
removed when storing the statusupdate.
This method removes any remaining temp attachments.
maxage is age in days beyond which attachments are removed.
"""
temp_attachments = IAttachmentStorage(workspace)
for key in temp_attachments.keys():
keyparts = key.split('-')
datestr = keyparts[1]
try:
date = datetime.strptime(datestr, '%Y%m%d%H%M%S%f')
except ValueError:
date = datetime.min # No proper datestr, treat as old
if datetime.now() - date > timedelta(maxage):
temp_attachments.remove(key)
log.info('Cleaned up temporary attachment {0} from '
'{1}'.format(key, workspace.Title()))
def clean_up_temporary_attachments(workspace, maxage=1):
"""Garbage collect temporary attachments on a workspace
-- these are used while creating a statusupdate but already
removed when storing the statusupdate.
This method removes any remaining temp attachments.
maxage is age in days beyond which attachments are removed.
"""
temp_attachments = IAttachmentStorage(workspace)
for key in temp_attachments.keys():
keyparts = key.split('-')
datestr = keyparts[1]
try:
date = datetime.strptime(datestr, '%Y%m%d%H%M%S%f')
except ValueError:
date = datetime.min # No proper datestr, treat as old
if datetime.now() - date > timedelta(maxage):
temp_attachments.remove(key)
log.info('Cleaned up temporary attachment {0} from '
'{1}'.format(key, workspace.Title()))
