前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python 写window服务(必须写在服务类里)

python 写window服务(必须写在服务类里)

作者头像
用户5760343
发布2019-08-06 10:24:50
5640
发布2019-08-06 10:24:50
举报
文章被收录于专栏:sktjsktj

encoding=utf-8

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 " # 服务的描述

代码语言:javascript
复制
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)

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019.08.02 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • encoding=utf-8
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档