Mercurial > public > sg101
comparison gpp/accounts/models.py @ 1:dbd703f7d63a
Initial import of sg101 stuff from private repository.
author | gremmie |
---|---|
date | Mon, 06 Apr 2009 02:43:12 +0000 |
parents | |
children | 75ea1a8be7f2 |
comparison
equal
deleted
inserted
replaced
0:900ba3c7b765 | 1:dbd703f7d63a |
---|---|
1 """Contains models for the accounts application""" | |
2 | |
3 import datetime | |
4 import random | |
5 import string | |
6 import hashlib | |
7 import base64 | |
8 | |
9 from django.db import models | |
10 from django.contrib import auth | |
11 from django.conf import settings | |
12 | |
13 | |
14 class IllegalUsername(models.Model): | |
15 """model to represent the list of illegal usernames""" | |
16 username = models.CharField(max_length = 30, db_index = True) | |
17 | |
18 def __unicode__(self): | |
19 return self.username | |
20 | |
21 class Meta: | |
22 ordering = ('username', ) | |
23 | |
24 | |
25 class IllegalEmail(models.Model): | |
26 """model to represent the list of illegal/restricted email addresses""" | |
27 email = models.EmailField(db_index = True) | |
28 | |
29 def __unicode__(self): | |
30 return self.email | |
31 | |
32 class Meta: | |
33 ordering = ('email', ) | |
34 | |
35 | |
36 class PendingUserManager(models.Manager): | |
37 """user manager for PendingUser model""" | |
38 | |
39 create_count = 0 | |
40 | |
41 def create_pending_user(self, username, email, password): | |
42 '''creates a new pending user and saves it to the database''' | |
43 | |
44 temp_user = auth.models.User() | |
45 temp_user.set_password(password) | |
46 | |
47 now = datetime.datetime.now() | |
48 pending_user = self.model(None, | |
49 username, | |
50 email, | |
51 temp_user.password, | |
52 now, | |
53 self._make_key()) | |
54 | |
55 pending_user.save() | |
56 self.create_count += 1 | |
57 return pending_user | |
58 | |
59 | |
60 def purge_expired(self): | |
61 expire_time = datetime.datetime.now() - datetime.timedelta(days = 1) | |
62 expired_pending_users = self.filter(date_joined__lt = expire_time) | |
63 expired_pending_users.delete() | |
64 | |
65 | |
66 def _make_key(self): | |
67 s = ''.join(random.sample(string.printable, 8)) | |
68 delta = datetime.date.today() - datetime.date(1846, 12, 28) | |
69 days = base64.urlsafe_b64encode(str(delta * 10)) | |
70 key = hashlib.sha1(settings.SECRET_KEY + | |
71 unicode(self.create_count) + | |
72 unicode(s) + | |
73 unicode(days)).hexdigest()[::2] | |
74 return key | |
75 | |
76 | |
77 class PendingUser(models.Model): | |
78 """model for holding users while they go through the email registration cycle""" | |
79 | |
80 username = models.CharField(max_length = 30, db_index = True) | |
81 email = models.EmailField() | |
82 password = models.CharField(max_length = 128) | |
83 date_joined = models.DateTimeField(default = datetime.datetime.now, db_index = True) | |
84 key = models.CharField(max_length = 20, editable = True) | |
85 | |
86 objects = PendingUserManager() | |
87 | |
88 def __unicode__(self): | |
89 return self.username | |
90 |