SRT 자동 예매

1. 개요

 잘 쓰던 Macro가 막혀버려서 직접 제작하였다. selenium을 사용하였기에 큰 어려움 없이 코드를 이해할 수 있을 것이다.

 

 

2. 코드

 탈컴공이 짠거이니 누구나 알 수 있을 것이다. 

from selenium import webdriver from html_table_parser import parser_functions as parser from bs4 import BeautifulSoup import pandas as pd driver = webdriver.Chrome('./chromedriver') driver.get('https://etk.srail.kr/cmc/01/selectLoginForm.do?pageId=TK0701000000') while(True): print("After Login, Do you go to the page with the seat you want? (y/n)") a= input() if( a == 'y'or a=='Y'): break driver.implicitly_wait(1) doc = driver.page_source soup = BeautifulSoup(doc,'html.parser') temp = soup.find_all('table') p=parser.make2d(temp[0]) print(p[0][0]) print("No." + "\t" + p[1][2] + "\t" + p[1][3] + "\t" + p[1][4] + "\t : 호실") count=1 select=[] for i in range(2,len(p)): if(p[i][5] == "매진"): print(str(count) + "\t" + p[i][2] + "\t" + p[i][3] + "\t" + p[i][4] + "\t : 특실") select.append([ i, 1,False]) count += 1 if(p[i][6] == "매진"): print(str(count) + "\t" + p[i][2] + "\t" + p[i][3] + "\t" + p[i][4] + "\t : 일반실") #i 1 == 2 select.append([ i, 2,False]) count += 1 print("\nInput Num you want.(ex: 1,4)") a=input() aa=a.split(",") for i in aa: select[int(i)-1][2]=True # while(True): driver.refresh() driver.implicitly_wait(2) for i in select: if(i[2] == True): txt="" if (i[1] == 1): txt = driver.find_element_by_xpath("//*[@id=\"result-form\"]/fieldset/div[6]/table/tbody/tr["+str(i[0]-1)+"]/td[6]/a/span").text a = driver.find_element_by_xpath("//*[@id=\"result-form\"]/fieldset/div[6]/table/tbody/tr["+str(i[0]-1)+"]/td[6]/a") elif (i[1] == 2): txt = driver.find_element_by_xpath("//*[@id=\"result-form\"]/fieldset/div[6]/table/tbody/tr["+str(i[0]-1)+"]/td[7]/a/span").text a = driver.find_element_by_xpath("//*[@id=\"result-form\"]/fieldset/div[6]/table/tbody/tr[" + str(i[0]-1) + "]/td[7]/a") if(txt=="예약하기"): a.click() break