Python重启深信服设备
cat > /sangfor/Pysangfor/sangfor_public.py<<EOF #!/bin/env python3# -*- coding:UTF-8 -*-
# _author:kaliarch
#导入模块
from pyvirtualdisplay import Display
from selenium import webdriver
import time
import os
import logging
#定义深信服重启类
class Glp_SangFor:
def __init__(self,logger):
self.logger = logger
self.logger.info("--------------start log----------------")
self.display = Display(visible=0,> self.display.start()
self.browser = webdriver.Firefox()
self.logger.info("start browser successfuly")
self.sangfor_url = "深信服公网url"
self.username = '深信服登录用户名'
self.password = '深信服登录密码'
def login(self):
self.browser.get(self.sangfor_url)
self.browser.implicitly_wait(5)
self.browser.find_element_by_name('user').send_keys(self.username)
self.browser.find_element_by_name('password').send_keys(self.password)
self.browser.find_element_by_class_name('buttons').click()
self.browser.implicitly_wait(5)
self.logger.info("loggin sangfor successfuly")
def client_reboot(self):
self.browser.find_element_by_id("ext-gen111").click()
print(self.browser.find_element_by_id("ext-gen111").text)
self.browser.implicitly_wait(15)
time.sleep(60)
self.logger.info("switch mainiframe start")
try:
print(self.browser.find_element_by_link_text("重启/重启服务/关机").text)
self.browser.find_element_by_link_text("重启/重启服务/关机").click()
self.browser.implicitly_wait(3)
self.browser.switch_to_frame("mainiframe")
self.browser.implicitly_wait(8)
time.sleep(10)
self.browser.find_element_by_xpath("//button[@id='ext-gen19']").click()
print(self.browser.find_element_by_xpath("//button[@id='ext-gen19']").text)
self.browser.implicitly_wait(10)
#self.browser.find_element_by_xpath("//button[@id='ext-gen42']").click()
print(self.browser.find_element_by_xpath("//button[@id='ext-gen42']").text)
except Exception as e:
self.logger.exception("reboot successful")
return 1
self.browser.close()
self.logger.info("browser close successful")
self.logger.info("--------------end log----------------")
return 0
#定义日志记录
class Glp_Log:
def __init__(self,filename):
self.filename = filename
def createDir(self):
_LOGDIR = os.path.join(os.path.dirname(__file__), 'publiclog')
print(_LOGDIR)
_TIME = time.strftime('%Y-%m-%d', time.gmtime()) + '-'
_LOGNAME = _TIME + self.filename
print(_LOGNAME)
LOGFILENAME = os.path.join(_LOGDIR, _LOGNAME)
print(LOGFILENAME)
if not os.path.exists(_LOGDIR):
os.mkdir(_LOGDIR)
return LOGFILENAME
print(LOGFILENAME)
def createlogger(self,logfilename):
logger= logging.getLogger()
logger.setLevel(logging.INFO)
handler = logging.FileHandler(logfilename)
handler.setLevel(logging.INFO)
formater = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formater)
logger.addHandler(handler)
return logger
#主函数调用
if __name__ == '__main__':
os.system("pkill firefox")
os.system("pkill Xvfb")
glploger = Glp_Log('public-***.log')
logfilename = glploger.createDir()
logger = glploger.createlogger(logfilename)
sangfor_oper = Glp_SangFor(logger)
sangfor_oper.login()
sangfor_oper.client_reboot()
EOF
页:
[1]