使用selenium控制光猫黑名单

需求

使用selenium无头模式对光猫中的mac地址黑名单进行控制

步骤

1.登陆光猫
2.点击对应页面
3.填写mac地址

恢复步骤

1.登陆光猫
2.点击对应页面
3.删除黑名单

代码

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options
import time
import sys

con = sys.argv[1]


chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument('--headless')
chrome_options.add_argument('--disable-gpu')
chrome_options.add_argument('--no-sandbox') # 这个配置很重要

def deny():
    driver = webdriver.Chrome(chrome_options=chrome_options,executable_path='/home/chromedriver')
    driver.get('http://192.168.1.1/') #按实际填写ip

    driver.find_element('id','Frm_Password').send_keys('xxxxx',Keys.ENTER) #按实际输入密码

    driver.find_element('id','menu3').click()
    time.sleep(1)

    driver.switch_to.frame('mainFrame')
    #按实际填写mac
    driver.find_element('id','smMACFilter').click()
    driver.find_element('id','mac1').send_keys('91')
    driver.find_element('id','mac2').send_keys('62')
    driver.find_element('id','mac3').send_keys('a1')
    driver.find_element('id','mac4').send_keys('13')
    driver.find_element('id','mac5').send_keys('91')
    driver.find_element('id','mac6').send_keys('f4')
    driver.find_element('id','Btn_Add').click()
    time.sleep(3)
    driver.close()
def permit():
    driver = webdriver.Chrome(chrome_options=chrome_options,executable_path='/home/chromedriver')
    driver.get('http://192.168.1.1/')
    driver.find_element('id','Frm_Password').send_keys('xxxxxx',Keys.ENTER)
    driver.find_element('id','menu3').click()
    time.sleep(1)
    driver.switch_to.frame('mainFrame')
    driver.find_element('id','smMACFilter').click()
    driver.find_element('id','Img_Delete0').click()
    time.sleep(3)
    driver.close()
def main(con):
    if con == 'permit':
        permit()
    if con == 'deny':
        deny()
main(con)