def login(self, cell_phone_num):
"""
这个方法适用于用户已经重手机端登出，
APP需要用户提供预设密码
先看Redis中是否有对应的对象，
如果有，直接从Redis中
返回用户的信息，并生成新的SessionId，更新redis数据
如果没有，从MySQL中提取数据，
并更新redis的数据
返回：用户的个人信息
"""
old_session_id = self.request.headers[Constant.AUTHORIZATION]
redis_dao = user_redis_dao.UserRedisDAO()
member_id = redis_dao.get_member_id_by_cell_phone(cell_phone_num)
if member_id and old_session_id:
auth_result = redis_dao.auth_session_id(member_id, old_session_id)
elif old_session_id == "" and member_id:
return redis_dao.login(cell_phone_num)
if auth_result:
return redis_dao.login(cell_phone_num)
else:
return_result = {}
return_result["member_id"] = ""
return_result["session_id"] = ""
return return_result
def login_directly(self, cell_phone_num, session_id):
"""
这个方法适用于用户没有从APP中登出，在APP中存在一个
上次登录时的SessionID
登录逻辑同login()方法基本相同。
唯一的不同，就是不需要验证用户的密码，而是去验证用户
上次登录时生成的SessionId
返回：用户的个人信息
"""
try:
isEnrol = user_redis_dao.UserRedisDAO.login(cell_phone_num)
if isEnrol:
redis_dao = user_redis_dao.UserRedisDAO()
new_session_id = utils.SessionGenerator.session_generate()
user = redis_dao.get_user_by_phone_number(cell_phone_num)
if session_id == user.get_session_id(
): # 用户终端存在sessionId，没有”登出“
user.set_session_id(new_session_id)
user_redis_dao.UserRedisDAO.update_user(
user) # update in redis
# --------------------------------------
# TODO(JIAS): 更新MySQL中用户的sessionId
# --------------------------------------
return user # 返回用户对象
else:
return # 没有注册过；或者已经“登出”
except user_redis_dao.UserNotExistException:
print('User NOT Exist Exception')
def fetch_discounts(self, member_id):
"""
返回用户收藏的优惠券列表
:param member_id 用户的ID
:return 优惠券列表
"""
redis_dao = user_redis_dao.UserRedisDAO()
return redis_dao.fetch_coupons(member_id)
def get_discount_detail(self, discount_id):
"""
返回优惠券的详细信息
:param discount_id 优惠券ID
:return 优惠券的详细信息
"""
redis_dao = user_redis_dao.UserRedisDAO()
return redis_dao.get_discount_by_id(discount_id)
def fetch_favors_in_range(self, member_id, page_id):
"""
返回用户关注的商铺列表（分页返回）
:param member_id 用户的ID
:param page_id 返回第几页的商户列表
:return 返回商铺列表
"""
redis_dao = user_redis_dao.UserRedisDAO()
return redis_dao.fetch_favors_in_range(member_id, page_id)
def fetch_favors(self, member_id):
"""
返回用户关注的商铺内容
:param member_id :用户的ID
:return
"""
redis_dao = user_redis_dao.UserRedisDAO()
shops = redis_dao.fetch_favors(member_id) # shops is a map
return shops
def use_groupon(self, member_id, groupon_id):
"""
消费一个团购
:param member_id:
:param groupon_id:
:return:
"""
redis_dao = user_redis_dao.UserRedisDAO()
count = redis_dao.use_groupon(member_id, groupon_id)
if count: # 如果已经删除,则需要往交易的redis中放置一条交易记录
pass
def enrol(self, cell_phone_num, password):
enrol_result = {"status": False, "memberId": '', "sessionId": ''}
if cell_phone_num is not None:
redis_dao = user_redis_dao.UserRedisDAO()
result = redis_dao.enrol(cell_phone_num, password)
if result:
enrol_result = {
"status": True,
"memberId": result[0],
"sessionId": result[1]
}
return enrol_result
def fetch_groupons(self, member_id):
"""
通过用户的UUID来提取优惠券
返回一个优惠券的列表
:param member_id
:return 返回一个groupons的map
"""
redis_dao = user_redis_dao.UserRedisDAO()
groupons = redis_dao.fetch_groupons(member_id)
groupons_dict = {}
groupons_dict[Constant.GROUPONS] = groupons
return groupons_dict
def user_discount(self, member_id, discount_id):
"""
消费一个折扣
:param member_id:
:param discount_id:
:return:
"""
redis_dao = user_redis_dao.UserRedisDAO()
count = redis_dao.use_discount(member_id, discount_id)
result = {}
result["count"] = count
return result
def remove_favor_shop(self, member_id, shop_id):
"""
提取用户关注的某个商铺的详细信息
:param member_id
:param shop_id
:return shop
"""
redis_dao = user_redis_dao.UserRedisDAO()
removeCount = redis_dao.remove_favor(member_id, shop_id)
result = {}
result["remove_count"] = removeCount
return result
def get_current_integral(self, cell_phone_num):
"""
:param cell_phone_num 用户的手机号码
:return 用户当前可用的积分总数
"""
integral = {}
score = 0
redis_dao = user_redis_dao.UserRedisDAO()
member_id = redis_dao.get_member_id_by_cell_phone(cell_phone_num)
if member_id:
score = redis_dao.get_current_integral(member_id)
integral["score"] = score
return integral
def get_member_info(self, member_id):
redis_dao = user_redis_dao.UserRedisDAO()
return redis_dao.getMemberInfo(member_id)
def fetch_discounts(self, member_id):
redis_dao = user_redis_dao.UserRedisDAO()
discounts = {}
discounts["discounts"] = redis_dao.fetch_discounts(member_id)
return discounts
class TestUserRedisDAO(unittest.TestCase):
userDAO = user_redis_dao.UserRedisDAO()
cell_phone = '13612077384'
def setUp(self):
pass
def tearDown(self):
pass
def test_cell_phone_number_exist(self):
# userDAO = user_redis_dao.UserRedisDAO()
is_exist = self.userDAO.cell_phone_number_exist(self.cell_phone)
self.assertEqual(1, is_exist)
def test_enrol(self):
pwd = '123456'
self.userDAO.enrol(self.cell_phone, pwd)
is_exist = self.userDAO.cell_phone_number_exist(self.cell_phone)
self.assertEqual(1, is_exist)
def test_login(self):
pwd = '123456'
self.userDAO.enrol(self.cell_phone, pwd)
result = self.userDAO.login(self.cell_phone)
self.assertTrue(result)
is_exist = self.userDAO.cell_phone_number_exist(self.cell_phone)
self.assertEqual(1, is_exist)
def test_getMemberInfo(self):
result = self.userDAO.login(self.cell_phone)
print("test_getMemberInfo() --- ")
print(result)
print(result['memberid'])
print("--------------------------")
member = self.userDAO.getMemberInfo(result['memberid'])
print("test_getMemberInfo() --- ")
print(member)
print("--------------------------")
# result = {}
def test_get_current_integral(self):
member_id = 'gST8epDEBF8ep4xdcJcGo2'
self.userDAO.clear_integral(member_id)
save_result = self.userDAO.save_or_update_integral(member_id, 100)
result = self.userDAO.get_current_integral(member_id)
self.assertEqual(100, result)
def test_fetch_favors(self):
member_id = 'gST8epDEBF8ep4xdcJcGo2'
shops_map = self.userDAO.fetch_favors(member_id)
shops = shops_map[Constant.JSON_HEAD_SHOPS]
for shop in shops:
print(shop)
def test_list_and_dict(self):
shops_list = [{"ID": "bsuE9msdeALT4ZhfyXRTRo", "shop_name": "周大福", "floor": "1F", "location": "33号",
"logo": "http://www.bac.org/logo.jpeg"},
{"ID": "kQNbn6HQproeUGkZSKBAkf", "shop_name": "周小福", "floor": "1F", "location": "33号",
"logo": "http://www.bac.org/logo.jpeg"}]
for shop in shops_list:
print(shop)
for (key, value) in shop.items():
print(key + ":" + value)
def test_get_discount_by_id(self):
discount_id = "pvqMYQLdMYohuhRaRSiquX"
discount = self.userDAO.get_discount_by_id(discount_id)
print(discount)
def test_fetch_discounts(self):
member_id = 'gST8epDEBF8ep4xdcJcGo2'
print(self.userDAO.fetch_discounts(member_id))
def test_use_discount(self):
member_id = 'gST8epDEBF8ep4xdcJcGo2'
discount_id = 'pvqMYQLdMYohuhRaRSiquX'
count = self.userDAO.use_discount(member_id,discount_id)
self.assertEqual(1,count)