本文给大家分享的小技巧是如何利用 Python 脚本获取 Windows 和 Linux 的系统版本信息,非常的简单实用,有需要的小伙伴可以查看下
Python 是一种面向对象、解释型计算机程序设计语言,由 Guido van Rossum 于 1989 年底发明,第一个公开发行版发行于 1991 年。Python 语法简洁而清晰,具有丰富和强大的类库。它常被昵称为胶水语言,它能够把用其他语言制作的各种模块(尤其是 C/C++)很轻松地联结在一起。
查看系统版本信息是一件家常便饭的事情,有时候需要将版本信息录入到资产管理系统中,如果每次手动的去查询这些信息再录入系统那么是一件令人呢头疼的事情,如果采用脚本去完成这件事情,那么情况就有所不同了。
在 Python 的世界里,获取 Windows 版本信息和 Linux 的版本信息都可以采用 platform 模块,但 platform 模块也不是万能的,有些特殊的信息(比如 Windows 的内部版本号)这个模块拿不到,那么只能另辟蹊径了。
在 Linux 系统中,可以简单的认为一切都是文件,那么就算没有现成的命令可用时,可以用 open() 文件的方法通过对文件的读写控制它。而在 Windows 的大部分信息在注册表中都能查到,因此可以从注册表上下手。Windows 注册表是一个好东西,拿数据就像在 Linux 下一切都是文件一样方便,如果想用 Python 访问注册表,除了权限外就是需要模块了,在 Python 中_winreg 是一个内置模块,通过这一模块可以对注册表进行读写。
本脚本收集了一些获取版本信息的常见方法,除了 platform 模块外,还有其他的模块可供使用,因为 platform 模块不是内置模块,因此需要额外安装。Windows 下运行脚本需要考虑权限问题和中文字符的问题,解决 Python 打印中文字符的问题是通过脚本中的 get_system_encoding() 函数实现的,这个函数取自 Django,经过测试这个函数还是非常好用的。
注:在 PyCharm 中,经常遇到 Run 窗口打印出的中文显示乱码,代码中没有经过正确转码是一方面,而 IDE 的编码设置也是一方面。如果是在 Windows 下开发,那么建议代码用 UTF-8 编写,IDE 的编码则设置为 "GBK",设置方法 "File"-->"Settings"-->"Editor"-->"File Encoding"-->"IDE Encoding" 选择 "<System Default (now GBK)>", "Project Encoding" 选择 UTF-8 保证代码的编码一致性。
脚本如下:
- #!/usr/bin/python
- # encoding: utf-8
- # -*- coding: utf8 -*-
- """
- Created by PyCharm.
- File: LinuxBashShellScriptForOps:getSystemVersion.py
- User: Guodong
- Create Date: 2016/12/16
- Create Time: 14:51
- """
- import sys
- import os
- import platform
- import subprocess
- import codecs
- import locale
- def get_system_encoding():
- """
- The encoding of the default system locale but falls back to the given
- fallback encoding if the encoding is unsupported by python or could
- not be determined. See tickets #10335 and #5846
- """
- try:
- encoding = locale.getdefaultlocale()[1] or 'ascii'
- codecs.lookup(encoding)
- except Exception:
- encoding = 'ascii'
- return encoding
- DEFAULT_LOCALE_ENCODING = get_system_encoding()
- mswindows = (sys.platform == "win32") # learning from 'subprocess' module
- linux = (sys.platform == "linux2")
- hidden_hostname = True
- if mswindows:
- uname = list(platform.uname())
- if hidden_hostname:
- uname[1] = "hidden_hostname"
- print uname
- import _winreg
- try:
- reg_key = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion")
- if reg_key:
- ProductName = _winreg.QueryValueEx(reg_key, "ProductName")[0] or None
- EditionId = _winreg.QueryValueEx(reg_key, "EditionId")[0] or None
- ReleaseId = _winreg.QueryValueEx(reg_key, "ReleaseId")[0] or None
- CurrentBuild = _winreg.QueryValueEx(reg_key, "CurrentBuild")[0] or None
- BuildLabEx = _winreg.QueryValueEx(reg_key, "BuildLabEx")[0][:9] or None
- print (ProductName, EditionId, ReleaseId, CurrentBuild, BuildLabEx)
- except Exception as e:
- print e.message.decode(DEFAULT_LOCALE_ENCODING)
- if linux:
- uname = list(platform.uname())
- if hidden_hostname:
- uname[1] = "hidden_hostname"
- print uname
- proc_obj = subprocess.Popen(r'uname -a', shell=True, stdout=subprocess.PIPE,
- stderr=subprocess.STDOUT)
- result = proc_obj.stdout.read().strip().decode(DEFAULT_LOCALE_ENCODING)
- if result:
- print result
- if os.path.isfile("/proc/version"):
- with open("/proc/version", 'r') as f:
- content = f.read().strip()
- if content != "":
- print content
- if os.path.isfile("/etc/issue"):
- with open("/etc/issue", 'r') as f:
- content = f.read().strip()
- if content != "":
- print content
截图如下:
(1)注册表信息获取位置:
(2)Windows 环境下的输出:
(3)Linux 环境下的输出:
来源: http://www.phperz.com/article/17/0318/315060.html