108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
# 获取Windows的已打的补丁号
|
||
import winreg
|
||
from winreg import *
|
||
import re
|
||
import os
|
||
import datetime
|
||
import hashlib
|
||
|
||
def subRegKey(key, pattern, patchlist):
|
||
# 个数
|
||
count = QueryInfoKey(key)
|
||
for index in range(len(QueryInfoKey(key))):
|
||
# 获取标题
|
||
name = EnumKey(key, index)
|
||
|
||
result = patch.match(name)
|
||
if result:
|
||
patchlist.append(result.group(1))
|
||
sub = OpenKey(key, name)
|
||
subRegKey(sub, pattern, patchlist)
|
||
CloseKey(sub)
|
||
|
||
|
||
def check_run_state():
|
||
#验证运行有效性逻辑：
|
||
#一、判断是否正常安装
|
||
#判断条件：软件安装时在注册表“HKEY_CURRENT_USER\\SOFTWARE\\Gutin\\Record“记录下安装目录
|
||
#二、非正常安装有效时间为三个月,且只能发生一次
|
||
|
||
#验证当前运行目录是否存在注册表中
|
||
qualified = False
|
||
run_dir = os.path.abspath('.')
|
||
# print('run_dir:%s' % run_dir)
|
||
reg_path = 'SOFTWARE\\Gutin\\Record'
|
||
feature_name ='InstallDir'
|
||
key = OpenKey(HKEY_CURRENT_USER, reg_path, access = KEY_READ)
|
||
items = QueryInfoKey(key)
|
||
for i in range(items[1]):
|
||
item = EnumValue(key, i)
|
||
name = item[0]
|
||
value = item[1]
|
||
type = item[2]
|
||
if name and name.find(feature_name)>=0:
|
||
if os.path.samefile(value, run_dir):
|
||
qualified = True
|
||
break;
|
||
|
||
if not qualified:
|
||
#查找非正常安装目录记录
|
||
#记录以运行目录的hash值作为键名，值为首次运行的时间
|
||
time_format= '%Y-%m-%d %H:%M:%S'
|
||
run_time = None
|
||
# run_dir_hash = hashlib.md5(run_dir.encode('utf-8')).hexdigest()
|
||
unqualified_key_name = 'Unqualified'
|
||
has_unqualified = False
|
||
for i in range(items[1]):
|
||
item = EnumValue(key, i)
|
||
name = item[0]
|
||
value = item[1]
|
||
type = item[2]
|
||
if name == unqualified_key_name:
|
||
has_unqualified = True
|
||
run_time = value
|
||
|
||
if not has_unqualified:
|
||
#如果不存在，创建
|
||
run_time = datetime.datetime.now().strftime(time_format)
|
||
key = OpenKey(HKEY_CURRENT_USER, reg_path, access = KEY_SET_VALUE)
|
||
SetValueEx(key, unqualified_key_name, 0, REG_SZ, run_time)
|
||
|
||
#判断时限
|
||
# now = datetime.datetime.strptime('2018-06-22 12:11:51', time_format)
|
||
now = datetime.datetime.now()
|
||
run_time_obj = datetime.datetime.strptime(run_time, time_format)
|
||
print('first run_time:%s' % run_time)
|
||
print('now:%s' % now.strftime(time_format))
|
||
|
||
# qual_days = 1 - (now - run_time_obj).days
|
||
# print('qualified days:%d' % (qual_days))
|
||
print('qualified hours:%d' % (((now - run_time_obj).total_seconds())//3600))
|
||
qual_hours = 18 - ((now - run_time_obj).total_seconds())//3600
|
||
# if qual_days > 0:
|
||
if qual_hours > 0:
|
||
qualified = True
|
||
|
||
CloseKey(key)
|
||
return qualified
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# patchlist = []
|
||
# updates = 'SOFTWARE\\Gutin\\Record'
|
||
# patch = re.compile('InstallDir.+')
|
||
# key = OpenKey(HKEY_CURRENT_USER, updates, access = KEY_READ)
|
||
# items = QueryInfoKey(key)
|
||
# print(items[1])
|
||
# for i in range(items[1]):
|
||
# item = EnumValue(key, i)
|
||
# name = item[0]
|
||
# value = item[1]
|
||
# type = item[2]
|
||
# if name and name.find('InstallDir')>=0:
|
||
# print(item)
|
||
|
||
print(check_run_state())
|
||
|
||
# print(items)
|
||
# CloseKey(key) |