[Selenium] Selenium 을 이용하여 복권 구매 자동화 만들어보기 #4

-

 이전 글에서 로그인 기능을 만들어 보았습니다. 이번 글에서는 복권 구매까지 진행을 해보겠습니다. 다음글에서는 마지막 코드 리펙토링을 진행하도록 하겠습니다. 

이 글을 끝에 구매 버튼이 있는데 진짜로 구매하시려면 예치금을 충전하시면 가능합니다.


복권 구매는 단계를 나누어 보면 아래와 같습니다.

1. 메인 페이지에서 복권 구매 버튼 클릭(클릭 시 팝업 열림)

2. 복권 구매 번호 랜덤으로 발행(자동번호 발행 기능이 이미 있지만 코드 예제를 위해 수동으로 진행)

3. 구매 버튼 클릭


단계를 나누어보면 새로 배울 개념이 없지만 여기서 문제라고하면 복권 구매 화면이 iframe 형식으로 구성이 되어있다는 것입니다. iframe 으로 되어있는 경우 selenium 을 통해 현재 frame 을 이동시켜줘야 원하는 element 등에 접근 가능합니다. 


우선 몇 개의 함수를 먼저 작성하겠습니다.


script 실행하기

def execute_script(script):
    driver.execute_script(script)


윈도우 중에서 마지막에 열린 윈도우로 현재 윈도우 변경하기

def switch_to_last_opened_window():
    driver.switch_to.window(driver.window_handles[-1])


iframe 으로 현재 프레임 변경하기

def switch_to_iframe(iframe):
    WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.TAG_NAME, "iframe"))
    )

    driver.switch_to.frame(iframe)


WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.TAG_NAME, "iframe"))
    )

의 코드가 필요한 이유는 iframe 이 나중에 로드가 되어 약간의 딜레이를 주지 않고 바로 find~ 등을 진행할 경우 때때로 실패하는 것을 확인할 수 있습니다. 그렇기에 넉넉히 10초간 iframe tag 가 나오기를 기다리는 코드를 작성한 것입니다.


로또 번호 6개 발행하기

def pick_numbers():
    picks = set([])
    while 6 != len(picks):
        n = random.randint(LOTTO_MIN_NUM, LOTTO_MAX_NUM)
    picks.add(n)

    return picks

화면 캡처하기

def save_screenshot():
    driver.save_screenshot(f'{datetime.now().strftime("%Y%m%d_%H%M%S")}.png')

이제 위 함수를 참고하여 코드를 작성해보겠습니다.

지난 글처럼 로그인을 진행하게 되면 다시금 팝업이 나타납니다. (팝업은 때 마다 다르니 팝업이 없으실 수 있습니다) 


팝업 닫기

sleep(1)
close_windows_if_not_main_url('common.do?method=main')


sleep(1) 을 추가한 이유는 팝업이 여러개 나타날 경우 팝업이 제대로 안 닫히는 경우가 있습니다. 그러므로 1초 동안 팝업이 다 열리기를 기다리는 것 입니다.


복권 구매 버튼 클릭 시 발생되는 script 실행

execute_script('goLottoBuy(2);')






저는 개인적으로 가능하다면 script 을 실행하는 편입니다. 그 이유는 selenium 을 통해 find~ 보다 변경이 적기 때문입니다. 또한 element 가 화면에 보이지 않을 경우 find 나 click 을 하려하면 에러를 발생시키는데 script 실행은 그에 비해 유연하기 때문입니다. 물론 상황에 따라서 적합한게 다를 것 입니다.


복권 구매 팝업 윈도우로 현재 윈도우 이동

switch_to_last_opened_window()



복권 번호 클릭을 위해 현재 frame 이동

switch_to_iframe(driver.find_elements(By.TAG_NAME, 'iframe')[0])




복권 번호 6개 클릭 후 번호 선택 확인 버튼 클릭

for num in pick_numbers():
    execute_script(f'$("#check645num{num}").prop("checked", true);')

click(driver.find_element(By.ID, 'btnSelectNum'))

