Python程序自动以管理员权限运行的实现方法

前言

由于Windows的安全机制，Python写的脚本缺少了管理员权限，运行就会受到一些限制。

文章介绍Python 脚本自动以管理员权限运行的方法，

也就是如果脚本不是以管理员运行，就自动提升到管理员权限。

核心API函数

将python提升到管理员权限运行需要调用ShellExecute这个API函数。

ShellExecute函数的用法如下（这里仅作了解 ）：

ShellExecute(hWnd,Operation,FileName,Parameters,Directory,ShowCmd)
hWnd: 整数 (指定父窗口句柄，如果没有可用None代替)
Operation: 字符串 (指定动作, 譬如: open、runas、print、edit、explore、find)
FileName: 字符串 (指定要打开的文件或程序)
Parameters: 字符串 (给要打开的程序指定参数)
Directory: 字符串 (缺省目录)
ShowCmd: 整数 (程序启动后的窗口显示方式，
正常为1、最小化为2、最大化为3)

代码实现(1)

使用Python自带的ctypes模块，调用API函数的代码：

import ctypes, sys

def is_admin():
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False

if is_admin():
    # 主程序写在这里
else:
    # 以管理员权限重新运行程序
    ctypes.windll.shell32.ShellExecuteW(None,"runas", sys.executable, __file__, None, 1)

如果是Python 2，把最后一行改成这样：

ctypes.windll.shell32.ShellExecuteW(None, u"runas", unicode(sys.executable), unicode(__file__), None, 1)

如果使用pyinstaller等库打包python程序，则应将ShellExecuteW的第四个参数替换为""。

另外，在ShellExecuteW的第4个参数中可加入其他的命令行参数，
在第6个参数中可指定窗口显示方式，如最小化(为2)、最大化(为3)等。

本方法优点：

• 不需要外部库，仅使用自带的ctypes模块。
• 适用于Python 2和Python 3。

本方法缺点：

在命令行窗口(cmd)中运行python时，会另外打开一个命令行窗口，造成输出的消息一闪而过。

(不过，据了解这似乎是Windows自身的安全机制造成的，而不是程序的bug，可能是用于防止漏洞)

代码实现(2)

相比于Python自带的ctypes库，win32api库更加广为人知、使用方便。

使用pip的安装方法：pip install pywin32使用win32api模块的代码：

import sys, os, win32api

def is_admin():
    # 由于win32api中没有IsUserAnAdmin函数,所以用了这种方法
    try:
        # 在c:\windows目录下新建一个文件test01.txt
        testfile=os.path.join(os.getenv("windir"),"test01.txt")
        open(testfile,"w").close()
    except OSError: # 不成功
        return False
    else: # 成功
        os.remove(testfile) # 删除文件
        return True

print(is_admin())
if is_admin():
    # 主程序写在这里
else:
    # 以管理员权限重新运行程序
    win32api.ShellExecute(None,"runas", sys.executable, __file__, None, 1)

如果是Python 2，将except OSError这一行改为except IOError。

本方法优点：

• 适用于Python 2和Python 3。

总结

到此这篇关于Python程序自动以管理员权限运行实现的文章就介绍到这了,更多相关Python自动以管理员权限运行内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

最新评论