Initial import

ebook_converter/ebooks/rtf2xml/check_brackets.py

@@ -0,0 +1,62 @@
+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.                            #
+#                                                                       #
+#                                                                       #
+#########################################################################
+
+
+from . import open_for_read
+
+
+class CheckBrackets:
+    """Check that brackets match up"""
+
+    def __init__(self, bug_handler=None, file=None):
+        self.__file=file
+        self.__bug_handler = bug_handler
+        self.__bracket_count=0
+        self.__ob_count = 0
+        self.__cb_count = 0
+        self.__open_bracket_num = []
+
+    def open_brack(self, line):
+        num = line[-5:-1]
+        self.__open_bracket_num.append(num)
+        self.__bracket_count += 1
+
+    def close_brack(self, line):
+        num = line[-5:-1]
+        try:
+            last_num = self.__open_bracket_num.pop()
+        except:
+            return False
+        if num != last_num:
+            return False
+        self.__bracket_count -= 1
+        return True
+
+    def check_brackets(self):
+        line_count = 0
+        with open_for_read(self.__file) as read_obj:
+            for line in read_obj:
+                line_count += 1
+                self.__token_info = line[:16]
+                if self.__token_info == 'ob<nu<open-brack':
+                    self.open_brack(line)
+                if self.__token_info == 'cb<nu<clos-brack':
+                    if not self.close_brack(line):
+                        return (False, "closed bracket doesn't match, line %s" % line_count)
+
+        if self.__bracket_count != 0:
+            msg = ('At end of file open and closed brackets don\'t match\n'
+                        'total number of brackets is %s') % self.__bracket_count
+            return (False, msg)
+        return (True, "Brackets match!")