def login_user():
'''
Request new password
:return:
'''
info=""
user_id = validat_session(request.cookies.get("samma_uid"))
if user_id != False:
pass
if request.method == 'POST':
email = str(request.form['email'])
password = str(request.form['password'])
user_id = login(email,password)
if user_id !=False:
#Uer has given correct user and password
#Creating session and login session cookie
session_id = session(user_id)
redirect_to_mypage = redirect('/home' , _scheme='https')
#resp = make_response(render_template('pro.html',info=info))
resp = make_response(redirect_to_mypage)
resp.set_cookie('samma_uid',value=session_id)
return resp
else:
info ="Not a valid email ore password"
return render_template('pro.html',info=info)
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