Login

another UserForeignKey

Author:
diverman
Posted:
July 12, 2009
Language:
Python
Version:
1.0
Tags:
foreignkey model user field users user-foreign-key
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. "Magic Link" Management Command by webology 1 month, 3 weeks ago
  2. Closest ORM models to a latitude/longitude point by simonw 1 month, 3 weeks ago
  3. Log the time taken to execute each DB query by kennyx46 1 month, 3 weeks ago
  4. django database snippet by ItsRLuo 2 months ago
  5. Serialize a model instance by chriswedgwood 2 months, 4 weeks ago

Comments

Please login first before commenting.