각 숫자는 checkbox 의 형식으로 되어있습니다. id 의 규칙을 보면 check645num 뒤에 1~45가 붙어있는 것임을 알 수 있습니다. 그렇기에 랜덤으로 6개의 숫자를 단순히 check645num 뒤에 붙여 checked 속성을 true 로 부여하였습니다.






구매하기 버튼 클릭

click(driver.find_element(By.ID, 'btnBuy'))
execute_script('closepopupLayerConfirm(true);')


구매하기 버튼을 클릭하면 한 번 더 묻는 컨펌 다이얼로그가 나옵니다. 여기서 저는 해당 확인 버튼을 클릭하기 보다 js script 을 실행시키도록 하였습니다.


위 코드까지 실행하면 아래와 같이 구매 완료 화면이 나옵니다.



구매내역 확인 캡처

구매만 하고 끝낼 수 있지만 저 구매내역 확인을 캡처해서 저장하고 싶을 수 있습니다. 그렇기에 캡처하고 브라우저 종료하도록 하겠습니다.

sleep(2)
save_screenshot()

driver.quit()



전체 코드
from datetime import datetime
import random
from time import sleep

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By

from dotenv import load_dotenv
import os

from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

load_dotenv()

LOTTO_MIN_NUM = 1
LOTTO_MAX_NUM = 45


def create_driver():
    return webdriver.Chrome(service=Service('./driver/chromedriver_mac64_m1_97'))


driver = create_driver()
driver.get('https://dhlottery.co.kr/common.do?method=main')


def get_main_handle(main_url):
    for window_handle in driver.window_handles:
        driver.switch_to.window(window_handle)

        print(f'현재 핸들러 name : {driver.current_window_handle}')
        print(f'현재 url :  {driver.current_url}')
        print('')

        if main_url in driver.current_url:
            return window_handle
    raise '메인 핸들러 찾지 못함'


def close_windows_if_not_main_url(main_url):
    main_handle = get_main_handle(main_url)

    for window_handle in driver.window_handles:
        if main_handle != window_handle:
            driver.switch_to.window(window_handle)
            driver.close()

    driver.switch_to.window(main_handle)


def send_keys(element, text):
    element.send_keys(text)


def click(element):
    element.click()


def execute_script(script):
    driver.execute_script(script)


def switch_to_last_opened_window():
    driver.switch_to.window(driver.window_handles[-1])


def switch_to_iframe(iframe):
    WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.TAG_NAME, "iframe"))
    )

    driver.switch_to.frame(iframe)


def pick_numbers():
    picks = set([])
    while 6 != len(picks):
        n = random.randint(LOTTO_MIN_NUM, LOTTO_MAX_NUM)
        picks.add(n)

    return picks


def save_screenshot():
    driver.save_screenshot(f'{datetime.now().strftime("%Y%m%d_%H%M%S")}.png')


close_windows_if_not_main_url('common.do?method=main')

click(driver.find_element(By.XPATH, '/html/body/div[1]/header/div[2]/div[2]/form/div/ul/li[1]/a'))

send_keys(driver.find_element(By.ID, 'userId'), os.environ.get('ID'))
send_keys(driver.find_element(By.NAME, 'password'), os.environ.get('PW'))
send_keys(driver.find_element(By.NAME, 'password'), Keys.ENTER)

sleep(1)

close_windows_if_not_main_url('common.do?method=main')

execute_script('goLottoBuy(2);')

sleep(1)

switch_to_last_opened_window()

switch_to_iframe(driver.find_elements(By.TAG_NAME, 'iframe')[0])

for num in pick_numbers():
    execute_script(f'$("#check645num{num}").prop("checked", true);')

click(driver.find_element(By.ID, 'btnSelectNum'))

click(driver.find_element(By.ID, 'btnBuy'))
execute_script('closepopupLayerConfirm(true);')

sleep(2)
save_screenshot()

driver.quit()




댓글

댓글 쓰기

이 블로그의 인기 게시물

[INTELLIJ] 인텔리제이 폴더 안 보이고 파일만 보이는 버그 해결 방법

