diff options
Diffstat (limited to 'DeDRM_plugin/ignoblekeyNookStudy.py')
| -rw-r--r-- | DeDRM_plugin/ignoblekeyNookStudy.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/DeDRM_plugin/ignoblekeyNookStudy.py b/DeDRM_plugin/ignoblekeyNookStudy.py index 6a5f1cf..4cf5246 100644 --- a/DeDRM_plugin/ignoblekeyNookStudy.py +++ b/DeDRM_plugin/ignoblekeyNookStudy.py @@ -54,15 +54,26 @@ def getNookLogFiles(): paths = set() if 'LOCALAPPDATA' in os.environ.keys(): # Python 2.x does not return unicode env. Use Python 3.x - path = winreg.ExpandEnvironmentStrings("%LOCALAPPDATA%") + if sys.version_info[0] == 2: + path = winreg.ExpandEnvironmentStrings(u"%LOCALAPPDATA%") + else: + path = winreg.ExpandEnvironmentStrings("%LOCALAPPDATA%") if os.path.isdir(path): paths.add(path) if 'USERPROFILE' in os.environ.keys(): # Python 2.x does not return unicode env. Use Python 3.x - path = winreg.ExpandEnvironmentStrings("%USERPROFILE%")+"\\AppData\\Local" + if sys.version_info[0] == 2: + path = winreg.ExpandEnvironmentStrings(u"%USERPROFILE%")+u"\\AppData\\Local" + else: + path = winreg.ExpandEnvironmentStrings("%USERPROFILE%")+"\\AppData\\Local" + if os.path.isdir(path): paths.add(path) - path = winreg.ExpandEnvironmentStrings("%USERPROFILE%")+"\\AppData\\Roaming" + + if sys.version_info[0] == 2: + path = winreg.ExpandEnvironmentStrings(u"%USERPROFILE%")+u"\\AppData\\Roaming" + else: + path = winreg.ExpandEnvironmentStrings("%USERPROFILE%")+"\\AppData\\Roaming" if os.path.isdir(path): paths.add(path) # User Shell Folders show take precedent over Shell Folders if present |