qa_bot/equip_crawling.py

#%%
import requests
from bs4 import BeautifulSoup as bs

base_url='http://family.wanhai.com'
loginURL = base_url + '/Login.jsp'
account='M1933'
password='1933whlM'
payload = {'Account': account,'Password': password}
meeting_rooms = None
massage_rooms = None
session_req = None
# session_req.post(loginURL,data=payload)

#%%
def login(id,pw):
    global session_req
    if session_req == None:
        session_req=requests.Session()
    res = session_req.post(loginURL,data={'Account': account,'Password': password})
    print(res.request.headers['Cookie'])

#%%
def isLogin():
    global session_req
    if session_req == None:
        session_req=requests.Session()
    route_url = base_url + '/MainPage.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 getEquipList(type='MEETING'):
    equipList_url = base_url + '/LeaseEquip/equipListOnePage.jsp'
    result = session_req.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 getEquipName(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 isEquipInUsed(b_date,b_time,e_time,equip_type,equip_id):
    if equip_type=='MASSAGE':
        print('Not Support Now!!!')
        return Fasle;
    equipUsageURL = base_url+'/LeaseEquip/equipUsage.jsp'
    payload = [('q_from_date', b_date),
               ('q_from_time',b_time),
               ('q_to_date',b_date),
               ('q_to_time',e_time),
               ('q_equip_type',equip_type),
               ('ID_keyD',equip_id)]
    result = session_req.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 = session_req.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:
        #     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 bookingEquip(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(session_req.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 cancalBookingEquip(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(session_req.post(bookingURL,data=payload).text,'html.parser')
    print(soup.prettify())

#%%
# 列出個人設備預約紀錄
def listRentedEquips():
    equipList_url = base_url + '/LeaseEquip/equipListOnePage.jsp'
    result = session_req.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   

#%%
listRentedEquips()

#%%
equipId = 'CF15-10F-DN-01'
print(getEquipName(equipId))
inUsed = isEquipInUsed('20190606','1900','2000','MEETING',equipId)
print('Equip in used? {}'.format(inUsed))                                                              
#%%
search_available_equips('MEETING','20190624','0900','1200')
#%%
search_available_time('MEETING','CF01-9F','20190624')

#%%
if not isLogin():
    login(account,password)
meeting_rooms = getEquipList('MEETING')
massage_rooms = getEquipList('MASSAGE')

for x in meeting_rooms:
    print(x,meeting_rooms.get(x))
print('\n')
for x in massage_rooms:
    print(x,massage_rooms.get(x))


#%%
if __name__ == '__main__':
   pass