mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-01 15:32:26 +01:00
gryf
0f9792df36
Here is the first batch of modules, which are needed for converting several formats to LRF. Some of the logic has been change, more cleanups will follow.
105 lines
3.0 KiB
Python
105 lines
3.0 KiB
Python
__license__ = 'GPL v3'
|
|
__copyright__ = '2009, Kovid Goyal <kovid@kovidgoyal.net>'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
|
|
import sys, os
|
|
|
|
from ebook_converter import config_dir
|
|
from ebook_converter.polyglot.builtins import builtins
|
|
|
|
|
|
user_dir = os.path.join(config_dir, 'resources')
|
|
|
|
|
|
class PathResolver(object):
|
|
|
|
def __init__(self):
|
|
sys.resources_location = os.path.abspath(os.path.dirname(__file__))
|
|
self.locations = [sys.resources_location]
|
|
self.cache = {}
|
|
|
|
def suitable(path):
|
|
try:
|
|
return os.path.exists(path) and os.path.isdir(path) and \
|
|
os.listdir(path)
|
|
except:
|
|
pass
|
|
return False
|
|
|
|
self.default_path = sys.resources_location
|
|
|
|
dev_path = os.environ.get('CALIBRE_DEVELOP_FROM', None)
|
|
self.using_develop_from = False
|
|
if dev_path is not None:
|
|
dev_path = os.path.join(os.path.abspath(
|
|
os.path.dirname(dev_path)), 'resources')
|
|
if suitable(dev_path):
|
|
self.locations.insert(0, dev_path)
|
|
self.default_path = dev_path
|
|
self.using_develop_from = True
|
|
|
|
self.user_path = None
|
|
if suitable(user_dir):
|
|
self.locations.insert(0, user_dir)
|
|
self.user_path = user_dir
|
|
|
|
def __call__(self, path, allow_user_override=True):
|
|
path = path.replace(os.sep, '/')
|
|
key = (path, allow_user_override)
|
|
ans = self.cache.get(key, None)
|
|
if ans is None:
|
|
for base in self.locations:
|
|
if not allow_user_override and base == self.user_path:
|
|
continue
|
|
fpath = os.path.join(base, *path.split('/'))
|
|
if os.path.exists(fpath):
|
|
ans = fpath
|
|
break
|
|
|
|
if ans is None:
|
|
ans = os.path.join(self.default_path, *path.split('/'))
|
|
|
|
self.cache[key] = ans
|
|
|
|
return ans
|
|
|
|
def set_data(self, path, data=None):
|
|
self.cache.pop((path, True), None)
|
|
fpath = os.path.join(user_dir, *path.split('/'))
|
|
if data is None:
|
|
if os.path.exists(fpath):
|
|
os.remove(fpath)
|
|
else:
|
|
base = os.path.dirname(fpath)
|
|
if not os.path.exists(base):
|
|
os.makedirs(base)
|
|
with open(fpath, 'wb') as f:
|
|
f.write(data)
|
|
|
|
|
|
_resolver = PathResolver()
|
|
|
|
|
|
def get_path(path, data=False, allow_user_override=True):
|
|
fpath = _resolver(path, allow_user_override=allow_user_override)
|
|
|
|
if data:
|
|
with open(fpath, 'rb') as f:
|
|
return f.read()
|
|
return fpath
|
|
|
|
|
|
def get_image_path(path, data=False, allow_user_override=True):
|
|
if not path:
|
|
return get_path('images', allow_user_override=allow_user_override)
|
|
return get_path('images/'+path, data=data, allow_user_override=allow_user_override)
|
|
|
|
|
|
def set_data(path, data=None):
|
|
return _resolver.set_data(path, data)
|
|
|
|
|
|
builtins.__dict__['P'] = get_path
|
|
builtins.__dict__['I'] = get_image_path
|