|
|
import requestsfrom bs4 import BeautifulSoup as bsimport bot_app.fulfillment.utility as utilfrom datetime import datetime
base_url = 'http://family.wanhai.com'account = Nonepassword = Nonepayload = {'Account': account, 'Password': password}meeting_rooms = Nonemassage_rooms = Nonesession_req = None
def getSession(check_login=True): global session_req if session_req is None: session_req = requests.Session() if check_login and not isLogin(): login(account, password) return session_req
def login(id, pw): global session_req if session_req is None: session_req = requests.Session() res = session_req.post( base_url + '/Login.jsp', data={'Account': id, 'Password': pw})
print(res.request.headers['Cookie'])
def isLogin(): global session_req if session_req is None: session_req = requests.Session() route_url = base_url + '/LeaseEquip/equipListOnePage.jsp' res = session_req.get(route_url, allow_redirects=False) if res.status_code == 200: print(res.request.headers['Cookie']) return True else: return False
# 取得可用設備列表def get_equip_list(type='MEETING'): equipList_url = base_url + '/LeaseEquip/equipListOnePage.jsp' result = getSession().get( equipList_url + '?file_num=61621&account_id=' + account + '&equip_type=' + type) print(result.request.headers['Cookie'])
soup = bs(result.text, 'html.parser') elements = soup.find_all('input', {'id': 'check2', 'name': 'ID_keyD'}) equipDict = dict() for v1, v2 in [(e['value'], e.find_parent().find_next_sibling().text) for e in elements]: equipDict[v1] = {'id': v1, 'type': type, 'name': v2} # print(equipDict) return equipDict
# 取得設備名稱def get_equip_name(equipId): if equipId in meeting_rooms.keys(): return meeting_rooms[equipId]['name'] elif equipId in massage_rooms.keys(): return massage_rooms[equipId]['name'] else: return None
# 查詢設備是否已被占用(by 特定日期/時段)def is_equip_in_use(equip_id, b_date=datetime.today(), b_time=datetime.now(), e_time=datetime.now(), equip_type='MEETING'): if equip_type == 'MASSAGE': print('Not Support Now!!!') return False equipUsageURL = base_url + '/LeaseEquip/equipUsage.jsp' payload = [('q_from_date', datetime.strftime(b_date, '%Y%m%d')), ('q_from_time', datetime.strftime(b_time, '%H:%M:%S')), ('q_to_date', datetime.strftime(b_date, '%Y%m%d')), ('q_to_time', datetime.strftime(e_time, '%H:%M:%S')), ('q_equip_type', equip_type), ('ID_keyD', equip_id)] result = getSession().post(equipUsageURL, data=payload) soup = bs(result.text, 'html.parser') elements = soup.select('td > b') if (len(elements) > 1) and ('目前無人預約' not in elements[1].text): return True else: return False return False
# 查詢設備可用時段(by 特定日期)def search_available_time(equip_type, equip_id, strDate): if equip_type == 'MASSAGE': print('Not Support Now!!!') return from interval import Interval equipUsageURL = base_url + '/LeaseEquip/equipUsage.jsp' payload = [('q_from_date', strDate), ('q_from_time', ''), ('q_to_date', strDate), ('q_to_time', ''), ('q_equip_type', equip_type), ('ID_keyD', equip_id)] result = session_req.post(equipUsageURL, data=payload) soup = bs(result.text, 'html.parser') # print(soup.prettify()) print('{}:\n{}'.format(equip_id, soup.select('td[nowrap]')[0].text))
# 查詢可用設備(by 特定日期/時段)def search_available_equips(equip_type, strDate, b_time, e_time): if equip_type == 'MASSAGE': print('Not Support Now!!!') return from interval import Interval equipUsageURL = base_url + '/LeaseEquip/equipUsage.jsp' payload = [('q_from_date', strDate), ('q_from_time', ''), ('q_to_date', strDate), ('q_to_time', ''), ('q_equip_type', equip_type)]
payload.extend([('ID_keyD', e) for e in meeting_rooms.keys()]) result = getSession().post(equipUsageURL, data=payload) soup = bs(result.text, 'html.parser') equips = [equip.text.strip() for equip in soup.select( 'div[id="equips"] td[align="center"]')] # print(equips) availList = list() for index, elem in enumerate(soup.select('td[nowrap]')): availList.insert(index, 1) if '目前無人預約' in elem.text: continue target = Interval(int(b_time), int(e_time), closed=False) periods = [x.text.split('~') for x in elem.select('b')] periodIntvs = [Interval(x[0], x[1]) for x in list( map(lambda bb: [int(x) for x in bb], periods))]
# print(equips[index]) # for pp in periodIntvs:ontext['lifespanCount'] = 0 # print(pp,'\n')
for periodIntv in periodIntvs: if target.overlaps(periodIntv): availList[index] = 0 break
# print(availList) for index, avail in enumerate(availList): if avail == 1: print('{}\n'.format(equips[index]))
# 預約設備def booking_equip(equip_id, equip_type, strDate, b_time, e_time): if equip_type == 'MASSAGE': print('Not Support Now!!!') bookingURL = base_url + '/LeaseEquip/equipBooking_db.jsp' payload = { 'item_choice1': 'F', 'equipment_id1': equip_id, 'lease_action1': 'book', 'str_yymmdd': strDate, 'f_rent_fm1': b_time, 'f_rent_to1': e_time, 'equip_count': '1', 'f_str_remark1': '#7253', 'contactor_mail1': 'x' # 'lease_time_code_1': '', # 'term_time_code1': '', # 'equip_name_c1': '<5F>台北廳 【可容納90人】', # 'lease_time_code1': 'enter_rent_time', # 'tea1': '0', # 'water1': '0', # 'coffee1': '0', # 'meeting_type1': 'IN', # 'selected_year': '2019', # 'selected_month': '06', # 'selected_day': '06', # 'equip_type': 'MEETING', # 'fm_date': '20190606', # 'to_date': '20190606', # 'select_equips': equip_id } soup = bs(getSession().post(bookingURL, data=payload).text, 'html.parser') print(soup.select('table tr:nth-child(2) td:nth-child(5)')[0].text) # print(soup.prettify())
# 取消預約設備def cancel_booking(equip_type, rent_no): if equip_type == 'MASSAGE': print('Not Support Now!!!') bookingURL = base_url + '/LeaseEquip/userBooking_db.jsp' payload = { # 'ID_key': '0u', # 'action_type_0u': 'U', 'action_type': 'D', 'ID_keyDD': rent_no, # 'i_rent_no_0u': rent_no # 'ID_key': '1u', # 'action_type_1u': 'U', # 'i_rent_no_1u': '350438', # 'q_equip_id':'', # 'q_rent_date':'', # 'q_rent_no':'', # 'q_su_id':'' } soup = bs(getSession().post(bookingURL, data=payload).text, 'html.parser') print(soup.prettify())
# 列出個人設備預約紀錄def get_rented_equips(): equipList_url = base_url + '/LeaseEquip/equipListOnePage.jsp' result = getSession().get( equipList_url + '?file_num=61621&account_id=' + account + '&equip_type=MEETING') soup = bs(result.text, 'html.parser') elements = soup.select('form[name="dataForm1"] input[name^="i_rent_no"]') elements = [x.find_parent().select('td[align="center"]') for x in elements] rentedList = [(x[0].input['value'], x[1].text, x[2].text, x[3].text) for x in elements] return rentedList
def list_rented_equips(fulfillment): rented_list = get_rented_equips() if len(rented_list) == 0: return util.simple_response(text_content=u'您目前沒有預約任何的設備') else: str_rented_list = u'您預約的設備如下：\n' str_rented_list += ', \n'.join([x[2] + ' ' + x[3] + ' ' + x[1] for x in rented_list]) return util.simple_response(text_content=str_rented_list)
def list_all_meeting_rooms(fulfillment): global meeting_rooms str_meeting_rooms = ', \n'.join([x['id'] + x['name'] for x in meeting_rooms.values()]) return util.simple_response(text_content=str_meeting_rooms)
def check_equip_in_use(fulfillment): try: date_pattern = '%Y-%m-%dT%H%M%S%z' params = fulfillment.get('queryResult').get('parameters') meeting_room = params.get('meeting_room') date = datetime.strptime(params.get('date').replace(':', ''), date_pattern) start_time = datetime.strptime(params.get('time-period').get('startTime').replace(':', ''), date_pattern) end_time = datetime.strptime(params.get('time-period').get('endTime').replace(':', ''), date_pattern) meeting_room_id = next([x['id'] for x in meeting_rooms if x['name'].find(meeting_room) >= 0], None)
if is_equip_in_use(meeting_room_id, date, start_time, end_time): return util.simple_response(text_content=meeting_room+u'已經被人預約') else: return util.simple_response(text_content=meeting_room+u'尚未被預約')
except Exception as e: return util.simple_response(text_content=str(e)) pass
def greet_cancel(fulfillment): util.reset_all_contexts(fulfillment) return util.simple_response(fulfillmentObj={ 'fulfillmentText': '好的，沒有問題！', 'outputContexts': fulfillment.get('queryResult').get('outputContexts') })
def init_app(app): global password, account, meeting_rooms account = app.config['WHL_FAMILY_ID'] password = app.config['WHL_FAMILY_PW'] login(account, password) meeting_rooms = get_equip_list()
|
