def key_word(): keyForm = KeyForm(request.form) companyForm = CompanyForm(request.form) key_word = None names = None company = None number = 0 if keyForm.validate_on_submit(): key_word = keyForm.key_word.data scrape = Scrape(key_word) number, driver = scrape.login() tags, names = scrape.get_company_list(driver) if companyForm.validate_on_submit(): company = companyForm.company.data scrape = Scrape(company) number, driver = scrape.login() driver = scrape.select_company(driver, company) lists = scrape.scrapy(driver) return render_template('company.html', lists=lists) return render_template('index.html', keyForm=keyForm, companyForm=companyForm, key_word=key_word, names=names, number=number, company=company)
def get_list(): keyWord = request.args.get("keyWord") scrape = Scrape(keyWord) number, driver = scrape.login() tags, names = scrape.get_company_list(driver) name_list = [] for key in names.keys(): name_list.append(key) return jsonify({"num": number, "list": name_list})