Uploaded Test files
This commit is contained in:
parent
f584ad9d97
commit
2e81cb7d99
16627 changed files with 2065359 additions and 102444 deletions
843
venv/Lib/site-packages/win32/lib/win32serviceutil.py
Normal file
843
venv/Lib/site-packages/win32/lib/win32serviceutil.py
Normal file
|
@ -0,0 +1,843 @@
|
|||
# General purpose service utilities, both for standard Python scripts,
|
||||
# and for for Python programs which run as services...
|
||||
#
|
||||
# Note that most utility functions here will raise win32api.error's
|
||||
# (which is == win32service.error, pywintypes.error, etc)
|
||||
# when things go wrong - eg, not enough permissions to hit the
|
||||
# registry etc.
|
||||
|
||||
import win32service, win32api, win32con, winerror
|
||||
import sys, pywintypes, os, warnings
|
||||
error = RuntimeError
|
||||
|
||||
def LocatePythonServiceExe(exeName = None):
|
||||
if not exeName and hasattr(sys, "frozen"):
|
||||
# If py2exe etc calls this with no exeName, default is current exe.
|
||||
return sys.executable
|
||||
|
||||
# Try and find the specified EXE somewhere. If specifically registered,
|
||||
# use it. Otherwise look down sys.path, and the global PATH environment.
|
||||
if exeName is None:
|
||||
if os.path.splitext(win32service.__file__)[0].endswith("_d"):
|
||||
exeName = "PythonService_d.exe"
|
||||
else:
|
||||
exeName = "PythonService.exe"
|
||||
# See if it exists as specified
|
||||
if os.path.isfile(exeName): return win32api.GetFullPathName(exeName)
|
||||
baseName = os.path.splitext(os.path.basename(exeName))[0]
|
||||
try:
|
||||
exeName = win32api.RegQueryValue(win32con.HKEY_LOCAL_MACHINE,
|
||||
"Software\\Python\\%s\\%s" % (baseName, sys.winver))
|
||||
if os.path.isfile(exeName):
|
||||
return exeName
|
||||
raise RuntimeError("The executable '%s' is registered as the Python " \
|
||||
"service exe, but it does not exist as specified" \
|
||||
% exeName)
|
||||
except win32api.error:
|
||||
# OK - not there - lets go a-searchin'
|
||||
for path in [sys.prefix] + sys.path:
|
||||
look = os.path.join(path, exeName)
|
||||
if os.path.isfile(look):
|
||||
return win32api.GetFullPathName(look)
|
||||
# Try the global Path.
|
||||
try:
|
||||
return win32api.SearchPath(None, exeName)[0]
|
||||
except win32api.error:
|
||||
msg = "%s is not correctly registered\nPlease locate and run %s, and it will self-register\nThen run this service registration process again." % (exeName, exeName)
|
||||
raise error(msg)
|
||||
|
||||
def _GetServiceShortName(longName):
|
||||
# looks up a services name
|
||||
# from the display name
|
||||
# Thanks to Andy McKay for this code.
|
||||
access = win32con.KEY_READ | win32con.KEY_ENUMERATE_SUB_KEYS | win32con.KEY_QUERY_VALUE
|
||||
hkey = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services", 0, access)
|
||||
num = win32api.RegQueryInfoKey(hkey)[0]
|
||||
longName = longName.lower()
|
||||
# loop through number of subkeys
|
||||
for x in range(0, num):
|
||||
# find service name, open subkey
|
||||
svc = win32api.RegEnumKey(hkey, x)
|
||||
skey = win32api.RegOpenKey(hkey, svc, 0, access)
|
||||
try:
|
||||
# find display name
|
||||
thisName = str(win32api.RegQueryValueEx(skey, "DisplayName")[0])
|
||||
if thisName.lower() == longName:
|
||||
return svc
|
||||
except win32api.error:
|
||||
# in case there is no key called DisplayName
|
||||
pass
|
||||
return None
|
||||
|
||||
# Open a service given either it's long or short name.
|
||||
def SmartOpenService(hscm, name, access):
|
||||
try:
|
||||
return win32service.OpenService(hscm, name, access)
|
||||
except win32api.error as details:
|
||||
if details.winerror not in [winerror.ERROR_SERVICE_DOES_NOT_EXIST,
|
||||
winerror.ERROR_INVALID_NAME]:
|
||||
raise
|
||||
name = win32service.GetServiceKeyName(hscm, name)
|
||||
return win32service.OpenService(hscm, name, access)
|
||||
|
||||
def LocateSpecificServiceExe(serviceName):
|
||||
# Given the name of a specific service, return the .EXE name _it_ uses
|
||||
# (which may or may not be the Python Service EXE
|
||||
hkey = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services\\%s" % (serviceName), 0, win32con.KEY_ALL_ACCESS)
|
||||
try:
|
||||
return win32api.RegQueryValueEx(hkey, "ImagePath")[0]
|
||||
finally:
|
||||
hkey.Close()
|
||||
|
||||
def InstallPerfmonForService(serviceName, iniName, dllName = None):
|
||||
# If no DLL name, look it up in the INI file name
|
||||
if not dllName: # May be empty string!
|
||||
dllName = win32api.GetProfileVal("Python", "dll", "", iniName)
|
||||
# Still not found - look for the standard one in the same dir as win32service.pyd
|
||||
if not dllName:
|
||||
try:
|
||||
tryName = os.path.join(os.path.split(win32service.__file__)[0], "perfmondata.dll")
|
||||
if os.path.isfile(tryName):
|
||||
dllName = tryName
|
||||
except AttributeError:
|
||||
# Frozen app? - anyway, can't find it!
|
||||
pass
|
||||
if not dllName:
|
||||
raise ValueError("The name of the performance DLL must be available")
|
||||
dllName = win32api.GetFullPathName(dllName)
|
||||
# Now setup all the required "Performance" entries.
|
||||
hkey = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services\\%s" % (serviceName), 0, win32con.KEY_ALL_ACCESS)
|
||||
try:
|
||||
subKey = win32api.RegCreateKey(hkey, "Performance")
|
||||
try:
|
||||
win32api.RegSetValueEx(subKey, "Library", 0, win32con.REG_SZ, dllName)
|
||||
win32api.RegSetValueEx(subKey, "Open", 0, win32con.REG_SZ, "OpenPerformanceData")
|
||||
win32api.RegSetValueEx(subKey, "Close", 0, win32con.REG_SZ, "ClosePerformanceData")
|
||||
win32api.RegSetValueEx(subKey, "Collect", 0, win32con.REG_SZ, "CollectPerformanceData")
|
||||
finally:
|
||||
win32api.RegCloseKey(subKey)
|
||||
finally:
|
||||
win32api.RegCloseKey(hkey)
|
||||
# Now do the "Lodctr" thang...
|
||||
|
||||
try:
|
||||
import perfmon
|
||||
path, fname = os.path.split(iniName)
|
||||
oldPath = os.getcwd()
|
||||
if path:
|
||||
os.chdir(path)
|
||||
try:
|
||||
perfmon.LoadPerfCounterTextStrings("python.exe " + fname)
|
||||
finally:
|
||||
os.chdir(oldPath)
|
||||
except win32api.error as details:
|
||||
print("The service was installed OK, but the performance monitor")
|
||||
print("data could not be loaded.", details)
|
||||
|
||||
def _GetCommandLine(exeName, exeArgs):
|
||||
if exeArgs is not None:
|
||||
return exeName + " " + exeArgs
|
||||
else:
|
||||
return exeName
|
||||
|
||||
def InstallService(pythonClassString, serviceName, displayName, startType = None, errorControl = None, bRunInteractive = 0, serviceDeps = None, userName = None, password = None, exeName = None, perfMonIni = None, perfMonDll = None, exeArgs = None,
|
||||
description = None, delayedstart = None):
|
||||
# Handle the default arguments.
|
||||
if startType is None:
|
||||
startType = win32service.SERVICE_DEMAND_START
|
||||
serviceType = win32service.SERVICE_WIN32_OWN_PROCESS
|
||||
if bRunInteractive:
|
||||
serviceType = serviceType | win32service.SERVICE_INTERACTIVE_PROCESS
|
||||
if errorControl is None:
|
||||
errorControl = win32service.SERVICE_ERROR_NORMAL
|
||||
|
||||
exeName = '"%s"' % LocatePythonServiceExe(exeName) # None here means use default PythonService.exe
|
||||
commandLine = _GetCommandLine(exeName, exeArgs)
|
||||
hscm = win32service.OpenSCManager(None,None,win32service.SC_MANAGER_ALL_ACCESS)
|
||||
try:
|
||||
hs = win32service.CreateService(hscm,
|
||||
serviceName,
|
||||
displayName,
|
||||
win32service.SERVICE_ALL_ACCESS, # desired access
|
||||
serviceType, # service type
|
||||
startType,
|
||||
errorControl, # error control type
|
||||
commandLine,
|
||||
None,
|
||||
0,
|
||||
serviceDeps,
|
||||
userName,
|
||||
password)
|
||||
if description is not None:
|
||||
try:
|
||||
win32service.ChangeServiceConfig2(hs,win32service.SERVICE_CONFIG_DESCRIPTION,description)
|
||||
except NotImplementedError:
|
||||
pass ## ChangeServiceConfig2 and description do not exist on NT
|
||||
if delayedstart is not None:
|
||||
try:
|
||||
win32service.ChangeServiceConfig2(hs,win32service.SERVICE_CONFIG_DELAYED_AUTO_START_INFO, delayedstart)
|
||||
except (win32service.error, NotImplementedError):
|
||||
## delayed start only exists on Vista and later - warn only when trying to set delayed to True
|
||||
if delayedstart:
|
||||
warnings.warn('Delayed Start not available on this system')
|
||||
win32service.CloseServiceHandle(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
InstallPythonClassString(pythonClassString, serviceName)
|
||||
# If I have performance monitor info to install, do that.
|
||||
if perfMonIni is not None:
|
||||
InstallPerfmonForService(serviceName, perfMonIni, perfMonDll)
|
||||
|
||||
def ChangeServiceConfig(pythonClassString, serviceName, startType = None, errorControl = None, bRunInteractive = 0,
|
||||
serviceDeps = None, userName = None, password = None,
|
||||
exeName = None, displayName = None, perfMonIni = None, perfMonDll = None,
|
||||
exeArgs = None, description = None, delayedstart = None):
|
||||
# Before doing anything, remove any perfmon counters.
|
||||
try:
|
||||
import perfmon
|
||||
perfmon.UnloadPerfCounterTextStrings("python.exe "+serviceName)
|
||||
except (ImportError, win32api.error):
|
||||
pass
|
||||
|
||||
# The EXE location may have changed
|
||||
exeName = '"%s"' % LocatePythonServiceExe(exeName)
|
||||
|
||||
# Handle the default arguments.
|
||||
if startType is None: startType = win32service.SERVICE_NO_CHANGE
|
||||
if errorControl is None: errorControl = win32service.SERVICE_NO_CHANGE
|
||||
|
||||
hscm = win32service.OpenSCManager(None,None,win32service.SC_MANAGER_ALL_ACCESS)
|
||||
serviceType = win32service.SERVICE_WIN32_OWN_PROCESS
|
||||
if bRunInteractive:
|
||||
serviceType = serviceType | win32service.SERVICE_INTERACTIVE_PROCESS
|
||||
commandLine = _GetCommandLine(exeName, exeArgs)
|
||||
try:
|
||||
hs = SmartOpenService(hscm, serviceName, win32service.SERVICE_ALL_ACCESS)
|
||||
try:
|
||||
|
||||
win32service.ChangeServiceConfig(hs,
|
||||
serviceType, # service type
|
||||
startType,
|
||||
errorControl, # error control type
|
||||
commandLine,
|
||||
None,
|
||||
0,
|
||||
serviceDeps,
|
||||
userName,
|
||||
password,
|
||||
displayName)
|
||||
if description is not None:
|
||||
try:
|
||||
win32service.ChangeServiceConfig2(hs,win32service.SERVICE_CONFIG_DESCRIPTION,description)
|
||||
except NotImplementedError:
|
||||
pass ## ChangeServiceConfig2 and description do not exist on NT
|
||||
if delayedstart is not None:
|
||||
try:
|
||||
win32service.ChangeServiceConfig2(hs,win32service.SERVICE_CONFIG_DELAYED_AUTO_START_INFO, delayedstart)
|
||||
except (win32service.error, NotImplementedError):
|
||||
## Delayed start only exists on Vista and later. On Nt, will raise NotImplementedError since ChangeServiceConfig2
|
||||
## doensn't exist. On Win2k and XP, will fail with ERROR_INVALID_LEVEL
|
||||
## Warn only if trying to set delayed to True
|
||||
if delayedstart:
|
||||
warnings.warn('Delayed Start not available on this system')
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
InstallPythonClassString(pythonClassString, serviceName)
|
||||
# If I have performance monitor info to install, do that.
|
||||
if perfMonIni is not None:
|
||||
InstallPerfmonForService(serviceName, perfMonIni, perfMonDll)
|
||||
|
||||
def InstallPythonClassString(pythonClassString, serviceName):
|
||||
# Now setup our Python specific entries.
|
||||
if pythonClassString:
|
||||
key = win32api.RegCreateKey(win32con.HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\%s\\PythonClass" % serviceName)
|
||||
try:
|
||||
win32api.RegSetValue(key, None, win32con.REG_SZ, pythonClassString);
|
||||
finally:
|
||||
win32api.RegCloseKey(key)
|
||||
|
||||
# Utility functions for Services, to allow persistant properties.
|
||||
def SetServiceCustomOption(serviceName, option, value):
|
||||
try:
|
||||
serviceName = serviceName._svc_name_
|
||||
except AttributeError:
|
||||
pass
|
||||
key = win32api.RegCreateKey(win32con.HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\%s\\Parameters" % serviceName)
|
||||
try:
|
||||
if type(value)==type(0):
|
||||
win32api.RegSetValueEx(key, option, 0, win32con.REG_DWORD, value);
|
||||
else:
|
||||
win32api.RegSetValueEx(key, option, 0, win32con.REG_SZ, value);
|
||||
finally:
|
||||
win32api.RegCloseKey(key)
|
||||
|
||||
def GetServiceCustomOption(serviceName, option, defaultValue = None):
|
||||
# First param may also be a service class/instance.
|
||||
# This allows services to pass "self"
|
||||
try:
|
||||
serviceName = serviceName._svc_name_
|
||||
except AttributeError:
|
||||
pass
|
||||
key = win32api.RegCreateKey(win32con.HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\%s\\Parameters" % serviceName)
|
||||
try:
|
||||
try:
|
||||
return win32api.RegQueryValueEx(key, option)[0]
|
||||
except win32api.error: # No value.
|
||||
return defaultValue
|
||||
finally:
|
||||
win32api.RegCloseKey(key)
|
||||
|
||||
|
||||
def RemoveService(serviceName):
|
||||
try:
|
||||
import perfmon
|
||||
perfmon.UnloadPerfCounterTextStrings("python.exe "+serviceName)
|
||||
except (ImportError, win32api.error):
|
||||
pass
|
||||
|
||||
hscm = win32service.OpenSCManager(None,None,win32service.SC_MANAGER_ALL_ACCESS)
|
||||
try:
|
||||
hs = SmartOpenService(hscm, serviceName, win32service.SERVICE_ALL_ACCESS)
|
||||
win32service.DeleteService(hs)
|
||||
win32service.CloseServiceHandle(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
|
||||
import win32evtlogutil
|
||||
try:
|
||||
win32evtlogutil.RemoveSourceFromRegistry(serviceName)
|
||||
except win32api.error:
|
||||
pass
|
||||
|
||||
def ControlService(serviceName, code, machine = None):
|
||||
hscm = win32service.OpenSCManager(machine,None,win32service.SC_MANAGER_ALL_ACCESS)
|
||||
try:
|
||||
|
||||
hs = SmartOpenService(hscm, serviceName, win32service.SERVICE_ALL_ACCESS)
|
||||
try:
|
||||
status = win32service.ControlService(hs, code)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
return status
|
||||
|
||||
def __FindSvcDeps(findName):
|
||||
if type(findName) is pywintypes.UnicodeType: findName = str(findName)
|
||||
dict = {}
|
||||
k = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Services")
|
||||
num = 0
|
||||
while 1:
|
||||
try:
|
||||
svc = win32api.RegEnumKey(k, num)
|
||||
except win32api.error:
|
||||
break
|
||||
num = num + 1
|
||||
sk = win32api.RegOpenKey(k, svc)
|
||||
try:
|
||||
deps, typ = win32api.RegQueryValueEx(sk, "DependOnService")
|
||||
except win32api.error:
|
||||
deps = ()
|
||||
for dep in deps:
|
||||
dep = dep.lower()
|
||||
dep_on = dict.get(dep, [])
|
||||
dep_on.append(svc)
|
||||
dict[dep]=dep_on
|
||||
|
||||
return __ResolveDeps(findName, dict)
|
||||
|
||||
|
||||
def __ResolveDeps(findName, dict):
|
||||
items = dict.get(findName.lower(), [])
|
||||
retList = []
|
||||
for svc in items:
|
||||
retList.insert(0, svc)
|
||||
retList = __ResolveDeps(svc, dict) + retList
|
||||
return retList
|
||||
|
||||
def WaitForServiceStatus(serviceName, status, waitSecs, machine=None):
|
||||
"""Waits for the service to return the specified status. You
|
||||
should have already requested the service to enter that state"""
|
||||
for i in range(waitSecs*4):
|
||||
now_status = QueryServiceStatus(serviceName, machine)[1]
|
||||
if now_status == status:
|
||||
break
|
||||
win32api.Sleep(250)
|
||||
else:
|
||||
raise pywintypes.error(winerror.ERROR_SERVICE_REQUEST_TIMEOUT, "QueryServiceStatus", win32api.FormatMessage(winerror.ERROR_SERVICE_REQUEST_TIMEOUT)[:-2])
|
||||
|
||||
def __StopServiceWithTimeout(hs, waitSecs = 30):
|
||||
try:
|
||||
status = win32service.ControlService(hs, win32service.SERVICE_CONTROL_STOP)
|
||||
except pywintypes.error as exc:
|
||||
if exc.winerror!=winerror.ERROR_SERVICE_NOT_ACTIVE:
|
||||
raise
|
||||
for i in range(waitSecs):
|
||||
status = win32service.QueryServiceStatus(hs)
|
||||
if status[1] == win32service.SERVICE_STOPPED:
|
||||
break
|
||||
win32api.Sleep(1000)
|
||||
else:
|
||||
raise pywintypes.error(winerror.ERROR_SERVICE_REQUEST_TIMEOUT, "ControlService", win32api.FormatMessage(winerror.ERROR_SERVICE_REQUEST_TIMEOUT)[:-2])
|
||||
|
||||
|
||||
def StopServiceWithDeps(serviceName, machine = None, waitSecs = 30):
|
||||
# Stop a service recursively looking for dependant services
|
||||
hscm = win32service.OpenSCManager(machine,None,win32service.SC_MANAGER_ALL_ACCESS)
|
||||
try:
|
||||
deps = __FindSvcDeps(serviceName)
|
||||
for dep in deps:
|
||||
hs = win32service.OpenService(hscm, dep, win32service.SERVICE_ALL_ACCESS)
|
||||
try:
|
||||
__StopServiceWithTimeout(hs, waitSecs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hs)
|
||||
# Now my service!
|
||||
hs = win32service.OpenService(hscm, serviceName, win32service.SERVICE_ALL_ACCESS)
|
||||
try:
|
||||
__StopServiceWithTimeout(hs, waitSecs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hs)
|
||||
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
|
||||
|
||||
def StopService(serviceName, machine = None):
|
||||
return ControlService(serviceName, win32service.SERVICE_CONTROL_STOP, machine)
|
||||
|
||||
def StartService(serviceName, args = None, machine = None):
|
||||
hscm = win32service.OpenSCManager(machine,None,win32service.SC_MANAGER_ALL_ACCESS)
|
||||
try:
|
||||
|
||||
hs = SmartOpenService(hscm, serviceName, win32service.SERVICE_ALL_ACCESS)
|
||||
try:
|
||||
win32service.StartService(hs, args)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
|
||||
def RestartService(serviceName, args = None, waitSeconds = 30, machine = None):
|
||||
"Stop the service, and then start it again (with some tolerance for allowing it to stop.)"
|
||||
try:
|
||||
StopService(serviceName, machine)
|
||||
except pywintypes.error as exc:
|
||||
# Allow only "service not running" error
|
||||
if exc.winerror!=winerror.ERROR_SERVICE_NOT_ACTIVE:
|
||||
raise
|
||||
# Give it a few goes, as the service may take time to stop
|
||||
for i in range(waitSeconds):
|
||||
try:
|
||||
StartService(serviceName, args, machine)
|
||||
break
|
||||
except pywintypes.error as exc:
|
||||
if exc.winerror!=winerror.ERROR_SERVICE_ALREADY_RUNNING:
|
||||
raise
|
||||
win32api.Sleep(1000)
|
||||
else:
|
||||
print("Gave up waiting for the old service to stop!")
|
||||
|
||||
def _DebugCtrlHandler(evt):
|
||||
if evt in (win32con.CTRL_C_EVENT, win32con.CTRL_BREAK_EVENT):
|
||||
assert g_debugService
|
||||
print("Stopping debug service.")
|
||||
g_debugService.SvcStop()
|
||||
return True
|
||||
return False
|
||||
|
||||
def DebugService(cls, argv = []):
|
||||
# Run a service in "debug" mode. Re-implements what pythonservice.exe
|
||||
# does when it sees a "-debug" param.
|
||||
# Currently only used by "frozen" (ie, py2exe) programs (but later may
|
||||
# end up being used for all services should we ever remove
|
||||
# pythonservice.exe)
|
||||
import servicemanager
|
||||
global g_debugService
|
||||
|
||||
print("Debugging service %s - press Ctrl+C to stop." % (cls._svc_name_,))
|
||||
servicemanager.Debugging(True)
|
||||
servicemanager.PrepareToHostSingle(cls)
|
||||
g_debugService = cls(argv)
|
||||
# Setup a ctrl+c handler to simulate a "stop"
|
||||
win32api.SetConsoleCtrlHandler(_DebugCtrlHandler, True)
|
||||
try:
|
||||
g_debugService.SvcRun()
|
||||
finally:
|
||||
win32api.SetConsoleCtrlHandler(_DebugCtrlHandler, False)
|
||||
servicemanager.Debugging(False)
|
||||
g_debugService = None
|
||||
|
||||
def GetServiceClassString(cls, argv = None):
|
||||
if argv is None:
|
||||
argv = sys.argv
|
||||
import pickle
|
||||
modName = pickle.whichmodule(cls, cls.__name__)
|
||||
if modName == '__main__':
|
||||
try:
|
||||
fname = win32api.GetFullPathName(argv[0])
|
||||
path = os.path.split(fname)[0]
|
||||
# Eaaaahhhh - sometimes this will be a short filename, which causes
|
||||
# problems with 1.5.1 and the silly filename case rule.
|
||||
filelist = win32api.FindFiles(fname)
|
||||
# win32api.FindFiles will not detect files in a zip or exe. If list is empty,
|
||||
# skip the test and hope the file really exists.
|
||||
if len(filelist) != 0:
|
||||
# Get the long name
|
||||
fname = os.path.join(path, filelist[0][8])
|
||||
except win32api.error:
|
||||
raise error("Could not resolve the path name '%s' to a full path" % (argv[0]))
|
||||
modName = os.path.splitext(fname)[0]
|
||||
return modName + "." + cls.__name__
|
||||
|
||||
def QueryServiceStatus(serviceName, machine=None):
|
||||
hscm = win32service.OpenSCManager(machine,None,win32service.SC_MANAGER_CONNECT)
|
||||
try:
|
||||
|
||||
hs = SmartOpenService(hscm, serviceName, win32service.SERVICE_QUERY_STATUS)
|
||||
try:
|
||||
status = win32service.QueryServiceStatus(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hs)
|
||||
finally:
|
||||
win32service.CloseServiceHandle(hscm)
|
||||
return status
|
||||
|
||||
def usage():
|
||||
try:
|
||||
fname = os.path.split(sys.argv[0])[1]
|
||||
except:
|
||||
fname = sys.argv[0]
|
||||
print("Usage: '%s [options] install|update|remove|start [...]|stop|restart [...]|debug [...]'" % fname)
|
||||
print("Options for 'install' and 'update' commands only:")
|
||||
print(" --username domain\\username : The Username the service is to run under")
|
||||
print(" --password password : The password for the username")
|
||||
print(" --startup [manual|auto|disabled|delayed] : How the service starts, default = manual")
|
||||
print(" --interactive : Allow the service to interact with the desktop.")
|
||||
print(" --perfmonini file: .ini file to use for registering performance monitor data")
|
||||
print(" --perfmondll file: .dll file to use when querying the service for")
|
||||
print(" performance data, default = perfmondata.dll")
|
||||
print("Options for 'start' and 'stop' commands only:")
|
||||
print(" --wait seconds: Wait for the service to actually start or stop.")
|
||||
print(" If you specify --wait with the 'stop' option, the service")
|
||||
print(" and all dependent services will be stopped, each waiting")
|
||||
print(" the specified period.")
|
||||
sys.exit(1)
|
||||
|
||||
def HandleCommandLine(cls, serviceClassString = None, argv = None, customInstallOptions = "", customOptionHandler = None):
|
||||
"""Utility function allowing services to process the command line.
|
||||
|
||||
Allows standard commands such as 'start', 'stop', 'debug', 'install' etc.
|
||||
|
||||
Install supports 'standard' command line options prefixed with '--', such as
|
||||
--username, --password, etc. In addition,
|
||||
the function allows custom command line options to be handled by the calling function.
|
||||
"""
|
||||
err = 0
|
||||
|
||||
if argv is None: argv = sys.argv
|
||||
|
||||
if len(argv)<=1:
|
||||
usage()
|
||||
|
||||
serviceName = cls._svc_name_
|
||||
serviceDisplayName = cls._svc_display_name_
|
||||
if serviceClassString is None:
|
||||
serviceClassString = GetServiceClassString(cls)
|
||||
|
||||
# Pull apart the command line
|
||||
import getopt
|
||||
try:
|
||||
opts, args = getopt.getopt(argv[1:], customInstallOptions,["password=","username=","startup=","perfmonini=", "perfmondll=", "interactive", "wait="])
|
||||
except getopt.error as details:
|
||||
print(details)
|
||||
usage()
|
||||
userName = None
|
||||
password = None
|
||||
perfMonIni = perfMonDll = None
|
||||
startup = None
|
||||
delayedstart = None
|
||||
interactive = None
|
||||
waitSecs = 0
|
||||
for opt, val in opts:
|
||||
if opt=='--username':
|
||||
userName = val
|
||||
elif opt=='--password':
|
||||
password = val
|
||||
elif opt=='--perfmonini':
|
||||
perfMonIni = val
|
||||
elif opt=='--perfmondll':
|
||||
perfMonDll = val
|
||||
elif opt=='--interactive':
|
||||
interactive = 1
|
||||
elif opt=='--startup':
|
||||
map = {"manual": win32service.SERVICE_DEMAND_START,
|
||||
"auto" : win32service.SERVICE_AUTO_START,
|
||||
"delayed": win32service.SERVICE_AUTO_START, ## ChangeServiceConfig2 called later
|
||||
"disabled": win32service.SERVICE_DISABLED}
|
||||
try:
|
||||
startup = map[val.lower()]
|
||||
except KeyError:
|
||||
print("'%s' is not a valid startup option" % val)
|
||||
if val.lower() == "delayed":
|
||||
delayedstart = True
|
||||
elif val.lower() == "auto":
|
||||
delayedstart = False
|
||||
## else no change
|
||||
elif opt=='--wait':
|
||||
try:
|
||||
waitSecs = int(val)
|
||||
except ValueError:
|
||||
print("--wait must specify an integer number of seconds.")
|
||||
usage()
|
||||
|
||||
arg=args[0]
|
||||
knownArg = 0
|
||||
# First we process all arguments which pass additional args on
|
||||
if arg=="start":
|
||||
knownArg = 1
|
||||
print("Starting service %s" % (serviceName))
|
||||
try:
|
||||
StartService(serviceName, args[1:])
|
||||
if waitSecs:
|
||||
WaitForServiceStatus(serviceName, win32service.SERVICE_RUNNING, waitSecs)
|
||||
except win32service.error as exc:
|
||||
print("Error starting service: %s" % exc.strerror)
|
||||
err = exc.winerror
|
||||
|
||||
elif arg=="restart":
|
||||
knownArg = 1
|
||||
print("Restarting service %s" % (serviceName))
|
||||
RestartService(serviceName, args[1:])
|
||||
if waitSecs:
|
||||
WaitForServiceStatus(serviceName, win32service.SERVICE_RUNNING, waitSecs)
|
||||
|
||||
elif arg=="debug":
|
||||
knownArg = 1
|
||||
if not hasattr(sys, "frozen"):
|
||||
# non-frozen services use pythonservice.exe which handles a
|
||||
# -debug option
|
||||
svcArgs = " ".join(args[1:])
|
||||
try:
|
||||
exeName = LocateSpecificServiceExe(serviceName)
|
||||
except win32api.error as exc:
|
||||
if exc.winerror == winerror.ERROR_FILE_NOT_FOUND:
|
||||
print("The service does not appear to be installed.")
|
||||
print("Please install the service before debugging it.")
|
||||
sys.exit(1)
|
||||
raise
|
||||
try:
|
||||
os.system("%s -debug %s %s" % (exeName, serviceName, svcArgs))
|
||||
# ^C is used to kill the debug service. Sometimes Python also gets
|
||||
# interrupted - ignore it...
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
else:
|
||||
# py2exe services don't use pythonservice - so we simulate
|
||||
# debugging here.
|
||||
DebugService(cls, args)
|
||||
|
||||
if not knownArg and len(args)!=1:
|
||||
usage() # the rest of the cmds don't take addn args
|
||||
|
||||
if arg=="install":
|
||||
knownArg = 1
|
||||
try:
|
||||
serviceDeps = cls._svc_deps_
|
||||
except AttributeError:
|
||||
serviceDeps = None
|
||||
try:
|
||||
exeName = cls._exe_name_
|
||||
except AttributeError:
|
||||
exeName = None # Default to PythonService.exe
|
||||
try:
|
||||
exeArgs = cls._exe_args_
|
||||
except AttributeError:
|
||||
exeArgs = None
|
||||
try:
|
||||
description = cls._svc_description_
|
||||
except AttributeError:
|
||||
description = None
|
||||
print("Installing service %s" % (serviceName,))
|
||||
# Note that we install the service before calling the custom option
|
||||
# handler, so if the custom handler fails, we have an installed service (from NT's POV)
|
||||
# but is unlikely to work, as the Python code controlling it failed. Therefore
|
||||
# we remove the service if the first bit works, but the second doesnt!
|
||||
try:
|
||||
InstallService(serviceClassString, serviceName, serviceDisplayName, serviceDeps = serviceDeps, startType=startup, bRunInteractive=interactive, userName=userName,password=password, exeName=exeName, perfMonIni=perfMonIni,perfMonDll=perfMonDll,exeArgs=exeArgs,
|
||||
description=description, delayedstart=delayedstart)
|
||||
if customOptionHandler:
|
||||
customOptionHandler(*(opts,))
|
||||
print("Service installed")
|
||||
except win32service.error as exc:
|
||||
if exc.winerror==winerror.ERROR_SERVICE_EXISTS:
|
||||
arg = "update" # Fall through to the "update" param!
|
||||
else:
|
||||
print("Error installing service: %s (%d)" % (exc.strerror, exc.winerror))
|
||||
err = exc.winerror
|
||||
except ValueError as msg: # Can be raised by custom option handler.
|
||||
print("Error installing service: %s" % str(msg))
|
||||
err = -1
|
||||
# xxx - maybe I should remove after _any_ failed install - however,
|
||||
# xxx - it may be useful to help debug to leave the service as it failed.
|
||||
# xxx - We really _must_ remove as per the comments above...
|
||||
# As we failed here, remove the service, so the next installation
|
||||
# attempt works.
|
||||
try:
|
||||
RemoveService(serviceName)
|
||||
except win32api.error:
|
||||
print("Warning - could not remove the partially installed service.")
|
||||
|
||||
if arg == "update":
|
||||
knownArg = 1
|
||||
try:
|
||||
serviceDeps = cls._svc_deps_
|
||||
except AttributeError:
|
||||
serviceDeps = None
|
||||
try:
|
||||
exeName = cls._exe_name_
|
||||
except AttributeError:
|
||||
exeName = None # Default to PythonService.exe
|
||||
try:
|
||||
exeArgs = cls._exe_args_
|
||||
except AttributeError:
|
||||
exeArgs = None
|
||||
try:
|
||||
description=cls._svc_description_
|
||||
except AttributeError:
|
||||
description=None
|
||||
print("Changing service configuration")
|
||||
try:
|
||||
ChangeServiceConfig(serviceClassString, serviceName, serviceDeps = serviceDeps, startType=startup, bRunInteractive=interactive, userName=userName,password=password, exeName=exeName, displayName = serviceDisplayName, perfMonIni=perfMonIni,perfMonDll=perfMonDll,exeArgs=exeArgs,
|
||||
description=description, delayedstart=delayedstart)
|
||||
if customOptionHandler:
|
||||
customOptionHandler(*(opts,))
|
||||
print("Service updated")
|
||||
except win32service.error as exc:
|
||||
print("Error changing service configuration: %s (%d)" % (exc.strerror,exc.winerror))
|
||||
err = exc.winerror
|
||||
|
||||
elif arg=="remove":
|
||||
knownArg = 1
|
||||
print("Removing service %s" % (serviceName))
|
||||
try:
|
||||
RemoveService(serviceName)
|
||||
print("Service removed")
|
||||
except win32service.error as exc:
|
||||
print("Error removing service: %s (%d)" % (exc.strerror,exc.winerror))
|
||||
err = exc.winerror
|
||||
elif arg=="stop":
|
||||
knownArg = 1
|
||||
print("Stopping service %s" % (serviceName))
|
||||
try:
|
||||
if waitSecs:
|
||||
StopServiceWithDeps(serviceName, waitSecs = waitSecs)
|
||||
else:
|
||||
StopService(serviceName)
|
||||
except win32service.error as exc:
|
||||
print("Error stopping service: %s (%d)" % (exc.strerror,exc.winerror))
|
||||
err = exc.winerror
|
||||
if not knownArg:
|
||||
err = -1
|
||||
print("Unknown command - '%s'" % arg)
|
||||
usage()
|
||||
return err
|
||||
|
||||
#
|
||||
# Useful base class to build services from.
|
||||
#
|
||||
class ServiceFramework:
|
||||
# Required Attributes:
|
||||
# _svc_name_ = The service name
|
||||
# _svc_display_name_ = The service display name
|
||||
|
||||
# Optional Attributes:
|
||||
_svc_deps_ = None # sequence of service names on which this depends
|
||||
_exe_name_ = None # Default to PythonService.exe
|
||||
_exe_args_ = None # Default to no arguments
|
||||
_svc_description_ = None # Only exists on Windows 2000 or later, ignored on windows NT
|
||||
|
||||
def __init__(self, args):
|
||||
import servicemanager
|
||||
self.ssh = servicemanager.RegisterServiceCtrlHandler(args[0], self.ServiceCtrlHandlerEx, True)
|
||||
servicemanager.SetEventSourceName(self._svc_name_)
|
||||
self.checkPoint = 0
|
||||
|
||||
def GetAcceptedControls(self):
|
||||
# Setup the service controls we accept based on our attributes. Note
|
||||
# that if you need to handle controls via SvcOther[Ex](), you must
|
||||
# override this.
|
||||
accepted = 0
|
||||
if hasattr(self, "SvcStop"): accepted = accepted | win32service.SERVICE_ACCEPT_STOP
|
||||
if hasattr(self, "SvcPause") and hasattr(self, "SvcContinue"):
|
||||
accepted = accepted | win32service.SERVICE_ACCEPT_PAUSE_CONTINUE
|
||||
if hasattr(self, "SvcShutdown"): accepted = accepted | win32service.SERVICE_ACCEPT_SHUTDOWN
|
||||
return accepted
|
||||
|
||||
def ReportServiceStatus(self, serviceStatus, waitHint = 5000, win32ExitCode = 0, svcExitCode = 0):
|
||||
if self.ssh is None: # Debugging!
|
||||
return
|
||||
if serviceStatus == win32service.SERVICE_START_PENDING:
|
||||
accepted = 0
|
||||
else:
|
||||
accepted = self.GetAcceptedControls()
|
||||
|
||||
if serviceStatus in [win32service.SERVICE_RUNNING, win32service.SERVICE_STOPPED]:
|
||||
checkPoint = 0
|
||||
else:
|
||||
self.checkPoint = self.checkPoint + 1
|
||||
checkPoint = self.checkPoint
|
||||
|
||||
# Now report the status to the control manager
|
||||
status = (win32service.SERVICE_WIN32_OWN_PROCESS,
|
||||
serviceStatus,
|
||||
accepted, # dwControlsAccepted,
|
||||
win32ExitCode, # dwWin32ExitCode;
|
||||
svcExitCode, # dwServiceSpecificExitCode;
|
||||
checkPoint, # dwCheckPoint;
|
||||
waitHint)
|
||||
win32service.SetServiceStatus( self.ssh, status)
|
||||
|
||||
def SvcInterrogate(self):
|
||||
# Assume we are running, and everyone is happy.
|
||||
self.ReportServiceStatus(win32service.SERVICE_RUNNING)
|
||||
|
||||
def SvcOther(self, control):
|
||||
try:
|
||||
print("Unknown control status - %d" % control)
|
||||
except IOError:
|
||||
# services may not have a valid stdout!
|
||||
pass
|
||||
|
||||
def ServiceCtrlHandler(self, control):
|
||||
return self.ServiceCtrlHandlerEx(control, 0, None)
|
||||
|
||||
# The 'Ex' functions, which take additional params
|
||||
def SvcOtherEx(self, control, event_type, data):
|
||||
# The default here is to call self.SvcOther as that is the old behaviour.
|
||||
# If you want to take advantage of the extra data, override this method
|
||||
return self.SvcOther(control)
|
||||
|
||||
def ServiceCtrlHandlerEx(self, control, event_type, data):
|
||||
if control==win32service.SERVICE_CONTROL_STOP:
|
||||
return self.SvcStop()
|
||||
elif control==win32service.SERVICE_CONTROL_PAUSE:
|
||||
return self.SvcPause()
|
||||
elif control==win32service.SERVICE_CONTROL_CONTINUE:
|
||||
return self.SvcContinue()
|
||||
elif control==win32service.SERVICE_CONTROL_INTERROGATE:
|
||||
return self.SvcInterrogate()
|
||||
elif control==win32service.SERVICE_CONTROL_SHUTDOWN:
|
||||
return self.SvcShutdown()
|
||||
else:
|
||||
return self.SvcOtherEx(control, event_type, data)
|
||||
|
||||
def SvcRun(self):
|
||||
self.ReportServiceStatus(win32service.SERVICE_RUNNING)
|
||||
self.SvcDoRun()
|
||||
# Once SvcDoRun terminates, the service has stopped.
|
||||
# We tell the SCM the service is still stopping - the C framework
|
||||
# will automatically tell the SCM it has stopped when this returns.
|
||||
self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
|
Loading…
Add table
Add a link
Reference in a new issue