mirror of
https://github.com/gryf/ebook-converter.git
synced 2025-12-28 20:42:27 +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.
88 lines
2.6 KiB
Python
88 lines
2.6 KiB
Python
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
__license__ = 'GPL v3'
|
|
__copyright__ = '2008, Kovid Goyal <kovid at kovidgoyal.net>'
|
|
|
|
''' Code to manage ebook library'''
|
|
|
|
|
|
import os
|
|
from ebook_converter.polyglot.builtins import range
|
|
|
|
|
|
def db(path=None, read_only=False):
|
|
from ebook_converter.db.legacy import LibraryDatabase
|
|
from ebook_converter.utils.config import prefs
|
|
return LibraryDatabase(os.path.expanduser(path) if path else prefs['library_path'],
|
|
read_only=read_only)
|
|
|
|
|
|
def generate_test_db(library_path, # {{{
|
|
num_of_records=20000,
|
|
num_of_authors=6000,
|
|
num_of_tags=10000,
|
|
tag_length=7,
|
|
author_length=7,
|
|
title_length=10,
|
|
max_authors=10,
|
|
max_tags=10
|
|
):
|
|
import random, string, os, sys, time
|
|
from ebook_converter.constants import preferred_encoding
|
|
|
|
if not os.path.exists(library_path):
|
|
os.makedirs(library_path)
|
|
|
|
letters = string.letters.decode(preferred_encoding)
|
|
|
|
def randstr(length):
|
|
return ''.join(random.choice(letters) for i in
|
|
range(length))
|
|
|
|
all_tags = [randstr(tag_length) for j in range(num_of_tags)]
|
|
print('Generated', num_of_tags, 'tags')
|
|
all_authors = [randstr(author_length) for j in range(num_of_authors)]
|
|
print('Generated', num_of_authors, 'authors')
|
|
all_titles = [randstr(title_length) for j in range(num_of_records)]
|
|
print('Generated', num_of_records, 'titles')
|
|
|
|
testdb = db(library_path)
|
|
|
|
print('Creating', num_of_records, 'records...')
|
|
|
|
start = time.time()
|
|
|
|
for i, title in enumerate(all_titles):
|
|
print(i+1, end=' ')
|
|
sys.stdout.flush()
|
|
authors = random.randint(1, max_authors)
|
|
authors = [random.choice(all_authors) for i in range(authors)]
|
|
tags = random.randint(0, max_tags)
|
|
tags = [random.choice(all_tags) for i in range(tags)]
|
|
from ebook_converter.ebooks.metadata.book.base import Metadata
|
|
mi = Metadata(title, authors)
|
|
mi.tags = tags
|
|
testdb.import_book(mi, [])
|
|
|
|
t = time.time() - start
|
|
print('\nGenerated', num_of_records, 'records in:', t, 'seconds')
|
|
print('Time per record:', t/num_of_records)
|
|
# }}}
|
|
|
|
|
|
def current_library_path():
|
|
from ebook_converter.utils.config import prefs
|
|
path = prefs['library_path']
|
|
if path:
|
|
path = path.replace('\\', '/')
|
|
while path.endswith('/'):
|
|
path = path[:-1]
|
|
return path
|
|
|
|
|
|
def current_library_name():
|
|
import posixpath
|
|
path = current_library_path()
|
|
if path:
|
|
return posixpath.basename(path)
|