mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-08 04:34:12 +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.
89 lines
3.5 KiB
Python
89 lines
3.5 KiB
Python
from __future__ import unicode_literals, absolute_import, print_function, division
|
|
#########################################################################
|
|
# #
|
|
# #
|
|
# copyright 2002 Paul Henry Tremblay #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU #
|
|
# General Public License for more details. #
|
|
# #
|
|
# #
|
|
#########################################################################
|
|
import os
|
|
from ebook_converter.ebooks.rtf2xml import copy
|
|
from ebook_converter.ptempfile import better_mktemp
|
|
from . import open_for_read, open_for_write
|
|
|
|
# note to self. This is the first module in which I use tempfile. A good idea?
|
|
"""
|
|
"""
|
|
|
|
|
|
class TableInfo:
|
|
"""
|
|
Insert table data for tables.
|
|
Logic:
|
|
"""
|
|
|
|
def __init__(self,
|
|
in_file,
|
|
bug_handler,
|
|
table_data,
|
|
copy=None,
|
|
run_level=1,):
|
|
"""
|
|
Required:
|
|
'file'--file to parse
|
|
'table_data' -- a dictionary for each table.
|
|
Optional:
|
|
'copy'-- whether to make a copy of result for debugging
|
|
'temp_dir' --where to output temporary results (default is
|
|
directory from which the script is run.)
|
|
Returns:
|
|
nothing
|
|
"""
|
|
self.__file = in_file
|
|
self.__bug_handler = bug_handler
|
|
self.__copy = copy
|
|
self.__table_data = table_data
|
|
self.__run_level = run_level
|
|
self.__write_to = better_mktemp()
|
|
# self.__write_to = 'table_info.data'
|
|
|
|
def insert_info(self):
|
|
"""
|
|
"""
|
|
read_obj = open_for_read(self.__file)
|
|
self.__write_obj = open_for_write(self.__write_to)
|
|
line_to_read = 1
|
|
while line_to_read:
|
|
line_to_read = read_obj.readline()
|
|
line = line_to_read
|
|
if line == 'mi<mk<tabl-start\n':
|
|
if len(self.__table_data) > 0:
|
|
table_dict = self.__table_data[0]
|
|
self.__write_obj.write('mi<tg<open-att__<table')
|
|
keys = table_dict.keys()
|
|
for key in keys:
|
|
self.__write_obj.write('<%s>%s' % (key, table_dict[key]))
|
|
self.__write_obj.write('\n')
|
|
self.__table_data = self.__table_data[1:]
|
|
else:
|
|
# this shouldn't happen!
|
|
if self.__run_level > 3:
|
|
msg = 'Not enough data for each table\n'
|
|
raise self.__bug_handler(msg)
|
|
self.__write_obj.write('mi<tg<open______<table\n')
|
|
elif line == 'mi<mk<table-end_\n':
|
|
self.__write_obj.write('mi<tg<close_____<table\n')
|
|
self.__write_obj.write(line)
|
|
read_obj.close()
|
|
self.__write_obj.close()
|
|
copy_obj = copy.Copy(bug_handler=self.__bug_handler)
|
|
if self.__copy:
|
|
copy_obj.copy_file(self.__write_to, "table_info.data")
|
|
copy_obj.rename(self.__write_to, self.__file)
|
|
os.remove(self.__write_to)
|