def get_user_or_default_user(uid=None):
email = config.get('default_user_email', '*****@*****.**')
if not uid:
user = DBSession.query(User).filter(User.email_address==email).first()
else:
try:
user = DBSession.query(User).filter(User.user_id==uid).first()
except:
user = DBSession.query(User).filter(User.email_address==email).first()
if not user:
user = DBSession.query(User).filter(User.email_address==email).first()
return user
def get_user_or_default_user(uid=None):
email = config.get('default_user_email', '*****@*****.**')
if not uid:
user = DBSession.query(User).filter(
User.email_address == email).first()
else:
try:
user = DBSession.query(User).filter(User.user_id == uid).first()
except:
user = DBSession.query(User).filter(
User.email_address == email).first()
if not user:
user = DBSession.query(User).filter(
User.email_address == email).first()
return user
def setUp(self):
"""Prepare model test fixture."""
try:
new_attrs = {}
new_attrs.update(self.attrs)
new_attrs.update(self.do_get_dependencies())
self.obj = self.klass(**new_attrs)
DBSession.add(self.obj)
DBSession.flush()
return self.obj
except:
DBSession.rollback()
raise
def setUp(self):
"""Prepare model test fixture."""
try:
new_attrs = {}
new_attrs.update(self.attrs)
new_attrs.update(self.do_get_dependencies())
self.obj = self.klass(**new_attrs)
DBSession.add(self.obj)
DBSession.flush()
return self.obj
except:
DBSession.rollback()
raise
def test_query_obj(self):
"""Model objects can be queried"""
obj = DBSession.query(self.klass).one()
for key, value in self.attrs.iteritems():
assert_equals(getattr(obj, key), value)
def tearDown(self):
"""Finish model test fixture."""
DBSession.rollback()
def test_query_obj(self):
"""Model objects can be queried"""
obj = DBSession.query(self.klass).one()
for key, value in self.attrs.iteritems():
assert_equals(getattr(obj, key), value)
def tearDown(self):
"""Finish model test fixture."""
DBSession.rollback()
def by_user_name(cls, username):
"""Return the user object whose user name is ``username``."""
return DBSession.query(cls).filter_by(user_name=username).first()
def by_email_address(cls, email):
"""Return the user object whose email address is ``email``."""
return DBSession.query(cls).filter_by(email_address=email).first()
