From 4721f07e4a5409e4fe82c78758793c9d2dd3c17e Mon Sep 17 00:00:00 2001 From: JayYoung <60842691+JayYoung2021@users.noreply.github.com> Date: Fri, 28 Aug 2020 15:39:23 +0800 Subject: [PATCH] FINAL VERSION --- AwesomePU_CLI.py | 116 +++++++++++++++++++++++++++++++++++++++++++++++ PUWebSpider.py | 23 +++++----- 2 files changed, 128 insertions(+), 11 deletions(-) create mode 100644 AwesomePU_CLI.py diff --git a/AwesomePU_CLI.py b/AwesomePU_CLI.py new file mode 100644 index 0000000..a1f6b6a --- /dev/null +++ b/AwesomePU_CLI.py @@ -0,0 +1,116 @@ +import sys +from PUWebSpider import * + +try: # Network monitoring module + requests.get('http://njtech.pocketuni.net/') +except requests.exceptions.ProxyError: + sys.exit() + +# example = PU(None, '10086', 'password') +# example = PU('480', '10086', 'password') +# example.set_college('斯莱特林学院', '魔法类9.75') +print('输入 1 使用手机号登录') +print('输入 2 使用学号登录（南京工业大学独占）') +while True: + while True: + mode = input() + if mode == '1': + account = str(input('手机号：')) + password = str(input('密码：')) + example = PU(None, account, password) + break + elif mode == '2': + sid = '480' + account = str(input('学号：')) + password = str(input('密码：')) + example = PU(sid, account, password) + break + else: + print('输入有误，请重试') + if example.verification(): + print('登陆成功') + break + else: + print('登录失败，请重试') + +college_list = example.get_college_list() +for i in range(len(college_list)): + print(i + 1, college_list[i]) +college = college_list[int(input('学院序号：')) - 1] +class_ = input('班级：') +example.set_college(college, class_) + +list_ = example.filtered_activities +page = 0 +totalCount = len(list_) # 内容总数量 +loadCount = 10 # 每页加载数量 +totalPage = (totalCount + loadCount - 1) // loadCount # 所求总页数 + + +def end(p): + if totalCount - loadCount * (p + 1) < 0: + return totalCount - loadCount * p + else: + return loadCount + + +if not list_: + print('无可用活动') +else: + for i in range(0, end(page)): + print(i + 1, list_[i]['title']) + while True: + print('查看上一页输入"<" 查看活动详情输入序号 查看下一页输入">"') + print('退出程序输入0') + cmd = input('请输入命令：') + if cmd == '<': + if page > 0: + page -= 1 + for i in range(loadCount * page, end(page)): + print(i % 10 + 1, list_[i]['title']) + elif cmd == '>': + if page < totalPage - 1: + page += 1 + for i in range(loadCount * page, end(page)): + print(i % 10 + 1, list_[i]['title']) + elif ('0' <= cmd <= '9') or cmd == '10': + if cmd == '0': + break + # for k, v in list_[loadCount * page + int(cmd) - 1].items(): + # print(k, v) + tmp = loadCount * page + int(cmd) - 1 + print('活动名称：' + list_[tmp]['title']) + print('归属组织：' + list_[tmp]['author']) + print('活动分类：' + list_[tmp]['class']) + print('活动地点：' + list_[tmp]['location']) + print('活动院系：', end='') + if list_[tmp]['college'] is None: + print("限制") + else: + print(list_[tmp]['college']) + print('活动年级：', end='') + if list_[tmp]['grade'] is None: + print("限制") + else: + print(list_[tmp]['grade']) + print('可参与部落：', end='') + if list_[tmp]['tribe'] is None: + print("无限制") + else: + print(list_[tmp]['tribe']) + print('活动时间：' + list_[tmp]['time']) + print('报名起止：' + list_[tmp]['reg']) + print('外勤打卡：' + list_[tmp]['clock_in']) + print('参加费用：' + list_[tmp]['cost']) + print('参加人数：' + list_[tmp]['exist']) + print('剩余名额：' + list_[tmp]['surplus']) + print('联系方式：', end='') + if list_[tmp]['contact'] is None: + print("无限制") + else: + print(list_[tmp]['contact']) + print('活动简介：' + list_[tmp]['intro']) + for i in range(loadCount * page, end(page)): + print(i % 10 + 1, list_[i]['title']) + else: + print('输入有误，请重试') diff --git a/PUWebSpider.py b/PUWebSpider.py index 6f27996..22de289 100644 --- a/PUWebSpider.py +++ b/PUWebSpider.py @@ -77,9 +77,9 @@ def parse_one_activity(html): # str -> list class PU: def __init__(self, sid, account, password): - self.password = str(password) + self.password = password if sid is None: # Judge login method - self.mobile = str(account) + self.mobile = account self.post_url = 'http://pocketuni.net/index.php?app=home&mod=Public&act=doMobileLogin' # Mobile login url post_data = { 'mobile': self.mobile, @@ -87,8 +87,8 @@ def __init__(self, sid, account, password): 'remember': 'on' } else: - self.sid = str(sid) - self.number = str(account) + self.sid = sid + self.number = account self.post_url = 'http://pocketuni.net/index.php?app=home&mod=Public&act=doLogin' # Sid and number login url post_data = { 'sid': self.sid, @@ -115,8 +115,8 @@ def verification(self): # Verify whether the login process is successful def set_college(self, college, tribe): # Set user's college and tribe self.college = college - self.tribe = tribe.split() - self.grade = '20' + re.search('\\d+', self.tribe[0])[0][:2] + self.tribe = tribe + self.grade = '20' + re.search('\\d+', self.tribe)[0][:2] def get_college_list(self): # Get the college list of user's school chrome_options = Options() @@ -129,6 +129,7 @@ def get_college_list(self): # Get the college list of user's school result = html.xpath('/html/body/div[2]/div/div[3]/div[2]/div[1]/div[1]/div[2]/a/text()') return result + @property def filtered_activities(self): # Filter out expired and other unavailable activities p = 0 list_ = [] @@ -140,18 +141,18 @@ def filtered_activities(self): # Filter out expired and other unavailable activ count = 0 for item in items: - if item['status'] != ['活动已结束']: - bool_ = False + if item['status'] != ['活动已结束'] and item['status'] !=['报名已结束']: + flag = False info_activity = parse_one_activity(self.session.get(item['link']).text) if info_activity['tribe'] is None: if (self.college in info_activity['college']) or (info_activity['college'] == '全部'): if (self.grade in info_activity['grade']) or (info_activity['grade'] == '全部'): - bool_ = True + flag = True else: for item_ in self.tribe: if item_ in info_activity['tribe']: - bool_ = True - if bool_ is True: + flag = True + if flag is True: list_.append(info_activity) else: count += 1