def login(): """用户登录 (GET|POST) :uri: /users/login :param name: 用户名 :param password: 密码 :param type: 登陆类型(name, phone,) :returns: {'user': object, 'ut': string} """ params = request.values login_type = params.get('type', 'name') name = params.get("name", None) password = params.get("password", None) if name is None or password is None or login_type not in ['name', 'phone']: return error.InvalidArguments user = User.login(name, password, login_type=login_type) if not user: return error.LoginFailed # 初始化用户任务 UserTask.create_and_init_user_tasks(str(user._id)) token = User.gen_token(str(user._id)) return {'user': user.format(), 'ut': token}
def refresh_token(): """刷新用户token时间 (GET|POST&LOGIN) :uri: /users/refresh_token :returns: {'user': Object} """ ut = request.values.get("ut", None) user = request.authed_user token = User.set_token(ut, str(user._id)) if token: User.recommend_users(str(user._id)) # 更新用户活跃时间 user.update_model({'$set': {'update_at': time.time()}}) # 初始化用户任务 UserTask.create_and_init_user_tasks(str(user._id)) return {'user': user.format(include_fields=['passid'])}
def register(): """用户注册 (GET|POST) :uri: /users/register :param name: 用户名 :param password: 密码 :param nickname: 昵称 :returns: {'user': object, 'ut': string} """ params = request.values.to_dict() name = params.get("name", None) # delete password from data so that we don't save it to mongo password = str(params.pop("password", None)) nickname = params.get('nickname', None) if not name or not password or not nickname: return error.InvalidArguments invalid_error = User.invalid_password(password) if invalid_error: return invalid_error invalid_error = User.invalid_nickname(nickname) if invalid_error: return invalid_error if User.get_by_name(name): return error.UserExists user = User.init() user.update(params) salt = os.urandom(const.PWD_HASH_LEN) pwd = User.gen_pwd_hash(password, salt) user._salt = Binary(salt) user._password = Binary(pwd) uid = user.create_model() new_user = User.get_one(uid) # 初始化用户任务 UserTask.create_and_init_user_tasks(str(new_user._id)) token = User.gen_token(str(uid)) return {'user': new_user.format(), 'ut': token}
def do_task(worker, job): from wanx.base.log import print_log from wanx.base import const, util, jpush from wanx.models.user import User from wanx.models.game import Game from wanx.models.live import Live_Activity from wanx.models.activity import ActivityConfig from wanx.models.game import GameActivity from wanx.models.task import BEGIN_LIVE, UserTask from wanx.base.xredis import Redis from wanx.base.xmysql import MYDB from wanx.platforms.xmatch import Xmatch data = job.data # 记录日志 print_log('create_live_activity', data) data = json.loads(data) # 记录日志 print_log('create_live_activity', '%s ==========================> Start' % (data['event_id'])) user = User.get_one(data['user_id'], check_online=False) if not user: return '' # 主播上线push推送 push_content = u"您关注的主播 {0} 正在直播,点击围观".format(user.nickname) message_content = u"您关注的主播 {0} 正在直播,速速去围观吧!".format(user.nickname) an_link = "playsdk://video_live/{0}/".format(data['event_id']) ios_link = "playsdk://live/{0}".format(data['event_id']) jpush.jpush_schedule_create(data['event_id'], user, push_content, message_content, an_link, ios_link) # 赛事预约直播上线推送 battle = Xmatch.getbattle(user._id, data['name']) if battle: battle = battle['data']['battle'] push_title = u'您预约的比赛正在进行' push_content = u"{0} vs {1}正在直播,点击围观".format(battle['team_1'], battle['team_2']) message_content = u"您预约的比赛{0} vs {1}正在直播!".format( battle['team_1'], battle['team_2']) if battle.get('players', '') != "": push_content = message_content = battle['live_name'] an_link = "playsdk://video_live/{0}/".format(data['event_id']) ios_link = "playsdk://live/{0}".format(data['event_id']) jpush.jpush_withtitle_create(data['event_id'], battle['_id'], push_title, push_content, message_content, an_link, ios_link) game = Game.get_one(data['game_id'], check_online=False) if not game: return '' aid = None aids = ActivityConfig.get_by_type(const.FROM_LIVE) for a in ActivityConfig.get_list(aids): gids = GameActivity.game_activity_ids(a['_id']) if gids and data['game_id'] not in gids: continue aid = a['_id'] break if not aid: return '' print_log('create_live_activity', '%s ==========================> activity_id' % (aid)) activity_live = Live_Activity.get_activity_live(str(user._id), aid) if not activity_live: key = 'lock:activity:live:%s:%s' % (str(user._id), aid) with util.Lockit(Redis, key) as locked: if locked: return '' activity_live = Live_Activity.init() activity_live.author = ObjectId(data['user_id']) activity_live.game = ObjectId(data['game_id']) activity_live.event_id = data['event_id'] activity_live.activity_id = ObjectId(aid) activity_live.create_model() # 如果没有活动任务则创建 UserTask.create_and_init_user_tasks(str(user._id)) if not MYDB.is_closed(): MYDB.close() try: UserTask.check_user_tasks(user._id, BEGIN_LIVE, 1, data['game_id'], aid) except Exception as e: print_log('create_live_activity', '%s ========' % e) if not MYDB.is_closed(): MYDB.close() # 记录日志 print_log('create_live_activity', '%s ==========================> Finished' % (data['event_id'])) return ''