Windows平台使用Python切换用户运行程序

在Windows平台,如果需要以某个指定用户来运行程序,此时可以通过使用 Windows CreateProcessAsUser API来实现。但是要实现这个功能首先需要有要切换用户的用户名和密码。

import psutil
import win32process
import win32security
import win32con
import win32api
import win32service
import win32file
import win32profile
import pywintypes

try:
    username = 'admin'
    password = 'admin'
    domain = ''
    token = win32security.LogonUser (
        username,
        domain,
        password,
        win32con.LOGON32_LOGON_SERVICE,
        win32con.LOGON32_PROVIDER_DEFAULT
    )
    win32security.ImpersonateLoggedOnUser(token)

    cmd = ''
    cwd = ''
    env = os.environ
    priority = win32con.NORMAL_PRIORITY_CLASS
    startup = win32process.STARTUPINFO()

    (hProcess, hThread, dwProcessId, dwThreadId) = \
        win32process.CreateProcessAsUser(token, None, cmd, None, None, True, priority, env, cwd, startup)

    # win32process.GetExitCodeProcess(hProcess) == win32con.STILL_ACTIVE
    process = psutil.Process(dwProcessId)
    return_code = process.wait()
    print 'return code: %s' % str(return_code)
except win32security.error as e:
    print e
  • 要使上面的程序生效,首先需要使用一个windows service来执行这个程序(目前我是这么测试的,有人说也可以在命令行以administrator来运行,但是我没有测试成功)。关于Python创建Windows Service的方法,可以参考我前面文章。
  • 要切换用户执行脚本,需要有被切换用户的用户名和密码。
  • 在用 CreateProcessAsUser API执行命令后,可以通过 GetExitCodeProcess api 来检查进程状态,如果返回结果是 win32con.STILL_ACTIVE,我们可以继续等待。
  • 显然,通过 GetExitCodeProcess api来检查进程是否退出不是很好用,所以这里我使用了psutil库来等待进程退出。

参考

  • http://timgolden.me.uk/pywin32-docs/contents.html

你可能感兴趣的:(Windows平台使用Python切换用户运行程序)