mirror of
https://github.com/noDRM/DeDRM_tools.git
synced 2024-11-05 05:26:09 +06:00
afa4ac5716
THIS IS ON THE MASTER BRANCH. The Master branch will be Python 3.0 from now on. While Python 2.7 support will not be deliberately broken, all efforts should now focus on Python 3.0 compatibility. I can see a lot of work has been done. There's more to do. I've bumped the version number of everything I came across to the next major number for Python 3.0 compatibility indication. Thanks everyone. I hope to update here at least once a week until we have a stable 7.0 release for calibre 5.0
78 lines
2.9 KiB
Python
78 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
# vim:ts=4:sw=4:softtabstop=4:smarttab:expandtab
|
|
|
|
import sys
|
|
import os, os.path
|
|
import shutil
|
|
|
|
class SimplePrefsError(Exception):
|
|
pass
|
|
|
|
class SimplePrefs(object):
|
|
def __init__(self, target, description):
|
|
self.prefs = {}
|
|
self.key2file={}
|
|
self.file2key={}
|
|
for keyfilemap in description:
|
|
[key, filename] = keyfilemap
|
|
self.key2file[key] = filename
|
|
self.file2key[filename] = key
|
|
self.target = target + 'Prefs'
|
|
if sys.platform.startswith('win'):
|
|
import _winreg as winreg
|
|
regkey = winreg.OpenKey(winreg.HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders\\")
|
|
path = winreg.QueryValueEx(regkey, 'Local AppData')[0]
|
|
prefdir = path + os.sep + self.target
|
|
elif sys.platform.startswith('darwin'):
|
|
home = os.getenv('HOME')
|
|
prefdir = os.path.join(home,'Library','Preferences','org.' + self.target)
|
|
else:
|
|
# linux and various flavors of unix
|
|
home = os.getenv('HOME')
|
|
prefdir = os.path.join(home,'.' + self.target)
|
|
if not os.path.exists(prefdir):
|
|
os.makedirs(prefdir)
|
|
self.prefdir = prefdir
|
|
self.prefs['dir'] = self.prefdir
|
|
self._loadPreferences()
|
|
|
|
def _loadPreferences(self):
|
|
filenames = os.listdir(self.prefdir)
|
|
for filename in filenames:
|
|
if filename in self.file2key:
|
|
key = self.file2key[filename]
|
|
filepath = os.path.join(self.prefdir,filename)
|
|
if os.path.isfile(filepath):
|
|
try :
|
|
data = file(filepath,'rb').read()
|
|
self.prefs[key] = data
|
|
except Exception as e:
|
|
pass
|
|
|
|
def getPreferences(self):
|
|
return self.prefs
|
|
|
|
def setPreferences(self, newprefs={}):
|
|
if 'dir' not in newprefs:
|
|
raise SimplePrefsError('Error: Attempt to Set Preferences in unspecified directory')
|
|
if newprefs['dir'] != self.prefs['dir']:
|
|
raise SimplePrefsError('Error: Attempt to Set Preferences in unspecified directory')
|
|
for key in newprefs:
|
|
if key != 'dir':
|
|
if key in self.key2file:
|
|
filename = self.key2file[key]
|
|
filepath = os.path.join(self.prefdir,filename)
|
|
data = newprefs[key]
|
|
if data != None:
|
|
data = str(data)
|
|
if data == None or data == '':
|
|
if os.path.exists(filepath):
|
|
os.remove(filepath)
|
|
else:
|
|
try:
|
|
file(filepath,'wb').write(data)
|
|
except Exception as e:
|
|
pass
|
|
self.prefs = newprefs
|
|
return
|