def create_diary_pic(up_file, entry):
"""
Given an uploaded file and a diary entry, it creates a DiaryPicture
belonging to an entry.
"""
diary_pic = DiaryPicture(entry=entry)
img = File(up_file)
diary_pic.image.save(up_file.name, img, save=False)
old_path = diary_pic.image.path
if os.path.exists(old_path):
resized = resize_image(diary_pic.image.path,
size=settings.DIARY_IMAGE_SIZE)
os.unlink(old_path)
diary_pic.image.save(diary_pic.image.name, resized, save=False)
diary_pic.thumbnail.save(up_file.name, img, save=False)
diary_pic.save()
# Generate thumbnail off thread
generate_entry_thumbnail.delay(diary_pic.pk)
return diary_pic
def setUp(self):
super(DeleteDraftsCronTests, self).setUp()
self.c = get_user(username='******')
grace_days = (datetime.now() -
timedelta(days=settings.DIARY_GRACE_DAYS + 1))
self.draft = DiaryEntry(creator=self.c, created=grace_days,
is_draft=True)
self.draft.save()
self.draft_pk = self.draft.pk
draft_pic = DiaryPicture(entry=self.draft)
draft_pic2 = DiaryPicture(entry=self.draft)
self.entry = DiaryEntry(creator=self.c, is_draft=False)
self.entry.save()
self.entry_pk = self.entry.pk
entry_pic = DiaryPicture(entry=self.entry)
entry_pic2 = DiaryPicture(entry=self.entry)
with open('apps/diary/tests/media/django.jpg') as f:
img = File(f)
draft_pic.image.save('cowboy', img, save=False)
draft_pic.thumbnail.save('cowboy', img, save=False)
entry_pic.image.save('cowboy_new', img, save=False)
entry_pic.thumbnail.save('cowboy_new', img, save=False)
draft_pic.save()
entry_pic.save()
with open('apps/diary/tests/media/hawk.jpg') as f:
img = File(f)
draft_pic2.image.save('hawk', img, save=False)
draft_pic2.thumbnail.save('hawk', img, save=False)
entry_pic2.image.save('hawk_new', img, save=False)
entry_pic2.thumbnail.save('hawk_new', img, save=False)
draft_pic2.save()
entry_pic2.save()
