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. Template tag - list punctuation for a list of items by shapiromatron 11 months, 2 weeks ago
  2. JSONRequestMiddleware adds a .json() method to your HttpRequests by cdcarter 11 months, 2 weeks ago
  3. Serializer factory with Django Rest Framework by julio 1 year, 6 months ago
  4. Image compression before saving the new model / work with JPG, PNG by Schleidens 1 year, 7 months ago
  5. Help text hyperlinks by sa2812 1 year, 7 months ago

Comments

Please login first before commenting.