def add_farm(self, **kw):
'''
'''
# import ipdb;ipdb.set_trace()
farm = model.Farm()
# check registed user
user_name = kw.get('user_name', '')
find_result = DBSession.query(
model.Farm).filter_by(user_name=user_name).first()
if find_result:
return USER_EXIST
[
setattr(farm, key, kw[key]) for key in FARM_KEYS
if key in kw and kw[key] != ''
]
try:
DBSession.add(farm)
transaction.commit()
return SUCCESS
except Exception, e:
transaction.abort()
print 'error of add_farm: {}'.format(e)
return UNKNOW_ERROR
def setUp(self):
"""Setup test fixture for each model test method."""
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 add_user_animals(self, **kw):
'''
'''
# import ipdb;ipdb.set_trace()
animals = model.UserAnimals()
[setattr(animals, key, int(kw[key])) for key in USER_ANIMALS_KEYS if key in kw]
try:
DBSession.add(animals)
transaction.commit()
return SUCCESS
except Exception, e:
transaction.abort()
print 'error of add_farm: {}'.format(e)
return UNKNOW_ERROR
def add_user_animals(self, **kw):
'''
'''
# import ipdb;ipdb.set_trace()
animals = model.UserAnimals()
[
setattr(animals, key, int(kw[key])) for key in USER_ANIMALS_KEYS
if key in kw
]
try:
DBSession.add(animals)
transaction.commit()
return SUCCESS
except Exception, e:
transaction.abort()
print 'error of add_farm: {}'.format(e)
return UNKNOW_ERROR
def add_farm(self, **kw):
'''
'''
# import ipdb;ipdb.set_trace()
farm = model.Farm()
# check registed user
user_name = kw.get('user_name', '')
find_result = DBSession.query(model.Farm).filter_by(user_name=user_name).first()
if find_result:
return USER_EXIST
[setattr(farm, key, kw[key]) for key in FARM_KEYS if key in kw and kw[key] != '']
try:
DBSession.add(farm)
transaction.commit()
return SUCCESS
except Exception, e:
transaction.abort()
print 'error of add_farm: {}'.format(e)
return UNKNOW_ERROR
def registe(self, user_name, password, email_address=''):
'''
'''
# import ipdb;ipdb.set_trace()
user = model.User()
user.user_name = user_name
user.password = password
user.email_address = email_address
# check registed user
find_result = DBSession.query(model.User).filter_by(user_name=user_name).first()
if find_result:
return USER_EXIST
try:
DBSession.add(user)
transaction.commit()
return SUCCESS
except Exception, e:
transaction.abort()
print 'error of registe: {}'.format(e)
raise UNKNOW_ERROR
def registe(self, user_name, password, email_address=''):
'''
'''
# import ipdb;ipdb.set_trace()
user = model.User()
user.user_name = user_name
user.password = password
user.email_address = email_address
# check registed user
find_result = DBSession.query(
model.User).filter_by(user_name=user_name).first()
if find_result:
return USER_EXIST
try:
DBSession.add(user)
transaction.commit()
return SUCCESS
except Exception, e:
transaction.abort()
print 'error of registe: {}'.format(e)
raise UNKNOW_ERROR
