示例#1
0
文件: endpoint.py 项目: yc10086/lost
    def post(self):
        # get data from parser
        data = login_parser.parse_args()
        dbm = access.DBMan(LOST_CONFIG)
        # find user in database
        if 'user_name' in data:
            user = dbm.find_user_by_user_name(data['user_name'])

        # check password
        if user and user.check_password(data['password']):
            dbm.close_session()
            expires = datetime.timedelta(minutes=LOST_CONFIG.session_timeout)
            expires_refresh = datetime.timedelta(
                minutes=LOST_CONFIG.session_timeout + 2)
            if FLASK_DEBUG:
                expires = datetime.timedelta(days=365)
                expires_refresh = datetime.timedelta(days=366)
            access_token = create_access_token(identity=user.idx,
                                               fresh=True,
                                               expires_delta=expires)
            refresh_token = create_refresh_token(user.idx,
                                                 expires_delta=expires_refresh)
            return {'token': access_token, 'refresh_token': refresh_token}, 200
        dbm.close_session()
        return {'message': 'Invalid credentials'}, 401
示例#2
0
文件: endpoint.py 项目: l3p-cv/lost
 def post(self):
     # get data from parser
     data = login_parser.parse_args()
     dbm = access.DBMan(LOST_CONFIG)
     user = dbm.find_user_by_user_name(data['user_name'])
     lm = LoginManager(dbm, data['user_name'], data['password'])
     response = lm.login()
     if LOST_CONFIG.worker_management == 'dynamic':
         if response[1] == 200:
             dask_session.ds_man.create_user_cluster(user)
     dbm.close_session()
     return response