Give User a status, also add uploader user field to MediaEntry

This commit is contained in:
Christopher Allan Webber 2011-04-23 08:52:23 -05:00
parent 7eba0306d8
commit 4d75522b91

View File

@ -21,10 +21,40 @@ from mongokit import Document, Set
from mediagoblin.auth import lib as auth_lib
class User(Document):
__collection__ = 'users'
structure = {
'username': unicode,
'email': unicode,
'created': datetime.datetime,
'plugin_data': dict, # plugins can dump stuff here.
'pw_hash': unicode,
'email_verified': bool,
'status': unicode,
}
required_fields = ['username', 'created', 'pw_hash', 'email']
default_values = {
'created': datetime.datetime.utcnow,
'email_verified': False,
# TODO: shouldn't be active by default, must have email registration
'status': 'active'}
def check_login(self, password):
"""
See if a user can login with this password
"""
return auth_lib.bcrypt_check_password(
password, self['pw_hash'])
class MediaEntry(Document):
__collection__ = 'media_entries'
structure = {
'uploader': User,
'title': unicode,
'created': datetime.datetime,
'description': unicode,
@ -45,31 +75,6 @@ class MediaEntry(Document):
def main_mediafile(self):
pass
class User(Document):
__collection__ = 'users'
structure = {
'username': unicode,
'email': unicode,
'created': datetime.datetime,
'plugin_data': dict, # plugins can dump stuff here.
'pw_hash': unicode,
'email_verified': bool,
}
required_fields = ['username', 'created', 'pw_hash', 'email']
default_values = {
'created': datetime.datetime.utcnow,
'email_verified': False}
def check_login(self, password):
"""
See if a user can login with this password
"""
return auth_lib.bcrypt_check_password(
password, self['pw_hash'])
REGISTER_MODELS = [MediaEntry, User]