[태그:] python

python,selenium을 활용
skt Tworld 로그인해서 남은 데이터량을 가져오는걸 만들어봤습니다.

통신사 자체적으로 sms로 사용량 안내를 해주기도 하고
앱이나 위젯으로도 확인할 수 있지만
어르신들은 익숙하지 않으신지  한번씩 사용량을 오버해서
요금이 더 나오는 경우가 있더라구요

zabbix와 연동해서 가족들 남은 데이터량 확인해서
모자라는 사람은 데이터선물하기를 통해 데이터를 나눠줄 생각입니다.

#!/usr/local/bin/python3.6
from selenium import webdriver
import time
import sys
import os

class Tworld():
    driver = ""
    user_id = ""
    user_pass = ""

    def __init__(self):
        #options = webdriver.ChromeOptions()
        #options.add_argument('headless')
        #options.add_argument("lang=ko_KR")
        #options.add_argument(
        #    "user-agent=Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36")

        #self.driver = webdriver.Chrome('./chromedriver', chrome_options=options)
        self.driver = webdriver.PhantomJS()
        self.driver.implicitly_wait(3)

    def login(self, user_id, user_pass):
        try:
            self.user_id = user_id
            self.user_pass = user_pass

            self.driver.get('http://www.tworld.co.kr/poc/html/main/MA.html')

            self.driver.switch_to.frame('login_area')
            self.driver.find_element_by_xpath('//*[@id="authLogin"]').click()
            self.driver.switch_to.window('loginPop')
            self.driver.find_element_by_id('userId').send_keys(self.user_id)
            self.driver.find_element_by_id('password').send_keys(self.user_pass)
            self.driver.find_element_by_id('authLogin').click()
            time.sleep(1)
            return True

        except:
            return False

    def get_available_data_in_mb(self):
        self.driver.switch_to.window('')
        self.driver.switch_to.frame('login_area')
        self.driver.switch_to.frame('freebillIframe')
        data = self.driver.find_element_by_xpath('/html/body/div[1]/div/div[1]/div[2]/span[1]').text
        value,standard = data.split(" ")

        if standard == "MB":
            data = value.replace(",", "")
        elif standard == "GB":
            data = float(value) * 1000
        else:
            data = "999"
        return data

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("arguments required.")
        exit(0)

    user_id = sys.argv[1]
    user_pass = sys.argv[2]

    tworld = Tworld()
    if tworld.login(user_id, user_pass):
        print(tworld.get_available_data_in_mb())