def club_pictures(rowkey):
if not check_auth(request):
abort(403)
if request.method == 'POST':
# check if the post request has the file part
if 'file' not in request.files:
return make_response('No file part', 400)
file = request.files['file']
# if user does not select file, browser also
# submit a empty part without filename
if file.filename == '':
return make_response('No file part', 400)
if not allowed_file(file.filename):
return make_response('The filetype is not allowed')
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
filename = rowkey + "." + filename.rsplit('.', 1)[1]
file.save(os.path.join(app.config['CLUB_PICTURE_UPLOAD_FOLDER'], filename))
db_picture = db_session.query(UserFile).filter_by(owner=rowkey, file_type='ProfilePicture')
if len(list(db_picture)) == 0:
db_picture = UserFile()
db_picture.file_type = 'ProfilePicture'
db_picture.owner = rowkey
db_session.add(db_picture)
db_session.commit()
return make_response("",200)
db_picture.update({"file_name":filename})
db_session.commit()
return make_response("", 200)
if request.method == 'GET':
filename = db_session.query(UserFile).filter_by(file_type='ProfilePicture', owner=rowkey)[0].file_name
return jsonify({"filename": str.replace(app.config['CLUB_PICTURE_UPLOAD_FOLDER'], "static\\Sloach\\", "") + "/" + filename})
def show_home():
blah = UserFile(None)
return render_template(
'index.html',
user_file=blah,
signed_policy=blah.sign_policy(app.config['S3_SECRET_KEY']),
access_key=app.config['S3_ACCESS_KEY'])
def upload (request):
currentPage = "upload"
auth = request.user.is_authenticated()
if (request.method=="POST"): # If the form has been submitted...
filename = request.FILES['file'].name
if (filename.endswith('.pdf')): # allow users to only upload pdf files
form = uploadfileform(request.POST, request.FILES) # A form bound to the POST data
if form.is_valid(): # All validation rules pass
data = form.cleaned_data
# generate default report title if not specified by user
if (data['title'] == ''):
data['title'] = "Report for " + filename
unid = unicode(uuid.uuid4())
filename = unid + ".pdf"
filepath = MEDIA_ROOT + "public/"
# if user has logged in, save uploaded file to user's folder
if auth:
filepath = request.user.get_profile().filepath
fileObj = UserFile(uid = unid, owner = request.user.username, title = data['title'], notes = data['notes'])
fileObj.file.save(filename, ContentFile(request.FILES['file'].read()))
fileObj.save()
process_file(filename, filepath) # call function to run evaluator on uploaded file
if auth:
user = request.user.get_profile()
user.filecount = user.filecount + 1
user.save()
return HttpResponseRedirect('/accounts/profile/managereports/')
else:
return render_to_response("upload/processing.html", {"uid": fileObj.uid})
else:
message = "Not a PDF file"
form = uploadfileform()
return render_to_response("upload/fileupload.htm", locals(), context_instance=RequestContext(request))
def upload():
msg = None
form = UploadFileForm()
if form.validate_on_submit():
try:
file = form.file.data
file_name = file_set.save(file)
UserFile.add(g.user.id, file_name)
return redirect(url_for('userfile.list'))
except Exception as e:
msg = '上传失败，请检查文件格式'
return render_template('userfile/uplaod.html',msg=msg, form=form)
def attach(request):
"Obtains the attachment and saves it to the disk"
if request.method == 'POST':
form = AttachmentForm(request.POST, request.FILES)
if form.is_valid():
f = request.FILES['attachment']
record = UserFile(name = f.name,
attachment = os.path.join(settings.ATTACHMENT_FOLDER, form.cleaned_data['attachment'].name),
mime_type = f.content_type)
handle_uploaded_file(f)
record.save()
return HttpResponseRedirect('/filemanager/list/')
else:
form = AttachmentForm()
return render_to_response('attach.html', {'form':form},
context_instance = RequestContext(request))
def upload(request):
""" Upload View """
t = loader.get_template('upload.html')
c = Context({
'files': UserFile.objects.all()
})
if request.POST:
uf = UserFile(user_file=request.FILES['userfile'], absolute_path=request.POST.get('absolute_path', ''),
content_type=request.FILES['userfile'].content_type)
uf.save()
if not uf.id:
raise Exception('Not saved!')
return HttpResponse(dumps({'name': uf.user_file.name, 'url': uf.user_file.url}), mimetype='application/json')
return HttpResponse(t.render(c))
def register(request):
if request.method == 'POST':
uf = UserForm(request.POST)
if uf.is_valid():
username = uf.cleaned_data['username']
password = uf.cleaned_data['password']
email = uf.cleaned_data['email']
user = UserFile()
user.username = username
user.password = password
user.email = email
user.save()
return render_to_response('success.html',{'username':username})
else:
uf = UserForm()
return render_to_response('register.html',{'uf':uf})
def list():
file_list = UserFile.find_by_userid(g.user.id)
return render_template('userfile/list.html', list=file_list)
