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})
