def post(self):
data = vendor_schema.load(request.get_json())
vendor = VendorModule(**data)
if (vendor.find_by_username()):
return {"msg": "TÊN ĐĂNG NHẬP ĐÃ TỒN TẠI"}, 400
if (vendor.find_by_email()):
return {"msg": "EMAIL ĐÃ TỒN TẠI"}, 400
token = generate_confirmation_token(vendor.email)
confirm_url = url_for('userregister', token=token, _external=True)
email_confirmation = q.enqueue(Mail.send_mail_confirmation_to_user, args=(
vendor.email, confirm_url,), result_ttl=0)
vendor.id = vendor.find_maxium_user()
vendor.password = vendor.hash_password
trigger_data = {
"id": vendor.id,
"username": vendor.username,
"email": vendor.email,
"tel": vendor.tel,
"role": vendor.role,
"status": vendor.status,
}
email_confirmation = q.enqueue(vendor.save_to_database, args=(
vendor_schema.dump(vendor), trigger_data,), result_ttl=0)
return {"msg": "XÁC THỰC EMAIL ĐỂ KÍCH HOẠT TÀI KHOẢN"}, 201
def console():
bugbit, bugtype = core.GetBit()
counts = core.GetCounts()
ports = core.GetPort()
services = core.GetServices()
target = core.GetTargetCount()
if 'targetscan' in session:
urls = session['targetscan'].split()
redispool.hincrby('targetscan', 'waitcount', len(urls))
for url in urls:
queue.enqueue(SZheScan, url)
# SZheScan.delay(url)
session.pop('targetscan')
try:
lastscantime = BaseInfo.query.order_by(BaseInfo.id.desc()).first().date
except:
lastscantime = "暂无扫描"
pass
if request.method == 'GET':
return render_template('console.html',
bugbit=bugbit,
bugtype=bugtype,
counts=counts,
lastscantime=lastscantime,
ports=ports,
services=services,
target=target)
else:
session['targetscan'] = request.form.get('urls')
return redirect(url_for('console'))
def post(self):
data = user_schema.load(request.get_json())
user = UserModule(**data)
if user.find_by_username():
return {"msg": "TÊN ĐĂNG NHẬP ĐÃ TỒN TẠI"}, 400
if user.find_by_email():
return {"msg": "EMAIL ĐÃ TỒN TẠI"}, 400
token = generate_confirmation_token(user.email)
confirm_url = url_for('userregister', token=token, _external=True)
email_confirmation = q.enqueue(Mail.send_mail_confirmation_to_user,
args=(
user.email,
confirm_url,
),
result_ttl=0)
user.id = user.find_maxium_user()
user.password = user.hash_password
user.save_to_database(user_schema.dump(user))
return {"msg": "XÁC THỰC EMAIL ĐỂ KÍCH HOẠT TÀI KHOẢN"}, 201