mirror of
https://github.com/noDRM/DeDRM_tools.git
synced 2024-12-29 04:37:39 +06:00
107 lines
3.9 KiB
Python
Executable File
107 lines
3.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
|
|
|
|
__license__ = 'GPL v3'
|
|
|
|
# Standard Python modules.
|
|
import os, sys
|
|
import traceback
|
|
|
|
|
|
#@@CALIBRE_COMPAT_CODE@@
|
|
|
|
|
|
try:
|
|
from calibre.utils.config import JSONConfig
|
|
except:
|
|
from standalone.jsonconfig import JSONConfig
|
|
|
|
from __init__ import PLUGIN_NAME
|
|
|
|
class DeDRM_Prefs():
|
|
def __init__(self, json_path=None):
|
|
if json_path is None:
|
|
JSON_PATH = os.path.join("plugins", PLUGIN_NAME.strip().lower().replace(' ', '_') + '.json')
|
|
else:
|
|
JSON_PATH = json_path
|
|
|
|
self.dedrmprefs = JSONConfig(JSON_PATH)
|
|
|
|
self.dedrmprefs.defaults['configured'] = False
|
|
self.dedrmprefs.defaults['deobfuscate_fonts'] = True
|
|
self.dedrmprefs.defaults['remove_watermarks'] = False
|
|
self.dedrmprefs.defaults['bandnkeys'] = {}
|
|
self.dedrmprefs.defaults['adeptkeys'] = {}
|
|
self.dedrmprefs.defaults['ereaderkeys'] = {}
|
|
self.dedrmprefs.defaults['kindlekeys'] = {}
|
|
self.dedrmprefs.defaults['androidkeys'] = {}
|
|
self.dedrmprefs.defaults['pids'] = []
|
|
self.dedrmprefs.defaults['serials'] = []
|
|
self.dedrmprefs.defaults['lcp_passphrases'] = []
|
|
self.dedrmprefs.defaults['adobe_pdf_passphrases'] = []
|
|
self.dedrmprefs.defaults['adobewineprefix'] = ""
|
|
self.dedrmprefs.defaults['kindlewineprefix'] = ""
|
|
|
|
# initialise
|
|
# we must actually set the prefs that are dictionaries and lists
|
|
# to empty dictionaries and lists, otherwise we are unable to add to them
|
|
# as then it just adds to the (memory only) dedrmprefs.defaults versions!
|
|
if self.dedrmprefs['bandnkeys'] == {}:
|
|
self.dedrmprefs['bandnkeys'] = {}
|
|
if self.dedrmprefs['adeptkeys'] == {}:
|
|
self.dedrmprefs['adeptkeys'] = {}
|
|
if self.dedrmprefs['ereaderkeys'] == {}:
|
|
self.dedrmprefs['ereaderkeys'] = {}
|
|
if self.dedrmprefs['kindlekeys'] == {}:
|
|
self.dedrmprefs['kindlekeys'] = {}
|
|
if self.dedrmprefs['androidkeys'] == {}:
|
|
self.dedrmprefs['androidkeys'] = {}
|
|
if self.dedrmprefs['pids'] == []:
|
|
self.dedrmprefs['pids'] = []
|
|
if self.dedrmprefs['serials'] == []:
|
|
self.dedrmprefs['serials'] = []
|
|
if self.dedrmprefs['lcp_passphrases'] == []:
|
|
self.dedrmprefs['lcp_passphrases'] = []
|
|
if self.dedrmprefs['adobe_pdf_passphrases'] == []:
|
|
self.dedrmprefs['adobe_pdf_passphrases'] = []
|
|
|
|
def __getitem__(self,kind = None):
|
|
if kind is not None:
|
|
return self.dedrmprefs[kind]
|
|
return self.dedrmprefs
|
|
|
|
def set(self, kind, value):
|
|
self.dedrmprefs[kind] = value
|
|
|
|
def writeprefs(self,value = True):
|
|
self.dedrmprefs['configured'] = value
|
|
|
|
def addnamedvaluetoprefs(self, prefkind, keyname, keyvalue):
|
|
try:
|
|
if keyvalue not in self.dedrmprefs[prefkind].values():
|
|
# ensure that the keyname is unique
|
|
# by adding a number (starting with 2) to the name if it is not
|
|
namecount = 1
|
|
newname = keyname
|
|
while newname in self.dedrmprefs[prefkind]:
|
|
namecount += 1
|
|
newname = "{0:s}_{1:d}".format(keyname,namecount)
|
|
# add to the preferences
|
|
self.dedrmprefs[prefkind][newname] = keyvalue
|
|
return (True, newname)
|
|
except:
|
|
traceback.print_exc()
|
|
pass
|
|
return (False, keyname)
|
|
|
|
def addvaluetoprefs(self, prefkind, prefsvalue):
|
|
# ensure the keyvalue isn't already in the preferences
|
|
try:
|
|
if prefsvalue not in self.dedrmprefs[prefkind]:
|
|
self.dedrmprefs[prefkind].append(prefsvalue)
|
|
return True
|
|
except:
|
|
traceback.print_exc()
|
|
return False
|