198 lines
7.9 KiB
Markdown
198 lines
7.9 KiB
Markdown
# 使用 Python 重启电脑
|
||
|
||
> 原文：<https://www.blog.pythonlibrary.org/2010/03/27/restarting-pcs-with-python/>
|
||
|
||
你有没有想过重新启动你的 Windows 电脑没有按下开始，关机或 CTRL+ALT+DEL？重启你讨厌的同事的电脑怎么样...就是那个不知道什么时候该闭嘴的人？Python 给出了答案，这个博客将告诉你如何去做！注意:我并不建议你随意重启邻居的电脑...
|
||
|
||
## 使用 PyWin32 重新启动
|
||
|
||
无论如何，当我第一次学习 Python 时，我偶然发现了一个关于如何做到这一点的 ActiveState 方法。当然，现在我似乎找不到那个配方了，但是我在这里找到了一个相似的:http://code.activestate.com/recipes/360649/.因此，我们将首先从这个方法开始，然后看一个稍微不同的方法。如果你想跟着做，那么你需要确保你有 [PyWin32 包](http://sourceforge.net/projects/pywin32/files/)。
|
||
|
||
这是我从我模糊的过去得到的食谱:
|
||
|
||
```py
|
||
|
||
# rebootServer.py
|
||
|
||
import win32security
|
||
import win32api
|
||
import sys
|
||
import time
|
||
from ntsecuritycon import *
|
||
|
||
def AdjustPrivilege(priv, enable=1):
|
||
# Get the process token
|
||
flags = TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY
|
||
htoken = win32security.OpenProcessToken(win32api.GetCurrentProcess(), flags)
|
||
# Get the ID for the system shutdown privilege.
|
||
idd = win32security.LookupPrivilegeValue(None, priv)
|
||
# Now obtain the privilege for this process.
|
||
# Create a list of the privileges to be added.
|
||
if enable:
|
||
newPrivileges = [(idd, SE_PRIVILEGE_ENABLED)]
|
||
else:
|
||
newPrivileges = [(idd, 0)]
|
||
# and make the adjustment
|
||
win32security.AdjustTokenPrivileges(htoken, 0, newPrivileges)
|
||
|
||
def RebootServer(message='Rebooting', timeout=30, bForce=0, bReboot=1):
|
||
AdjustPrivilege(SE_SHUTDOWN_NAME)
|
||
try:
|
||
win32api.InitiateSystemShutdown(None, message, timeout, bForce, bReboot)
|
||
finally:
|
||
# Now we remove the privilege we just added.
|
||
AdjustPrivilege(SE_SHUTDOWN_NAME, 0)
|
||
|
||
def AbortReboot():
|
||
AdjustPrivilege(SE_SHUTDOWN_NAME)
|
||
try:
|
||
win32api.AbortSystemShotdown(None)
|
||
finally:
|
||
AdjustPrivilege(SE_SHUTDOWN_NAME, 0)
|
||
|
||
if __name__ == '__main__':
|
||
RebootServer()
|
||
time.sleep(10)
|
||
print 'Aborting shutdown'
|
||
AbortReboot()
|
||
|
||
```
|
||
|
||
在这个代码片段中，我实际上去掉了重启远程机器的功能，因为我不想这么做。如果要重新启动网络上的计算机，请更改以下行:
|
||
|
||
```py
|
||
|
||
win32api.InitiateSystemShutdown(None, message, timeout, bForce, bReboot)
|
||
|
||
```
|
||
|
||
到
|
||
|
||
```py
|
||
|
||
win32api.InitiateSystemShutdown("someMachineName", message, timeout, bForce, bReboot)
|
||
|
||
```
|
||
|
||
或者只需向 *RebootServer* 函数添加一个主机参数，并根据需要修改 InitiateSystemShutdown。如果你只是传递 *None* 进去，那么你只是重新启动你自己的机器。 *AdjustPrivilege* 方法用于改变进程的特权，这样它就可以关闭电脑。我认为这只适用于你作为一个拥有有限权限的用户运行脚本的情况，但是我几乎从来没有这样运行过，所以我不能确定。我记得菜谱上说您需要删除您添加的特权，这就是 *finally* 语句的作用。然而，我认为一旦重启完成，这些特权无论如何都会被取消。
|
||
|
||
## 用普通的 Python 重启
|
||
|
||
启动关机的另一种方法是学习 Windows 命令行的魔力。有一个关机命令，你可以把你的运行对话框(进入开始->运行)，这将重新启动你的电脑。这里有一个我说的:shutdown -r -t 1 *注意这只会重启本地机器！*
|
||
|
||
现在我们只需要弄清楚如何让 Python 为我们调用它。可能最简单的方法是导入 *os* 模块并调用它的*系统*方法。让我们来看看:
|
||
|
||
```py
|
||
|
||
import os
|
||
os.system("shutdown -r -t 1")
|
||
|
||
```
|
||
|
||
这比 PyWin32 方法短得多，但我不知道它是否同样健壮。我在一个快速而肮脏的 wxPython 应用程序中使用了这个方法，这个应用程序是我为工作中的 Sun Ray 虚拟机创建的。Sun Ray 系统的最大问题之一是用户使用远程桌面连接到他们的虚拟机。远程桌面的一部分是隐藏关机按钮，所以用户没有简单的方法来重新启动他们的电脑，除非他们安装了某种修复程序。有时候重启机器是一件好事，所以我将上面的脚本放入下面的应用程序中:
|
||
|
||
[](https://www.blog.pythonlibrary.org/wp-content/uploads/2010/03/restarter.png)
|
||
|
||
```py
|
||
|
||
import os
|
||
import wx
|
||
from wx.lib.buttons import GenBitmapButton
|
||
|
||
########################################################################
|
||
class RestarterPanel(wx.Panel):
|
||
""""""
|
||
|
||
#----------------------------------------------------------------------
|
||
def __init__(self, parent):
|
||
"""Constructor"""
|
||
wx.Panel.__init__(self, parent=parent)
|
||
|
||
img = wx.Bitmap("cute-logoff.png")
|
||
logoutBtn = GenBitmapButton(self, wx.ID_ANY, img, style=wx.BORDER_NONE)
|
||
logoutBtn.Bind(wx.EVT_BUTTON, self.onLogout)
|
||
|
||
img = wx.Bitmap("quick_restart.png")
|
||
restartBtn = GenBitmapButton(self, wx.ID_ANY, img, style=wx.BORDER_NONE)
|
||
restartBtn.Bind(wx.EVT_BUTTON, self.onRestart)
|
||
|
||
cancelBtn = wx.Button(self, label="Cancel")
|
||
cancelBtn.Bind(wx.EVT_BUTTON, self.onCancel)
|
||
|
||
vSizer = wx.BoxSizer(wx.VERTICAL)
|
||
sizer = wx.BoxSizer(wx.HORIZONTAL)
|
||
sizer.Add(self.buttonBuilder("Logout", logoutBtn), 0, wx.CENTER)
|
||
sizer.Add(self.buttonBuilder("Restart", restartBtn), 0, wx.CENTER)
|
||
vSizer.Add(sizer, 0, wx.CENTER)
|
||
vSizer.Add(cancelBtn, 0, wx.ALIGN_RIGHT|wx.ALL, 5)
|
||
self.SetSizer(vSizer)
|
||
|
||
#----------------------------------------------------------------------
|
||
def buttonBuilder(self, label, button):
|
||
"""
|
||
Creates a button with a label underneath it and puts them into
|
||
a vertical BoxSizer, which is then returned
|
||
"""
|
||
font = wx.Font(12, wx.SWISS, wx.NORMAL, wx.BOLD)
|
||
sizer = wx.BoxSizer(wx.VERTICAL)
|
||
sizer.Add(button, 0, wx.ALL, 3)
|
||
|
||
lbl = wx.StaticText(self, label=label)
|
||
lbl.SetFont(font)
|
||
sizer.Add(lbl, 0, wx.CENTER|wx.BOTTOM, 4)
|
||
|
||
return sizer
|
||
|
||
#----------------------------------------------------------------------
|
||
def onCancel(self, event):
|
||
"""
|
||
Close the dialog
|
||
"""
|
||
self.GetParent().Close()
|
||
|
||
#----------------------------------------------------------------------
|
||
def onLogout(self, event):
|
||
"""
|
||
Logs the current user out
|
||
"""
|
||
os.system("shutdown -t 0 -l")
|
||
|
||
#----------------------------------------------------------------------
|
||
def onRestart(self, event):
|
||
"""
|
||
Restarts the PC
|
||
"""
|
||
os.system("shutdown -r -t 1")
|
||
|
||
########################################################################
|
||
class RestarterFrame(wx.Frame):
|
||
""""""
|
||
|
||
#----------------------------------------------------------------------
|
||
def __init__(self):
|
||
"""Constructor"""
|
||
wx.Frame.__init__(self, None, size=(273, 169))
|
||
panel = RestarterPanel(self)
|
||
self.Center()
|
||
|
||
########################################################################
|
||
class Main(wx.App):
|
||
""""""
|
||
|
||
#----------------------------------------------------------------------
|
||
def __init__(self, redirect=False, filename=None):
|
||
"""Constructor"""
|
||
wx.App.__init__(self, redirect, filename)
|
||
dlg = RestarterFrame()
|
||
dlg.Show()
|
||
|
||
#----------------------------------------------------------------------
|
||
if __name__ == "__main__":
|
||
app = Main()
|
||
app.MainLoop()
|
||
|
||
```
|
||
|
||
我看了这个项目的很多图标，我想我最后只是搜索了一下[关机](http://commons.wikimedia.org/wiki/File:Quick_restart.png)和[注销](http://www.iconfinder.net/icondetails/32136/128/?q=cute)图标，然后下载了我最喜欢的图标。这两个都是免费的或者有知识共享许可。
|
||
|
||
现在您知道了重启本地和远程机器的技巧。明智地使用这些知识！ |