import win32serviceutil import win32service import win32event import os import logging import inspect import servicemanager import sys
class RenewLinuxOdpCookie(win32serviceutil.ServiceFramework): svc_name = "RenewLinuxOdpCookie" # 服务名 svc_display_name = "RenewLinuxOdpCookie" # 服务在windows系统中显示的名称 svc_description = "RenewLinuxOdpCookie " # 服务的描述
def getcookiefromchrome(self, host='wspms.haplat.net'):
import sqlite3
from win32.win32crypt import CryptUnprotectData
cookiepath = r"C:\Users\huangat\AppData\Local\Google\Chrome\User Data\Default\Cookies"
self.logger.info(cookiepath)
sql = "select host_key,name,encrypted_value from cookies where host_key='%s'" % host
with sqlite3.connect(cookiepath) as conn:
cu = conn.cursor()
cookies = {name: CryptUnprotectData(encrypted_value)[1].decode() for host_key, name, encrypted_value in
cu.execute(sql).fetchall()}
return cookies
def renewodpcookie(self, cookie):
import paramiko
tbjip = "10.8.0.35"
tbjport = 57891
tbjuser = "huangat"
tbjpass = "Lyw6778286@"
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(tbjip, tbjport, tbjuser, tbjpass)
ssh.exec_command("sed -i '/^Cookie/c{}' abc.sh".format(cookie))
def renewrun(self):
sid = self.getcookiefromchrome("odp.haplat.net").get("sid", "")
cookie = "Cookie:lang=zh-cn; theme=default; tenantCode=WS; _ga=GA1.2.1071081989.1552469516; UM_distinctid=16993b3b93a338-04034b46a28df4-b781636-100200-16993b3b93b352; grafana_sess=d4d49117819b91d2; grafana_user=huangat; grafana_remember=5440b2ea0bf1c4e633230e0482e840e91dd3df828655cd2d77aa265c; sid={}" \
.format(sid)
self.renewodpcookie(cookie)
def __init__(self, args):
win32serviceutil.ServiceFramework.__init__(self, args)
self.hWaitStop = win32event.CreateEvent(None, 0, 0, None)
self.logger = self._getLogger()
self.run = True
def _getLogger(self):
logger = logging.getLogger('[RenewLinuxOdpCookie]')
this_file = inspect.getfile(inspect.currentframe())
dirpath = os.path.abspath(os.path.dirname(this_file))
handler = logging.FileHandler(os.path.join(dirpath, "renewLinuxOdpCookie.log"))
formatter = logging.Formatter('%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(logging.INFO)
return logger
def SvcDoRun(self):
import time
self.logger.info("service is run....")
try:
self.renewrun()
except Exception as e:
self.logger.info(e)
while self.run:
self.logger.info("I am runing....")
time.sleep(2)
def SvcStop(self):
self.logger.info("service is stop....")
self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
win32event.SetEvent(self.hWaitStop)
self.run = False
if name == 'main': if len(sys.argv) == 1: evtsrc_dll = os.path.abspath(servicemanager.file) servicemanager.PrepareToHostSingle(RenewLinuxOdpCookie) servicemanager.Initialize('RenewLinuxOdpCookie', evtsrc_dll) servicemanager.StartServiceCtrlDispatcher() else: win32serviceutil.HandleCommandLine(RenewLinuxOdpCookie)