Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Dont depend on win32api in polyglot

Browse Source
This commit is contained in:
Kovid Goyal 2019-06-10 18:21:38 +05:30
parent f1199035f6
commit 057b873933
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/polyglot/builtins.py
View File

@ -151,14 +151,24 @@ else:
return x
if hasattr(sys, 'getwindowsversion'):
import ctypes
gev = ctypes.windll.kernel32.GetEnvironmentVariableW
def getenv(x, default=None):
from win32api import GetEnvironmentVariableW
if isinstance(x, bytes):
x = x.decode('mbcs', 'replace')
ans = GetEnvironmentVariableW(x)
if ans is None:
ans = default
return ans
n = 1024
buf = ctypes.create_unicode_buffer(n)
res = gev(x, buf, n)
if res == 0:
return default
if res > n - 4:
n = res + 8
buf = ctypes.create_unicode_buffer(n)
res = gev(x, buf, n)
if res == 0:
return default
return buf.value
else:
def getenv(x, default=None):
ans = os.getenv(x, default)