def index(): domain = settings.db.domain.find({"UID":"secret"},{'UID':False}).sort('_id',-1).limit(1) print(str(domain)) if request.args.get('d'): d_id = request.args.get('d') domain = settings.db.domain.find({"_id": ObjectId(d_id)},{'_id': False}) all = settings.db.domain.find().sort('_id',-1).limit(15) info="" if request.method == 'POST': domain = str(request.form['domain']) add_to_que(domain,'secret') info="Scan started" return render_template('base.html',domain=domain,info=info,all=all)
def mypage(): info="" user_id = validat_session(request.cookies.get("samma_uid")) if user_id != False: session_id = session(user_id) user_id_mongo=user_id.decode("utf-8") user_data = get_user_data(user_id_mongo) user_r = settings.db.domain.find({"UID":user_id_mongo}) user_domain=[] baseline = settings.db.baseline.find({'UID':user_id_mongo}) for d in user_r: user_domain.append(d) if request.method == 'POST': domain = str(request.form['domain']) add_to_que(domain,user_id_mongo) info="Scan started" if request.method=='GET': if request.args.get('bs'): if user_data['baslinestatus'] =="Full": ''' full baseline cant add more ''' info="Cant add more baseline it's full" else: add_baselinecount(user_id_mongo) bs_id = request.args.get('bs') domain = request.args.get('domain') settings.db.baseline.insert( { 'bs':bs_id, 'domain':domain, 'UID':user_id_mongo,'Date':format(datetime.datetime.utcnow()),'Status':'Not matced'} ) #resp = make_response(render_template('mypage.html',info=info,domain=user_domain,baseline=baseline,user_data=user_data)) #resp.set_cookie('northamlin_uid',value=session_id) #return resp return redirect('home') elif request.args.get('del'): remove_baselinecount(user_id_mongo) bs_id = request.args.get('del') settings.db.baseline.remove({'_id':ObjectId(bs_id), 'UID':user_id_mongo}) #resp = make_response(render_template('mypage.html',info=info,domain=user_domain,baseline=baseline,user_data=user_data)) #resp.set_cookie('northamlin_uid',value=session_id) #return resp return redirect('home') elif request.args.get('show'): domain=request.args.get('show') scan = settings.db.domain.find_one({"_id": ObjectId(domain),"UID":user_id_mongo}) resp = make_response(render_template('mypage.html',info=info,domain=user_domain,scan=scan,baseline=baseline,user_data=user_data)) resp.set_cookie('northamlin_uid',value=session_id) return resp resp = make_response(render_template('mypage.html',info=info,domain=user_domain,baseline=baseline,user_data=user_data)) resp.set_cookie('samma_uid',value=session_id) return resp else: redirect_to_index = redirect('/' ,_scheme='https') response = make_response(redirect_to_index ) response.set_cookie('samma_uid',value='nope') return response