【Selenium】Windows平台使用python自动登陆网关

#coding=utf8
# ========================================================
#   Copyright (C) 2017 All rights reserved.
#   
#   filename : CDLogin.py
#   author   : chendian@baidu.com
#   date     : 2017-08-29
#   desc     : Auto_Login Test
# ======================================================== 

import os
import time
import selenium
from selenium import webdriver

import platform
# "Windows" or "Linux"
OS_TYPE = platform.system()

def auto_connect(un, pw):
    print('open a phantomJS browser...\n')
    try:
        browser = webdriver.PhantomJS()
    except selenium.common.exceptions.WebDriverException as e:
        # Message: 'phantomjs' executable needs to be in PATH.
        print("Current Version only contains Windows PhantomJS.")
        print("You need phantomJS for your own platform as <%s>" % OS_TYPE)
        print("Please visit http://phantomjs.org/download.html for downloading")
        print("Download and unzip and find PhantomJS, copy it beside this python file.")
        print("For further info, visit http://blog.csdn.net/okcd00/article/details/77677326#t10")
        os.system('pause')
        exit()
    
    browser.get('http://159.226.39.22')

    username, password = un, pw

    browser.find_element_by_id('username').send_keys(username)
    browser.find_element_by_id('password').send_keys(password)
    #browser.find_element_by_id('button').click()
    browser.find_element_by_id('form2').submit()

    print('waiting 10s for posting...')
    time.sleep(10)
    browser.close()

def read_conf():
    un = ''
    pw = ''
    ps = ''
    try:
        with open('user.conf', 'r') as file:
            for line in file.readlines():
                line = line.strip()
                if line == '':
                    continue
                if line[0] == '#':
                    continue
                if line.split('=')[0].strip() == 'username':
                    un = line.split('=')[1].strip()
                elif line.split('=')[0].strip() == 'password':
                    pw = line.split('=')[1].strip()
                elif line.split('=')[0].strip() == 'pingsite':
                    ps = line.split('=')[1].strip()
                else:
                    print('user.conf is a error format. \n')
    except IOError as e:
        print('cannot read user.conf file--' + str(e) + ', please create user.conf file. \n')
    return un, pw, ps

def is_connect(ps):
    if OS_TYPE == 'Windows':
        flag = os.system('ping ' + ps)
    elif OS_TYPE == 'Linux':
        flag = os.system('ping -c 3 ' + ps)
    if flag == 0:
        return True
    return False

def con_connect(un, pw, ps):
    if is_connect(ps):
        return 1
    else:
        auto_connect(un, pw)
        if is_connect(ps):
            print('\nConnect internet success.\n')
        else:
            print('\nConnect internet Failed.\n')
        return 1


if __name__ == '__main__':
    print("starting...")
    os.system('MODE con: COLS=128 LINES=32 & TITLE Welcome to CDLogin')
    
    from logo import show_logo
    # show_logo()
    os.system('cls')
    
    time.sleep(1)
    os.system("color 0A")
    os.system('MODE con: COLS=64 LINES=16 & TITLE CD_Auto_Login is working')
    
    un, pw, ps = read_conf()
    
    while 1:
        print('=== current time ===')
        os.system("DATE /T & TIME /T")
        
        print('If you want to stop, input Ctrl+C to break.')
        con_connect(un, pw, ps)
        
        print('\nNext check in 10 mins ...\n')
        time.sleep(10*60) #Test
        os.system('cls')
    

#coding=utf8
import os
import time
import random
import urllib
import urllib2
import platform
import cookielib

uname = "test" # Change to your username
passwd = "123456" # Change to your password
ip = "10.63.0.222" # Modify this to your own ip 
interval_time = 5*60


def login(uname, passwd):
    cj = cookielib.CookieJar()
    opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
    opener.addheaders = [('User-agent','Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)')]
    urllib2.install_opener(opener)
    data = {
        "action": "login",
        "username": uname,
        "password": passwd,
        "ac_id": 1,
        "user_ip": ip,
        "ip": ip,
        "nas_ip": '',
        "user_mac": '',
        "save_me": 0,
        "type": 1,
        "n": 100,
        "ajax": 1
    }
    req = urllib2.Request("http://159.226.39.22/include/auth_action.php", urllib.urlencode(data))
    req.add_header("Referer","http://159.226.39.22/")
    try:
        resp = urllib2.urlopen(req)
        res = resp.read()
        print res
        if 'login_ok' in res:
            return True
        else:
            return False
    except:
        print 'log in error'
        return False


def logout(uname, passwd):
    cj = cookielib.CookieJar()
    opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
    opener.addheaders = [('User-agent','Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)')]
    urllib2.install_opener(opener)
    data = {
        "action": "logout",
        "username": uname,
        "password": passwd,
        "ajax": 1
    }
    req = urllib2.Request("http://159.226.39.22/include/auth_action.php", urllib.urlencode(data))
    req.add_header("Referer","http://159.226.39.22/")
    try:
        resp = urllib2.urlopen(req)
        res = resp.read()
        print res
        if "网络已断开" in res or "未曾连接到网络" in res:
            return True
        else:
            return False
    except:
        print 'log out error'
        return False


def internet_on():
    os_type = platform.system()
    if os_type == "Windows":
        flag = os.system('ping www.baidu.com')
    else:
        flag = os.system('ping -c 3 -t 3 www.baidu.com')
    return flag == 0


def logout_login():
    if logout(uname, passwd):
        print 'logout at: ', time.asctime()
        time.sleep(5)
        if login(uname, passwd):
            print 'login at: ', time.asctime()
        else:
            print 'login fails at: ', time.asctime()
    else:
        print 'logout fails at: ', time.asctime()


if __name__ == '__main__':
    if uname == "lihongwei_test":
        print "please set username, password and ip!"
    else:
        while True:
            if not internet_on():
                logout_login()
            time.sleep(interval_time)