def save(self):
request = self.context['request']
upload = self.validated_data['upload']
self.instance = Attachment(
secret=Attachment.generate_new_secret(),
filetype=self.filetype,
size=upload.size,
uploader=request.user,
uploader_name=request.user.username,
uploader_slug=request.user.slug,
uploader_ip=request.user_ip,
filename=upload.name,
)
if self.is_upload_image(upload):
try:
self.instance.set_image(upload)
except IOError:
raise serializers.ValidationError({
'upload': [_("Uploaded image was corrupted or invalid.")],
})
else:
self.instance.set_file(upload)
self.instance.save()
return self.instance
def create_attachment(self, request):
upload = request.FILES.get('upload')
if not upload:
raise UploadError(_("No file has been uploaded."))
user_roles = set(r.pk for r in request.user.get_roles())
filetype = validate_filetype(upload, user_roles)
validate_filesize(upload, filetype, request.user.acl_cache['max_attachment_size'])
attachment = Attachment(
secret=Attachment.generate_new_secret(),
filetype=filetype,
size=upload.size,
uploader=request.user,
uploader_name=request.user.username,
uploader_slug=request.user.slug,
uploader_ip=request.user_ip,
filename=upload.name,
)
if is_upload_image(upload):
try:
attachment.set_image(upload)
except IOError:
raise UploadError(_("Uploaded image was corrupted or invalid."))
else:
attachment.set_file(upload)
attachment.save()
add_acl(request.user, attachment)
return Response(AttachmentSerializer(attachment, context={'user': request.user}).data)
def move_attachments(stdout, style):
query = 'SELECT * FROM misago_attachment ORDER BY id'
posts = []
attachment_types = {}
for attachment_type in AttachmentType.objects.all():
for mimetype in attachment_type.mimetypes_list:
attachment_types[mimetype] = attachment_type
for attachment in fetch_assoc(query):
if attachment['content_type'] not in attachment_types:
stdout.write(
style.WARNING("Skipping attachment: %s (invalid type)" %
attachment['name']))
continue
if not attachment['post_id']:
stdout.write(
style.WARNING("Skipping attachment: %s (orphaned)" %
attachment['name']))
continue
filetype = attachment_types[attachment['content_type']]
post_pk = movedids.get('post', attachment['post_id'])
post = Post.objects.get(pk=post_pk)
if post_pk not in posts:
posts.append(post_pk)
uploader = None
if attachment['user_id']:
uploader_pk = movedids.get('user', attachment['user_id'])
uploader = UserModel.objects.get(pk=uploader_pk)
file_path = os.path.join(OLD_FORUM['ATTACHMENTS'], attachment['path'])
upload = OldAttachmentFile(open(file_path, 'rb'), attachment)
new_attachment = Attachment(
secret=Attachment.generate_new_secret(),
filetype=filetype,
post=post,
uploaded_on=localise_datetime(attachment['date']),
uploader=uploader,
uploader_name=attachment['user_name'],
uploader_slug=attachment['user_name_slug'],
uploader_ip=attachment['ip'],
filename=attachment['name'],
size=attachment['size'],
)
if attachment['content_type'] in IMAGE_TYPES:
new_attachment.set_image(upload)
else:
new_attachment.set_file(upload)
new_attachment.save()
movedids.set('attachment', attachment['id'], new_attachment.pk)
update_attachments_caches(posts)