mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-04 09:54:11 +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.
122 lines
4.1 KiB
Python
122 lines
4.1 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 sys, os
|
|
from ebook_converter.polyglot.builtins import raw_input
|
|
from . import open_for_read, open_for_write
|
|
# , codecs
|
|
|
|
|
|
class Output:
|
|
"""
|
|
Output file
|
|
"""
|
|
|
|
def __init__(self,
|
|
file,
|
|
orig_file,
|
|
output_dir=None,
|
|
out_file=None,
|
|
no_ask=True
|
|
):
|
|
"""
|
|
Required:
|
|
'file' -- xml file ready to output
|
|
orig_file -- original rtf file
|
|
Optional:
|
|
output_file -- the file to output to
|
|
Returns:
|
|
nothing
|
|
"""
|
|
self.__file = file
|
|
self.__orig_file = orig_file
|
|
self.__output_dir = output_dir
|
|
self.__no_ask = no_ask
|
|
self.__out_file = out_file
|
|
|
|
def output(self):
|
|
"""
|
|
Required:
|
|
nothing
|
|
Returns:
|
|
nothing
|
|
Logic:
|
|
output the line to the screen if no output file given. Otherwise, output to
|
|
the file.
|
|
"""
|
|
if self.__output_dir:
|
|
self.__output_to_dir_func()
|
|
elif self.__out_file:
|
|
self.__output_to_file_func()
|
|
# self.__output_xml(self.__file, self.__out_file)
|
|
else:
|
|
self.__output_to_standard_func()
|
|
|
|
def __output_to_dir_func(self):
|
|
"""
|
|
Requires:
|
|
nothing
|
|
Returns:
|
|
nothing
|
|
Logic:
|
|
Create a file within the output directory.
|
|
Read one file at a time. Output line to the newly-created file.
|
|
"""
|
|
base_name = os.path.basename(self.__orig_file)
|
|
base_name, ext = os.path.splitext(base_name)
|
|
output_file = os.path.join(self.__output_dir, '%s.xml' % base_name)
|
|
# change if user wants to output to a specific file
|
|
if self.__out_file:
|
|
output_file = os.path.join(self.__output_dir, self.__out_file)
|
|
user_response = 'o'
|
|
if os.path.isfile(output_file) and not self.__no_ask:
|
|
msg = 'Do you want to overwrite %s?\n' % output_file
|
|
msg += ('Type "o" to overwrite.\n'
|
|
'Type any other key to print to standard output.\n')
|
|
sys.stderr.write(msg)
|
|
user_response = raw_input()
|
|
if user_response == 'o':
|
|
with open_for_read(self.__file) as read_obj:
|
|
with open_for_write(self.output_file) as write_obj:
|
|
for line in read_obj:
|
|
write_obj.write(line)
|
|
else:
|
|
self.__output_to_standard_func()
|
|
|
|
def __output_to_file_func(self):
|
|
"""
|
|
Required:
|
|
nothing
|
|
Returns:
|
|
nothing
|
|
Logic:
|
|
read one line at a time. Output to standard
|
|
"""
|
|
with open_for_read(self.__file) as read_obj:
|
|
with open_for_write(self.__out_file) as write_obj:
|
|
for line in read_obj:
|
|
write_obj.write(line)
|
|
|
|
def __output_to_standard_func(self):
|
|
"""
|
|
Required:
|
|
nothing
|
|
Returns:
|
|
nothing
|
|
Logic:
|
|
read one line at a time. Output to standard
|
|
"""
|
|
with open_for_read(self.__file) as read_obj:
|
|
for line in read_obj:
|
|
sys.stdout.write(line)
|