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

Comments

Please login first before commenting.