【python】使用python写windows服务

背景

    运维windows服务器的同学都知道，windows服务器进行批量管理的时候非常麻烦，没有比较顺手的工具，虽然saltstack和ansible都能够支持windows操作，但是使用起来总感觉不太舒服，比如ansible使用的winrm进行远程操作，需要提前在windows上进行设置，并且要求操作系统版本和powershell版本，根据个人使用经验，经常存在链接不上，连接慢，或者推送失败的问题，较高的失败率对于自动化运维体系的建立不利，所以与其受制于人，不如自己动手写一个agent，完全自己控制。

技术平台

    服务器：windows服务器，测试过win2008r2、win2012、win7和win10.其他版本不知道怎么样。

    开发语言：python

    python模块：pywin32（python中的windows库），pyinstaller（用于将py文件生成exe程序），flask（web服务器）。以上三个模块都可以通过pip安装。以及一些python内置模块

    优势：完全自己控制，根据自己需要开发功能。服务器端不需要安装任何其他依赖（不需要安装python环境，exe自带运行环境）

流程结构

代码实现

agent的实现代码如下，代码有详细的注释，就不解释了。

# -*- coding:utf-8 -*-
import win32serviceutil
import win32service
import win32event
from flask import Flask
from flask import request
import sys
import os
import zipfile
import requests
import shutil
import re
import time
#设置编码
reload(sys)
sys.setdefaultencoding('utf-8')
################################################################
##########自定义函数区##########################################
################################################################
#下载文件函数
def get_url_file(url):
    #下载到临时目录，如果临时目录不存在就创建一个
    download_path = "D:\tmp\%s" % (time.time())
    if os.path.exists(download_path):
        pass
    else:
        os.makedirs(download_path)
    r = requests.get(url, stream=True)
    #解析文件名，如url为：http://www.ftp.com/test.zip那么文件名就是test.zip，并下载和写入文件
    filename = "%s\%s" % (download_path, url.split('/')[-1])
    with open(filename, "wb") as pdf:
        for chunk in r.iter_content(chunk_size=1024):
            if chunk:
                pdf.write(chunk)
        return {'filename':filename,'download_path':download_path}


def config_zabbix(hostname):
    #复制一份C:\zabbix\zabbix_agentd_win.conf.bak的zabbix模板文件，并修改其中的Hostname=new为新的hostname，然后生成新的配置文件。
    try:
        open('C:\zabbix\zabbix_agentd_win.conf', 'w').write(re.sub(r'Hostname=new', 'Hostname=%s' % hostname, open('C:\zabbix\zabbix_agentd_win.conf.bak').read()))
    except:
        return "error"
    # 重启zabbix agent服务
    service_list = os.popen('net start').read()
    if service_list.find('Zabbix Agent') == -1:
        try:
            os.system('net start "Zabbix Agent"')
        except:
            return "error"
    else:
        try:
            os.system('net stop "Zabbix Agent')
        except:
            return "error"
        try:
            os.system('net start "Zabbix Agent')
        except:
            return "error"
        return "ok"


