Пример #1
0
def upload_doc():
    if 'doc' not in request.files:
        # flash('no doc part in request')
        session.close()
        print('no doc part in request')
        return redirect(request.url)
    photo = request.files['doc']
    obj = upload_file(photo, 'DOC')
    if obj['status'] == 'BAD REQUEST':
        print(obj['message'])
        session.close()
        return obj
    if obj['status'] == 'OK':
        print(obj['message'])
        filename = obj['filename']
        info = UploadedFile(filename=filename, file_type='DOC')
        session.add(info)
        try:
            session.commit()
        except:
            sesssion.rollback()
            flash(config.UNEXPECTED_ERROR)
        session.close()
        return redirect('/dashboard')
    session.close()
    return {
        'status':'ERROR',
        'message':'UNPREDICTED ERROR OCCURRED'
    }
Пример #2
0
def handle_uploaded_file(f, user):
    username = user.username
    os.system('mkdir %(user)s'%{'user':username}) 
    destination = open('%(user)s/%(name)s'%{'user':username, 'name':f.name},'wb+')
    for chunk in f.chunks():
        destination.write(chunk)
    destination.close()
    for old in UploadedFile.objects.filter(name = f.name):
        old.delete()
    uFile = UploadedFile(name = f.name, username = username, local_path = '%(user)s/%(name)s'%{'user':username, 'name':f.name})
    uFile.save()
    #prof = Profile.objects.get(user = user)
    acc = Account.objects.get(user = user)
    acc.files.add(uFile)
    acc.save()
    return uFile
Пример #3
0
def upload_image():
    if 'photo' not in request.files:
        # flash('no photo part in request')
        print('no photo part in request')
        session.close()
        return redirect(request.url)
    photo = request.files['photo']
    obj = upload_file(photo, 'IMG')
    if obj['status'] == 'BAD REQUEST':
        print(obj['message'])
        return obj
    if obj['status'] == 'OK':
        print(obj['message'])
        filename = obj['filename']
        info = UploadedFile(filename=filename, file_type='IMAGE')
        session.add(info)
        session.commit()
        session.close()
        return redirect('/dashboard')
        # return {
        #     'status':'OK',
        #     'message':'SUCCESSFULLY ADDED IMAGE'
        # }, 200
    session.close()
    return {
        'status':'ERROR',
        'message':'UNPREDICTED ERROR OCCURRED'
    }
 def test_transmute(self):
     
     doctype = DocumentType.objects.create(name='transmuted')
     doc = Document(
         document_type = doctype,
     )
     doc.save()
     uf = UploadedFile.transmute_document(
         doc,
         uploader=self.test_user,
     )
     uf.save()
     self.assertEqual(uf.document_ptr_id, doc.id)
     self.assertEqual(uf.document_ptr, doc)
     self.assertEqual(uf.document_type, doc.document_type)
     self.assertEqual(doc.uploadedfile, uf)
     
     self.assertRaises(ValueError, UploadedFile.transmute_document, doc, uploader=self.test_user)
Пример #5
0
def upload_doc():
    if request.method == 'POST':
        if 'doc' not in request.files:
            # flash('no doc part in request')
            print('no doc part in request')
            return redirect(request.url)
        photo = request.files['doc']
        obj = upload_file(photo, 'DOC')
        if obj['status'] == 'BAD REQUEST':
            print(obj['message'])
            return obj
        if obj['status'] == 'OK':
            print(obj['message'])
            filename = obj['filename']
            info = UploadedFile(filename=filename,
                                event_id=request.data['event_id'])
            session.add(info)
            session.commit()
            return {'status': 'OK', 'message': 'SUCCESSFULLY ADDED DOC'}, 200
        return {'status': 'ERROR', 'message': 'UNPREDICTED ERROR OCCURRED'}
    else:
        return render_template('upload-file.html',
                               filetype='doc',
                               url=request.url)
    def test(self):
        filename = 'up-' + rand_string()
        contents = 'upladded contestsn'
    
        a = UploadedFile(
            uploader = self.test_user,
        )
        a.uploaded_file.save(filename, ContentFile(contents))
        a.clean()
        a.save()

        # TODO self.assertEqual(a.url,?)
        # TODO self.assertEqual(a.path,?)

        self.assertEqual(a.specific_instance(), a)
        self.assertEqual(a.document_ptr.__class__, Document)
        self.assertNotEqual(a.document_ptr.__class__, UploadedFile)
        self.assertEqual(a.document_ptr.specific_instance(), a)
        self.assertEqual(a.document_ptr.specific_instance().__class__, UploadedFile)
def add_document(request, documentable_id):
    
    obj = Documentable.objects.get(id=documentable_id).specific_instance()
    
    # even though only one subclass of Document is listed here, this view is
    # written with adding new ones in mind
    form_classes = {
        'document': NewDocumentForm,
        'uploaded_file': NewUploadedFileForm,
    }

    form_kwargs = {}
    for name in form_classes.keys():
        form_kwargs[name] = {
            'prefix': name,
        }
    
    forms = {}
    
    if request.method == 'POST':
        # we have to instantiate the 'document' form in order to know which other
        # forms to bind
        form_kwargs['document']['data'] = request.POST
        forms['document'] = form_classes['document'](**form_kwargs['document'])
        # we only want to bind the forms the user acutally filled in
        if forms['document'].is_valid():
            is_upload = forms['document'].cleaned_data['is_uploadedfile']
            if is_upload:
                form_kwargs['uploaded_file']['data'] = request.POST
                form_kwargs['uploaded_file']['files'] = request.FILES
        
        
        # now instantiate the forms besides the 'document' form
        for name in filter(lambda fn: fn != 'document', form_classes.keys()):
            forms[name] = form_classes[name](**form_kwargs[name])
        
        if forms['document'].is_valid() and (not is_upload or (is_upload and forms['uploaded_file'].is_valid())):
            
            document = forms['document'].save(commit=False)
            document.attached_to = obj
            document.save()
            forms['document'].save_m2m()

            if is_upload:
                uploaded_file = UploadedFile.transmute_document(document, **forms['uploaded_file'].cleaned_data)
                uploaded_file.uploader = request.user
                uploaded_file.clean()
                uploaded_file.save()
                
            return redirect(obj)
    else: # request.method != 'POST'
        for name in form_classes.keys():
            forms[name] = form_classes[name](**form_kwargs[name])
    
    # the default template uses the radiohider script with forms['models']
    # select widget. TODO currently no way to pass in a custom Media to go with 
    # a custom template.
    template_media = Media(
        js= (settings.JQUERY_FILE, 'checkboxhider.js'),
    )
    media = reduce( lambda m, f: m + f.media, forms.values(), template_media)

    return render_to_response(
        'documents/add_attachment.html',
        {
            'object': obj,
            'forms': forms,
            'media': media,
        },
        context_instance= RequestContext(request),
    )