def login(): if request.method == 'POST': username = request.form.get('username') password = request.form.get('password') crawler = Crawler(username, password) status_code = crawler.login() if status_code == 'FAILED': return '<h1>没有该用户或者密码错误!</h1>' courses_info = crawler.get_courses_schedule() schedule_info = Generate_schedule_table(courses_info).generate() return render_template('schedule.html', schedule_info=schedule_info, cur_user=username) else: return '<html><body><h1>请先登录</h1></body></html>'
class Test(unittest.TestCase): def setUp(self): self.cw = Crawler() def test_get_verify_code(self): with open('verify.txt', 'r') as f: verifyCode = f.read() print(self.cw.filter_verify_code(verifyCode)) # self.assertEqual(True, True) def tearDown(self): pass
def setUp(self): self.cw = Crawler()