bia-fitness/lifting/views.py
2015-04-02 17:58:26 -04:00

31 lines
836 B
Python

import tempfile
from django.shortcuts import render
from django import forms
from django.contrib.auth.decorators import login_required
import os
from . import importers
class FitnotesUploadForm(forms.Form):
file = forms.FileField()
@login_required
def fitnotes_upload(request):
if request.method == 'POST':
form = FitnotesUploadForm(request.POST, request.FILES)
if form.is_valid():
_, fname = tempfile.mkstemp()
with open(fname, 'wb') as tmp:
for chunk in request.FILES['file'].chunks():
tmp.write(chunk)
try:
importers.import_fitnotes_db(fname, request.user)
finally:
os.remove(fname)
else:
form = FitnotesUploadForm()
return render(request, 'upload.html', {'form': form})