# 加压缩文件函数
def un_zip(file_name,dest_path):
    zip_file = zipfile.ZipFile(file_name)
    file_pre = file_name.split('\')[-1].split('.')[0]
    for names in zip_file.namelist():
        zip_file.extract(names, dest_path)
    zip_file.close()
    return (dest_path + '\' + file_pre)
###################################################################
#############自定义函数区结束######################################
###################################################################

#windows服务中显示的名字
class zlsService(win32serviceutil.ServiceFramework):
    _svc_name_ = 'zls_agent' ###可以根据自己喜好修改
    _svc_display_name_ = 'zls_agent'  ###可以根据自己喜好修改
    _svc_description_ = 'zls_agent'  ###可以根据自己喜好修改


    def __init__(self,args):
        win32serviceutil.ServiceFramework.__init__(self,args)
        self.stop_event = win32event.CreateEvent(None, 0, 0, None)
        self.run = True

    def SvcDoRun(self):
        app = Flask(__name__)
        ##############################################################
        #########flask路由设置区，自定义功能也放在这里################
        ##############################################################
        #推送文件服务，比如传入一个url和目标地址，装有本agent的客户端就会下载这个url并把文件放在指定位置
        @app.route('/transferfile', methods=['GET', 'POST'])
        def transferfile():
            if request.method == "GET":
                url = request.args.get('url')
                dest = request.args.get('dest')
                try:
                    down_data = get_url_file(url)
                    filename = down_data['filename']
                    download_path = down_data['download_path']
                except BaseException, e:
                    return "download url file error : %s" % e
                # 加压缩文件
                try:
                    filepath = un_zip(filename,download_path)
                except BaseException, e:
                    return "un zip error : %s" % e
                # 删除压缩文件
                try:
                    os.remove(filename)
                except:
                    pass
                # 复制文件内容到指定目录
                try:
                    shutil.copytree('%s' % filepath.replace('\', '\\'), '%s' % dest)
                except BaseException, e:
                    return "copy file error: %s" % e
                # d删除解压缩文件夹
                try:
                    shutil.rmtree(filepath.replace('\', '\\'))
                except:
                    pass
                return "ok"
        #修改配置文件路由，用来远程修改zabbix配置文件，比如curl http://server_ip:50000/zabbix?hostname="zabbix_agent"就会把位于服务器上的zabbix_agentd配置文件中的Hostname修改为Hostname=zabbix_agent。
        @app.route('/zabbix', methods=['GET', 'POST'])
        def config_zabbix_func():
            if request.method == "GET":
                hostname = request.args.get('hostname')
                result = config_zabbix(hostname)
                return result
        #测试路由，用来测试agent是否正在运行，返回ok表示正在运行
        @app.route('/', methods=['GET', 'POST'])
        def test():
            if request.method == "GET":
                return "ok"
        #使用flask自带的web服务器，监听本地所有地址的50000端口
        ##############################################################
        #############自定义功能区结束#################################
        ##############################################################
        app.run(host='0.0.0.0', port=50000)
    def SvcStop(self):
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        win32event.SetEvent(self.stop_event)
        self.ReportServiceStatus(win32service.SERVICE_STOPPED)
        self.run = False

if __name__ == '__main__':
    import sys
    import servicemanager
    if len(sys.argv) == 1:
        try:
            evtsrc_dll = os.path.abspath(servicemanager.__file__)
            servicemanager.PrepareToHostSingle(zlsService) #如果修改过名字，名字要统一
            servicemanager.Initialize('zlsService',evtsrc_dll) #如果修改过名字，名字要统一
            servicemanager.StartServiceCtrlDispatcher()
        except win32service.error as details:
            import winerror
            if details == winerror.ERROR_FAILED_SERVICE_CONTROLLER_CONNECT:
                win32serviceutil.usage()
    else:
        win32serviceutil.HandleCommandLine(zlsService) #如果修改过名字，名字要统一

生成exe程序

安装完成pyinstaller以后，打开命令行窗口

输入：pyinstaller -F windows_service.py

说明：-F表示生成一个单一的exe文件。其他选项使用pyinstaller --help查看。

执行完成后，提示生成exe程序

找到这个程序

将程序复制到测试服务器上，安装

以管理员身份打开cmd，中输入 windows_servcie.exe install

提示安装成功

启动服务，然后设置成自动启动

 测试，浏览器中访问IP:50000端口，返回ok表示agent运行正常

后续：

想要卸载服务，可以使用

 重要说明：

以上的测试，均在测试环境中进行，不要用在正式环境，因为这个agent没有加密通信，存在被人攻击的危险！！！

实际使用过程中，一定要对通信进行加密，可以使用https或者在命令中加入token，对token进行认证，只有正确的token才执行命令，或者使用rsa证书加密，这里就不再举例。

此agent只是给了三个示例性应用，实际用的时候，可以根据自己需要在函数区和路由定义区加入自己的程序，比如把这个agent做成监控系统，获取本地信息穿给服务端，在此抛砖引玉，剩下的就看自己的发挥了。