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. codigo alto nivel by MrRocklion 6 days, 2 hours ago
  2. Load template from specific app by Krzysiek555 3 weeks, 5 days ago
  3. PostgreSQL JSON subqueries by dolamroth 4 weeks ago
  4. "Magic Link" Management Command by webology 6 months ago
  5. Closest ORM models to a latitude/longitude point by simonw 6 months ago

Comments

Please login first before commenting.