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