You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
280 lines
9.7 KiB
280 lines
9.7 KiB
import requests
|
|
from bs4 import BeautifulSoup as bs
|
|
import bot_app.fulfillment.utility as util
|
|
from datetime import datetime
|
|
|
|
base_url = 'http://family.wanhai.com'
|
|
account = None
|
|
password = None
|
|
payload = {'Account': account, 'Password': password}
|
|
meeting_rooms = None
|
|
massage_rooms = None
|
|
session_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()
|