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

Function to get a best guess book path from a book id

Browse Source
This commit is contained in:
Kovid Goyal 2023-04-06 13:51:35 +05:30
parent dd186d060d
commit acc1629efb
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/calibre/db/backend.py
View File

@ -55,6 +55,7 @@ from polyglot.builtins import (
# }}} # }}}
BOOK_ID_PATH_TEMPLATE = ' ({})'
CUSTOM_DATA_TYPES = frozenset(('rating', 'text', 'comments', 'datetime', CUSTOM_DATA_TYPES = frozenset(('rating', 'text', 'comments', 'datetime',
'int', 'float', 'bool', 'series', 'composite', 'enumeration')) 'int', 'float', 'bool', 'series', 'composite', 'enumeration'))
WINDOWS_RESERVED_NAMES = frozenset('CON PRN AUX NUL COM1 COM2 COM3 COM4 COM5 COM6 COM7 COM8 COM9 LPT1 LPT2 LPT3 LPT4 LPT5 LPT6 LPT7 LPT8 LPT9'.split()) WINDOWS_RESERVED_NAMES = frozenset('CON PRN AUX NUL COM1 COM2 COM3 COM4 COM5 COM6 COM7 COM8 COM9 LPT1 LPT2 LPT3 LPT4 LPT5 LPT6 LPT7 LPT8 LPT9'.split())
@ -425,13 +426,8 @@ class DB:
restore_all_prefs=False, progress_callback=lambda x, y:True, restore_all_prefs=False, progress_callback=lambda x, y:True,
load_user_formatter_functions=True): load_user_formatter_functions=True):
self.is_closed = False self.is_closed = False
try: if isbytestring(library_path):
if isbytestring(library_path): library_path = library_path.decode(filesystem_encoding)
library_path = library_path.decode(filesystem_encoding)
except:
import traceback
traceback.print_exc()
self.field_metadata = FieldMetadata() self.field_metadata = FieldMetadata()
self.library_path = os.path.abspath(library_path) self.library_path = os.path.abspath(library_path)
@ -1338,7 +1334,7 @@ class DB:
''' '''
Construct the directory name for this book based on its metadata. Construct the directory name for this book based on its metadata.
''' '''
book_id = ' (%d)' % book_id book_id = BOOK_ID_PATH_TEMPLATE.format(book_id)
l = self.PATH_LIMIT - (len(book_id) // 2) - 2 l = self.PATH_LIMIT - (len(book_id) // 2) - 2
author = ascii_filename(author)[:l] author = ascii_filename(author)[:l]
title = ascii_filename(title.lstrip())[:l].rstrip() title = ascii_filename(title.lstrip())[:l].rstrip()
@ -1433,8 +1429,22 @@ class DB:
pprint.pprint(table.metadata) pprint.pprint(table.metadata)
raise raise
def format_abspath(self, book_id, fmt, fname, path): def find_path_for_book(self, book_id):
path = os.path.join(self.library_path, path) q = BOOK_ID_PATH_TEMPLATE.format(book_id)
for author_dir in os.scandir(self.library_path):
if not author_dir.is_dir():
continue
try:
book_dir_iter = os.scandir(author_dir.path)
except OSError:
pass
else:
for book_dir in book_dir_iter:
if book_dir.name.endswith(q) and book_dir.is_dir():
return book_dir.path
def format_abspath(self, book_id, fmt, fname, book_path):
path = os.path.join(self.library_path, book_path)
fmt = ('.' + fmt.lower()) if fmt else '' fmt = ('.' + fmt.lower()) if fmt else ''
fmt_path = os.path.join(path, fname+fmt) fmt_path = os.path.join(path, fname+fmt)
if os.path.exists(fmt_path): if os.path.exists(fmt_path):