Initial import

ebook_converter/ebooks/rtf2xml/table_info.py

@@ -0,0 +1,88 @@
+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 calibre.ebooks.rtf2xml import copy
+from calibre.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)