diff --git a/micorsoft_api.py b/micorsoft_api.py index dca4432..5a8dbe8 100644 --- a/micorsoft_api.py +++ b/micorsoft_api.py @@ -87,10 +87,10 @@ def is_same_os(a, b): def get_build_number(): k = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r'SOFTWARE\Microsoft\Windows NT\CurrentVersion') - major = winreg.QueryValueEx(k, 'CurrentMajorVersionNumber') - minor = winreg.QueryValueEx(k, 'CurrentMinorVersionNumber') - build_number = winreg.QueryValueEx(k, 'CurrentBuildNumber') - ubr = winreg.QueryValueEx(k, 'UBR') + major = winreg.QueryValueEx(k, 'CurrentMajorVersionNumber')[0] + minor = winreg.QueryValueEx(k, 'CurrentMinorVersionNumber')[0] + build_number = winreg.QueryValueEx(k, 'CurrentBuildNumber')[0] + ubr = winreg.QueryValueEx(k, 'UBR')[0] return f'{major}.{minor}.{build_number}.{ubr}' @@ -189,4 +189,4 @@ def download_security_update(cve, product_id, out_file_path): if __name__ == '__main__': - download_security_update('CVE-2022-38044', 11800, r'e:\CVE-2022-38044.msu') \ No newline at end of file + download_security_update('CVE-2022-38044', 11800, r'e:\CVE-2022-38044.msu')