CODEWITHSUNDEEP
django
Praful Bagai
Praful Bagai

Reputation: 17382

Django uploading files for user

I am able to upload the files in django. I have also managed to register different users. But when the users log in , they all see the set of files that were uploaded by anyone. instead I want the user views only those set of files which he/she has uploaded.

Here is my models.py 

class Document(models.Model):
    docfile = models.FileField(upload_to='documents/%Y/%m/%d')

views.py

def list(request):

    if request.method == 'POST':
        form = DocumentForm(request.POST, request.FILES)
        if form.is_valid():
            newdoc = Document(docfile = request.FILES['docfile'])
            newdoc.save()

            # Redirect to the document list after POST
            return HttpResponse("File Uploaded.. Kindly Refresh!!")
    else:
        form = DocumentForm() # A empty, unbound form

    # Load documents for the list page
    documents = Document.objects.all()

    # Render list page with the documents and the form
    return render_to_response(
        'login/list.html',
        {'documents': documents, 'form': form},
        context_instance=RequestContext(request)
    )

LIST.html

 <table cellpadding=20>
    {% for document in documents %}
        <tr>
            <td><a href="{{ document.docfile.url }}">{{ document.docfile.name }}</a> </td>
            <td></td>
            <td></td>
            <td><a href="delete/{{ document.docfile.name }}"> Delete</a></td>

        </tr>
    {% endfor %}
    </table>

  {% else %}
    <p>No documents.</p>
  {% endif %}

Upvotes: 1

Views: 2732

Answers (1)

boltsfrombluesky
boltsfrombluesky

Reputation: 422

change models.py to this 

from django.contrib.auth.models import User
class Document(models.Model):
    docfile = models.FileField(upload_to='documents/%Y/%m/%d')
    user = models.ForeignKey(User)

change views.py to this

def list(request):
    if request.method == 'POST':
        form = DocumentForm(request.POST, request.FILES)
        if form.is_valid():
            newdoc = Document(docfile = request.FILES['docfile'])
            newdoc.user = request.user
            newdoc.save()

            # Redirect to the document list after POST
            return HttpResponse("File Uploaded.. Kindly Refresh!!")
    else:
        form = DocumentForm() # A empty, unbound form

    # Load documents for the list page
    documents = Document.objects.filter(user=request.user)

    # Render list page with the documents and the form
    return render_to_response(
        'login/list.html',
        {'documents': documents, 'form': form},
        context_instance=RequestContext(request)
    )

Upvotes: 4

Related Questions