-
이미지
오늘의 포스팅은 인텔리제이에서 폴더 안 보이고 파일만 보이는 버그 해결 방법에 대한 것입니다. 버그 상황 파일은 보이나 폴더가 보이지 않습니다. 해결 방법 1. File -> Project Structure 클릭을 합니다. 2. Modules 클릭 합니다. 3. + 을 눌러 Import Module 클릭 합니다. 4. 자신의 프로젝트 Root 폴더 를 선택하여줍니다. 5. Next 해줍니다. 6. Ok 해줍니다. 7. 짜잔! 이 방법이 안된다면 루트 디렉토리 안에 .idea 폴더를 삭제했다가 재실행 시 오류가 해결된다고도 합니다! 도움이 되셨기를 바라며 마치겠습니다. 다음에는 더 좋은 포스팅을 하도록 하겠습니다. 감사합니다.
자세한 내용 보기

[생활] 짝눈(부동시)로 신체검사 4급(공익) 받은 방법.

-
저는 2000년도 태어나서 2019년에 신체검사 대상자입니다. 저는 초등학교 2학년부터 짝눈인 것을 알았습니다. 흔히 말하는 시력으로는 좌안 1.0 우안0.1 이 나왔습니다. 사실 이때 약시 치료를 받았어야했는데 의사의 오진?!으로 치료 시기를 놓쳤습니다. 운이 좋게?!도 지금 20살이 되도록 좌안은 0.8 우안 0.01 이 나옵니다. 디옵토로 말하자면 좌안 -1.00 우안 -6.75 입니다. 병무청 신체검사 기준  을 보았을 때 양안 차이가 -5.0 이상 차이날 때 4급이라 저는 4급에 해당했습니다. 하지만 병무청 신체검사로 판정이 나지 않기때문에 병원에서 검사를 받고 서류를 제출해야합니다. 일단 저는 약시로 병무용 진단서로 진찰을 받았습니다. 진단서는 보험 처리가 안되기 때문에 비쌉니다. 약시 병무청 진단 비용은 약 13만원 정도 사용했습니다. 저는 생각이 짧았는지 약시가 부동시로 오기때문에 좀 더 넓은 범위가 약시라 약시 진단을 끊었습니다. 그러나 병무청에서는 약시는 3년전 기록이 필요하고 부동시로 진단서를 발급 받아야 한다는 것이였습니다. 돈 13만원이 너무 아깝지만 현역보단 낫겠지라는 생각으로 동네 안과에 갔습니다. 부동시는 병무용 진단서가 아니라 근처 병원에서 4가지 검사 후 의무기록지 사본만 제출하면 된다고합니다. 다행히 의무기록지 사본을 발급 받는 비용은 약 4만원 정도로 병사용 진단서보다는 저렴했습니다. 의무기록지 사본을 병무청 등기로 보낸 후 이틀 후 4급 판정을 우편으로 통지 받았습니다. 가끔... 저를 부러워 하시는 분들이 있는데 이렇게 짝눈이 심하면 한쪽눈으로 생활하게 되고. 한쪽눈으로 생활을 하다보니 두통과 어지러움을 굉장히 자주 느낍니다. 일상생활이 안될 때도 있습니다. 그러니 부러워하지 않으셔도 됩니다!! 이상 짝눈으로 4급 판정 받은 과정이였습니다.
자세한 내용 보기

[스프링부트] @Mapper import 실패시(빨간줄) 해결 방법

-
@Mapper 찾지못하는 오류 현상 해결 방법 오류 사항 import org.apache.ibatis.annotations.Mapper ; 위에 import문에서 Mapper에서만 빨간 밑줄 이 그어진다면 gradle 속성을 살펴보아야한다. 아래와 같이  gradle 속성을 바꾸어주었더니 IDE에서 인식이 가능하였다. 기존 compile 일부 compile "org.mybatis:mybatis:3.3.0" compile "org.mybatis:mybatis-spring:1.2.3" 추가한 compile 일부 compile "org.springframework.boot:spring-boot-starter-jdbc" compile( "org.mybatis.spring.boot:mybatis-spring-boot-starter:1.3.2" ) 속성만 바꿔준다고해서 적용 이 되지않는다. gradle 속성을 바꿔준다음 gradle Tool View에서 Refersh all gradle projects 를 해줘야 적용이 된다.
자세한 내용 보기