Login

another UserForeignKey

Author:
diverman
Posted:
July 12, 2009
Language:
Python
Version:
1.0
Score:
2 (after 2 ratings)

This is another foreign key to User model. User is automatically associated before save.

Requires: ThreadlocalsMiddleware

Inspired by: snippet 509

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
class UserForeignKey(models.ForeignKey):
    def __init__(self, **kwargs):
        kwargs['editable'] = False
        to = 'auth.user'

        super(UserForeignKey, self).__init__(to, **kwargs)

    def pre_save(self, model_instance, add):
        def set_current_user():
            setattr(model_instance, self.name, get_current_user())
            return super(UserForeignKey, self).pre_save(model_instance, add)

        return super(UserForeignKey, self).pre_save(model_instance, add) or set_current_user()

More like this

  1. Serializer factory with Django Rest Framework by julio 3 months, 1 week ago
  2. Image compression before saving the new model / work with JPG, PNG by Schleidens 3 months, 4 weeks ago
  3. Help text hyperlinks by sa2812 4 months, 3 weeks ago
  4. Stuff by NixonDash 7 months ago
  5. Add custom fields to the built-in Group model by jmoppel 9 months ago

Comments

Please login first before commenting.