Add first_name and last_name

master
Steven Loria 11 years ago
parent 748b6ea83e
commit 43a4083eff
  1. 4
      {{cookiecutter.app_name}}/{{cookiecutter.app_name}}/tests/test_models.py
  2. 2
      {{cookiecutter.app_name}}/{{cookiecutter.app_name}}/tests/webtest_tests.py
  3. 13
      {{cookiecutter.app_name}}/{{cookiecutter.app_name}}/user/models.py

@ -25,5 +25,9 @@ class TestUser(DbTestCase):
assert_true(user.check_password('foobarbaz123')) assert_true(user.check_password('foobarbaz123'))
assert_false(user.check_password("barfoobaz")) assert_false(user.check_password("barfoobaz"))
def test_full_name(self):
user = UserFactory(first_name="Foo", last_name="Bar")
assert_equal(user.full_name, "Foo Bar")
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

@ -27,7 +27,7 @@ class TestLoggingIn(DbTestCase):
form['username'] = self.user.username form['username'] = self.user.username
form['password'] = 'myprecious' form['password'] = 'myprecious'
# Submits # Submits
res = form.submit().follow() res = form.submit().maybe_follow()
assert_equal(res.status_code, 200) assert_equal(res.status_code, 200)
def test_sees_error_message_if_password_is_incorrect(self): def test_sees_error_message_if_password_is_incorrect(self):

@ -3,8 +3,8 @@ import datetime as dt
from flask.ext.login import UserMixin from flask.ext.login import UserMixin
from {{cookiecutter.app_name}}.database import db, CRUDMixin from myflaskapp.database import db, CRUDMixin
from {{cookiecutter.app_name}}.extensions import bcrypt from myflaskapp.extensions import bcrypt
class User(UserMixin, CRUDMixin, db.Model): class User(UserMixin, CRUDMixin, db.Model):
@ -14,10 +14,13 @@ class User(UserMixin, CRUDMixin, db.Model):
email = db.Column(db.String(80), unique=True, nullable=False) email = db.Column(db.String(80), unique=True, nullable=False)
password = db.Column(db.String, nullable=False) # The hashed password password = db.Column(db.String, nullable=False) # The hashed password
created_at = db.Column(db.DateTime(), nullable=False) created_at = db.Column(db.DateTime(), nullable=False)
first_name = db.Column(db.String(30), nullable=True)
last_name = db.Column(db.String(30), nullable=True)
active = db.Column(db.Boolean()) active = db.Column(db.Boolean())
is_admin = db.Column(db.Boolean()) is_admin = db.Column(db.Boolean())
def __init__(self, username=None, email=None, password=None, def __init__(self, username=None, email=None, password=None,
first_name=None, last_name=None,
active=False, is_admin=False): active=False, is_admin=False):
self.username = username self.username = username
self.email = email self.email = email
@ -26,6 +29,8 @@ class User(UserMixin, CRUDMixin, db.Model):
self.active = active self.active = active
self.is_admin = is_admin self.is_admin = is_admin
self.created_at = dt.datetime.utcnow() self.created_at = dt.datetime.utcnow()
self.first_name = first_name
self.last_name = last_name
def set_password(self, password): def set_password(self, password):
self.password = bcrypt.generate_password_hash(password) self.password = bcrypt.generate_password_hash(password)
@ -33,5 +38,9 @@ class User(UserMixin, CRUDMixin, db.Model):
def check_password(self, password): def check_password(self, password):
return bcrypt.check_password_hash(self.password, password) return bcrypt.check_password_hash(self.password, password)
@property
def full_name(self):
return "{0} {1}".format(self.first_name, self.last_name)
def __repr__(self): def __repr__(self):
return '<User "{username}">'.format(username=self.username) return '<User "{username}">'.format(username=self.username)

Loading…
Cancel
Save