fangying7
# -*- coding: utf-8 -*-
import sys

import win32api
import win32con
import win32event
import win32service
import win32serviceutil
import servicemanager
import logging
import os
from http.server import HTTPServer, CGIHTTPRequestHandler


class HTTPFilerServer(win32serviceutil.ServiceFramework):
    _svc_name_ = "HTTPFileServer"
    _svc_display_name_ = "HTTP File Server"
    _svc_description_ = "HTTP File Server"
    _svc_data_dir = \'D:\\ScreenRecorder\'

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

    def _getLogger(self):

        logger = logging.getLogger(\'[HTTPFileServer]\')
        dirpath = os.path.abspath(self._svc_data_dir)
        handler = logging.FileHandler(os.path.join(dirpath, self._svc_name_ + ".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):
        self.ReportServiceStatus(win32service.SERVICE_START_PENDING)
        try:
            self.ReportServiceStatus(win32service.SERVICE_RUNNING)
            self.logger.info(\'HTTP File Server is Starting ...\')
            self.start()
            self.logger.info(\'HTTP File Server Started\')
            import time
            time.sleep(3)
            os.chdir(self._svc_data_dir)
            httpd = HTTPServer((\'\', 8080), CGIHTTPRequestHandler)
            httpd.serve_forever()

            win32event.WaitForSingleObject(self.stop_event, win32event.INFINITE)
            self.logger.info(\'HTTP File Server Shutdown\')
        except BaseException as e:
            self.logger.warn(\'Exception : %s\' % e)
            self.SvcStop()

    def SvcStop(self):
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        self.logger.info(\'HTTP File Server is Stopping ...\')
        self.stop()
        self.logger.info(\'HTTP File Server Service Stopped\')
        win32event.SetEvent(self.stop_event)
        self.ReportServiceStatus(win32service.SERVICE_STOPPED)

    def start(self):
        if not os.path.exists(self._svc_data_dir):
            os.mkdir(self._svc_data_dir)
            win32api.SetFileAttributes(self._svc_data_dir, win32con.FILE_ATTRIBUTE_HIDDEN)
        else:
            pass

    def stop(self):
        pass

    def log(self, msg):
        servicemanager.LogInfoMsg(str(msg))


if __name__ == "__main__":
    if len(sys.argv) == 1:
        servicemanager.Initialize()
        servicemanager.PrepareToHostSingle(HTTPFilerServer)
        servicemanager.StartServiceCtrlDispatcher()
    else:
        win32serviceutil.HandleCommandLine(HTTPFilerServer)

分类:

技术点:

相关文章:

  • 2021-11-11
  • 2021-12-07
  • 2018-05-12
  • 2021-04-03
  • 2021-12-19
  • 2021-11-08
  • 2021-04-11
  • 2021-12-07
猜你喜欢
  • 2021-12-02
  • 2021-12-16
  • 2021-11-10
  • 2021-05-31
  • 2022-01-19
  • 2021-11-07
  • 2021-11-17
相关资源
相似解决方案