Return to spellcheck.py CVS log

Up to [Repository] / kupuMPIWG / python

Annotation of kupuMPIWG/python/spellcheck.py, revision 1.1.1.1

1.1       dwinter     1: #!/usr/bin/python
                      2: 
                      3: """SpellChecker for Kupu"""
                      4: 
                      5: COMMAND = 'aspell -a'
                      6: 
                      7: import popen2, re
                      8: 
                      9: try:
                     10:     from Globals import ClassSecurityInfo
                     11: except ImportError:
                     12:     pass
                     13: else:
                     14:     # hmmm... Zope 2...
                     15:     __allow_access_to_unprotected_subobjects__ = 1
                     16: 
                     17: class SpellChecker:
                     18:     """Simple spell checker, uses ispell (or aspell) with pipes"""
                     19: 
                     20:     __allow_access_to_unprotected_subobjects__ = 1
                     21: 
                     22:     reg_unknown = re.compile('^& (.*?) \d* \d*: (.*)$', re.U)
                     23:     reg_unknown_no_replacement = re.compile('^\# (.*?) \d*.*$', re.U)
                     24: 
                     25:     def __init__(self):
                     26:         self.chout, self.chin = popen2.popen2(COMMAND)
                     27:         # throw away intro
                     28:         self.read_line()
                     29: 
                     30:     def __del__(self):
                     31:         self.chout.close()
                     32:         self.chin.close()
                     33: 
                     34:     def check(self, text):
                     35:         """checks a line of text
                     36:         
                     37:             returns None if spelling was okay, and an HTML string with words 
                     38:             that weren't recognized marked (with a span class="wrong_spelling")
                     39:         """
                     40:         result = {}
                     41:         for line in text.split('\n'):
                     42:             line = line.strip()
                     43:             if line:
                     44:                 self.write_line(line)
                     45:             while 1:
                     46:                 resline = self.read_line()
                     47:                 if not resline.strip():
                     48:                     break
                     49:                 if resline.strip() != '*':
                     50:                     match = self.reg_unknown.match(resline)
                     51:                     have_replacement = True
                     52:                     if not match:
                     53:                         match = self.reg_unknown_no_replacement.match(resline)
                     54:                         have_replacement = False
                     55:                     assert match, 'Unknown formatted line: %s' % resline
                     56:                     word = match.group(1)
                     57:                     if result.has_key(word):
                     58:                         continue
                     59:                     replacements = []
                     60:                     if have_replacement:
                     61:                         replacements = match.group(2).split(', ')
                     62:                     result[word] = replacements
                     63:         return result
                     64: 
                     65:     def read_line(self):
                     66:         buf = []
                     67:         while 1:
                     68:             char = self.read_char()
                     69:             if char == '\n':
                     70:                 return ''.join(buf)
                     71:             buf.append(char)
                     72: 
                     73:     def write_line(self, line):
                     74:         try:
                     75:             self.chin.write('%s\n' % line)
                     76:             self.chin.flush()
                     77:             return
                     78:         except IOError:
                     79:             self.reconnect()
                     80:             self.chin.write('%s\n' % line)
                     81:             self.chin.flush()
                     82:             return
                     83:         raise
                     84: 
                     85:     def read_char(self):
                     86:         try:
                     87:             return self.chout.read(1)
                     88:         except IOError:
                     89:             self.reconnect()
                     90:             return self.chout.read(1)
                     91:         raise
                     92: 
                     93:     def reconnect(self):
                     94:         try:
                     95:             self.chout.close()
                     96:         except IOError:
                     97:             pass
                     98:         try:
                     99:             self.chin.close()
                    100:         except IOError:
                    101:             pass
                    102:         self.chout, self.chin = popen2.popen2(COMMAND)
                    103: 
                    104: def format_result(result):
                    105:     """convert the result dict to XML"""
                    106:     buf = ['<?xml version="1.0" encoding="UTF-8" ?>\n<spellcheck_result>']
                    107:     for key, value in result.items():
                    108:         buf.append('<incorrect><word>')
                    109:         buf.append(key)
                    110:         buf.append('</word><replacements>')
                    111:         buf.append(' '.join(value))
                    112:         buf.append('</replacements></incorrect>')
                    113:     buf.append('</spellcheck_result>')
                    114:     return ''.join(buf)
                    115: 
                    116: if __name__ == '__main__':
                    117:     c = SpellChecker()
                    118:     while 1:
                    119:         line = raw_input('Enter text to check: ')
                    120:         if line == 'q':
                    121:             break
                    122:         ret = c.check(line)
                    123:         if ret is None:
                    124:             print 'okay'
                    125:         else:
                    126:             print ret