def post(self): current = get_jwt_identity() access_token = create_access_token(identity=current) access_jti = get_jti(encoded_token=access_token) db.set("z:tokens:{jti}".format(jti=access_jti), "false", current_app.config.get("ACCESS_EXPIRES") * 1.2) return jsonify({"msg": "ok", "access_token": access_token})
def get_real_whois_domain(domain): """docstring for get_real_whois_domain""" server = WHOIS_SERVER.get(domain.suffix, '') res = pythonwhois.net.get_whois_raw(domain.registered_domain, server=server, never_cut=False) res.reverse() res = convert_newline('\n'.join(res)) db.set(REDIS_KEY_DOMAIN % domain.registered_domain, res) return res
def add(cls, ip): created = "%s" % now() new_ip = cls(ip=ip, created=created) if db.sismember("ips", ip): raise BadRequest("IP address %s already registered" % ip) db.sadd("ips", ip) db.set("ip:%s" % ip, json.dumps({"created": created})) return new_ip
def init_wechat_sdk(): """ 初始化微信sdk :return: WechatBasic """ access_token = db.get("wechat:access_token") jsapi_ticket = db.get("wechat:jsapi_ticket") token_expires_at = db.get("wechat:access_token_expires_at") ticket_expires_at = db.get("wechat:jsapi_ticket_expires_at") if all((access_token, jsapi_ticket, token_expires_at, ticket_expires_at)): wechat = WechatBasic(appid=app.config['APP_ID'], appsecret=app.config['APP_SECRET'], token=app.config['TOKEN'], access_token=access_token, access_token_expires_at=int(token_expires_at), jsapi_ticket=jsapi_ticket, jsapi_ticket_expires_at=int(ticket_expires_at)) else: wechat = WechatBasic(appid=app.config['APP_ID'], appsecret=app.config['APP_SECRET'], token=app.config['TOKEN']) access_token = wechat.get_access_token() db.set("wechat:access_token", access_token['access_token'], 7000) db.set("wechat:access_token_expires_at", access_token['access_token_expires_at'], 7000) jsapi_ticket = wechat.get_jsapi_ticket() db.set("wechat:jsapi_ticket", jsapi_ticket['jsapi_ticket'], 7000) db.set("wechat:jsapi_ticket_expires_at", jsapi_ticket['jsapi_ticket_expires_at'], 7000) return wechat
def save_snapshot(live_analysis_mode, save_button_clicked, ticker, snapshot_uuid, df_dict): if 1 in live_analysis_mode: # generate a fresh UUID snapshot_uuid = str(uuid.uuid5(uuid.UUID(snapshot_uuid), ticker)) if save_button_clicked: # df_dict[ticker] = {**df_dict[ticker], **dcf_dict[ticker]} if 'analysis_timestamp' in df_dict[ticker][ 'stats_dict']: # >= v0.6-alpha.3 df_dict[ticker]['stats_dict'][ 'analysis_timestamp'] += f',\n{snapshot_uuid} : Analysis saved @ {datetime.now().strftime("%b %-m %Y %H:%M:%S")}' db.set(ticker + '-' + snapshot_uuid, json.dumps(df_dict)) return '/apps/dcf/' + ticker + '/' + snapshot_uuid, False, not save_button_clicked else: return dash.no_update, True, True
def add(cls, source_ip, destination_ip, result): created = "%s" % now() new_ip = cls( source_ip=source_ip, destination_ip=destination_ip, result=result, created=created ) db.set( "result:%s:%s:%s" % (source_ip, destination_ip, time.strftime("%Y%m%d%H%M%S", time.localtime())), json.dumps({ "created": created, "source_ip": source_ip, "destination_ip": destination_ip, "result": result }) ) return new_ip
def update_wechat_token(): """刷新微信 token """ wechat = init_wechat_sdk() wechat.grant_token() wechat.grant_jsapi_ticket() access_token = wechat.get_access_token() db.set("wechat:access_token", access_token['access_token'], 7000) db.set("wechat:access_token_expires_at", access_token['access_token_expires_at'], 7000) jsapi_ticket = wechat.get_jsapi_ticket() db.set("wechat:jsapi_ticket", jsapi_ticket['jsapi_ticket'], 7000) db.set("wechat:jsapi_ticket_expires_at", jsapi_ticket['jsapi_ticket_expires_at'], 7000)
def login(self, uid, user=None): access_token = create_access_token(identity=uid, fresh=True) refresh_token = create_refresh_token(identity=uid) user = db.hgetall("z:users:{uid}".format( uid=uid)) if user == None else user # store tokens in db for security and logout blacklisting access_jti = get_jti(encoded_token=access_token) refresh_jti = get_jti(encoded_token=refresh_token) db.set("z:tokens:{jti}".format(jti=access_jti), "false", current_app.config.get("ACCESS_EXPIRES") * 1.2) db.set("z:tokens:{jti}".format(jti=refresh_jti), "false", current_app.config.get("REFRESH_EXPIRES") * 1.2) return jsonify( merge_dict( filter_dict(user, self.exposedFields), { "msg": "ok", "access_token": access_token, "refresh_token": refresh_token, "uid": uid }))
def get_real_whois_ip(ip): """docstring for get_real_whois_ip""" ip = ipwhois.IPWhois(ip) res = convert_newline(ip.lookup(inc_raw=True)['raw']) db.set(REDIS_KEY_IP % ip, res) return res
def delete(self): jti = get_raw_jwt()['jti'] db.set("z:tokens:{jti}".format(jti=jti), "true", current_app.config.get("REFRESH_EXPIRES") * 1.2) return jsonify({"msg": "refresh token revoked"})