Skip to main content

A reusable app to graceously handle files that your users might upload, such as profile pictures, foto albums etc...

Project description

Almost all modern web apps allow their users to upload content such as audio, video or images. This raises a number of issues if that content should not be visible to the whole world by default.

If you have a UserProfile model and add an ImageField to it, you need to come up with a good idea on how to save those images. It is probably not a good idea to keep the original filenames as they might disturb your server’s file system and open doors for hackers, who might try to brute-force against your /media/user_profiles/ in the hope to steal some valuable files.

Since it seems inevitable to implement a function for Django’s FileField’s upload_to attribute I thought that this might be a candidate for a reusable app.

This project is experimental. We are using it on two completely different live projects and will hopefully come up with an implementation that is so generic that it can safely be used by anyone.

Since we are dealing with files here and not only with a database, backwards incompatible changes might turn out to be a pain in the ass to deploy on your production sites. You have been warned.

Prerequisites

You need at least the following packages in your virtualenv:

  • Django 1.4

  • South

TODO: Test if this really is all we need

Installation

To get the latest stable release from PyPi:

$ pip install django-user-media

To get the latest commit from GitHub:

$ pip install -e git://github.com/bitmazk/django-user-media.git#egg=user_media

Add the app to your INSTALLED_APPS:

INSTALLED_APPS = [
    ...
    'user_media',
]

Hook the app into your main urls.py:

urlpatterns += patterns('',
    ...
    url(r'umedia/', include('user_media.urls')),
)

Run the south migrations to create the app’s database tables:

$ ./manage.py migrate user_media

Usage

Let’s assume that you have a UserProfile model and you want to add an avatar field to that model.

First you might want to add a GenericRelation to your UserProfile model:

from django.contrib.contenttypes import generic

class UserProfile(models.Model):
    ...
    user = models.ForeignKey('auth.User')

    avatar = generic.GenericRelation(
        'user_media.UserMediaImage',
    )

Now you will be able to get all uploaded images that belong to a UserProfile by doing this:

profile = UserProfile.objects.get(pk=1)
images = profile.avatar.filter(user=profile.user)

It makes sense to add a convenience method to your UserProfile model:

class UserProfile(models.Model):
    ...
    def get_avatar(self):
        return self.avatar.filter(user=self.user)

In your templates you can now provide a link to the image creation view like this (assuming that your UserProfile object is called object in the template’s context):

{% load url from future %}
<a href="{% url "user_media_create_image" content_type="userprofile" object_id=object.pk %}">Upload your picture</a>

Note that userprofile is the model name that the ContentType of your UserProfile model would return. You can figure this out with ./manage.py shell for example:

$ ./manage.py shell
In [1]: from django.contrib.contenttypes.models import ContentType
In [2]: from your_app.models import UserProfile
In [3]: ContentType.objects.get_for_model(UserProfile).model
Out [1]: u'userprofile'

When visiting that link, the user should see an image upload form. You might want to override that template (user_media/usermediaimage_form.html). After uploading the image the view should redirect back to the absolute url of your UserProfile. If you want to redirect to another URL, you can provide a next URL parameter via POST or GET:

<a href="{% url "user_media_create_image" content_type="userprofile" object_id=object.pk %}?next=/foo/bar">Upload your picture</a>

Now you should have all building blocks that you need to add links or buttons to your templates that call the views of this application. On your UserProfile detail view you could display the avatar, if available:

{% if object.get_avatar %}
    <img src="{{ MEDIA_URL }}{{ object.get_avatar.0.image }}" />
{% endif %}

Or in your UserProfile update view you could display a link to upload a new image or to delete the existing image:

{% if form.instance.get_avatar %}
    <p><img src="{{ MEDIA_URL }}{{ form.instance.get_avatar.0.image }}" /></p>
    <a href="{% url "user_media_image_delete" pk=form.instance.get_avatar.0.pk %}">Delete picture</a>
{% else %}
    <a href="{% url "user_media_image_create" content_type="userprofile" object_id=form.instance.pk %}">Add profile picture</a>
{% endif %}

The delete link in this example will render the user_media/usermediaimage_confirm_delete.html template, which you might want to override in your project.

Contribute

If you want to contribute to this project, please perform the following steps:

# Fork this repository
# Clone your fork
$ mkvirtualenv -p python2.7 django-user-media
$ pip install -r requirements.txt

$ git co -b feature_branch master
# Implement your feature and tests
$ git add . && git commit
$ git push -u origin feature_branch
# Send us a pull request for your feature branch

Testing

If you want to contribute to this project you can run the tests without setting up a Django project. Just clone this repository and execute the runtests.py:

$ ./user_media/tests/runtests.py

Sometimes a new feature needs new South migrations, in this case you should do the following:

$ rm db.sqlite
$ ./manage.py syncdb --migrate
$ ./manage.py schemamigration user_media --auto

Discuss

If you have questions or issues, please open an issue on GitHub.

If we don’t react quickly, please don’t hesitate to ping me on Twitter (@mbrochh)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

django-user-media-0.1.tar.gz (209.1 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page