text.py

'''! Utilities for manipulating and searching text strings. '''
 
# import os
import re
from os.path import *
from string import *
from pscfpp.file import *
 
 
class Wrapper:
 
   
   def __init__(self, eol = '\n', nIndent = 0):
      self.eol = eol
      self.nIndent = nIndent
      self.column = 0
      self.text = ''
      self.limit = 78 
 
   
   def clear(self):
      self.column = 0
      self.text = ''
 
   
   def append(self, string):
      size = len(string)
      if (self.column + 1 + size > self.limit):
         self.text += self.eol
         self.column = 0
         for i in range(self.nIndent):
            self.text += ' '
         self.column = self.nIndent
      elif (self.column > self.nIndent):
         self.text += ' '
         self.column += 1
      self.text += string
      self.column += size
 
   
   def __str__(self):
      return self.text
 
   
   def __repr__(self):
      return self.text
 
 
class Record:
 
   
   def __init__(self, line):
      if line[-1] == '\n':
         line = line[:-1]
      self.line   = line
      self.spaces  = []
      self.fields  = []
      n = 0
      blank  = True
      gap = ''
      for i in range(len(line)):
         if blank:
            if line[i] != ' ':
               begin = i
               blank = False
            else:
               gap = gap + ' '
         else:
            if line[i] == ' ':
               end = i
               blank = True
               self.fields.append(line[begin:end])
               self.spaces.append(gap)
               gap = ' '
               n += 1
      if not blank:
         end = len(line)
         self.fields.append(line[begin:])
         self.spaces.append(gap)
         n +=1
      self.size = n
 
   
   def __str__(self):
      line = ''
      for i in range(self.size):
         line += self.spaces[i]
         line += self.fields[i]
      return line
 
 
class RecordEditor:
 
   
   def __init__(self):
      self.hasFilter = False
      self.hasNew    = False
      self.isReady   = False
      self.isTest    = True
 
   
   def setFilter(self, filter, field = 0):
      self.filter      = filter
      self.filterField = field
      self.hasFilter   = True
      if self.hasNew:
         self.isReady = True
 
   
   def setNew(self, new, field):
      self.new  = new
      self.dataField = field
      self.hasNew = True
      if self.hasFilter:
         self.isReady = True
 
   
   def setIsTest(self, isTest):
      self.isTest = isTest
 
   
   def editFile(self, filename):
      if (not self.isReady):
         print('RecordEditor is not ready')
         return
 
      oldfile = open(filename, 'r')
      lines   = oldfile.readlines()
      n       = len(lines)
      oldfile.close()
   
      if (not self.isTest): 
         newfile = open (filename, 'w')
      found = False
      for line in lines:
         record = Record(line)
         if (self.filter == record.fields[self.filterField]):
            record.fields[self.dataField] = self.new
         if (not self.isTest):
            newfile.write(str(record) + '\n')
      if (not self.isTest):
         newfile.close()
 
 
 
def readLabelledLine(file, label):
   line = file.readline()
   groups = line.strip().split('=')
   if groups[0].strip() != label:
      print('Error: Expected label = ' + label)
      print('     : Found          = ' + groups[0].strip())
      raise
   if len(groups) > 2:
      print('Error: More than one = sign in line')
      raise
   if len(groups) == 1:
      return ''
   else:
     return groups[1].strip()
 
 
def readLabelledList(file, label):
   line = readLabelledLine(file, label)
   list = []
   if line != '':
      groups = line.split(' ')
      for i in range(len(groups)):
         if groups[i] != '':
            list.append(groups[i])
   return list
 
 
 
class Grep:
 
   
   def __init__(self):
      self.nFilter = 0
      self.filters = []
      self.results = []
 
   
   def addFilter(self, filter):
      self.filters.append(re.compile(filter))
      self.results.append(0)
      self.nFilter += 1
 
   
   def clearResults(self):
      for i in range(self.nFilter):
         self.results[i] = 0
 
   
   def grep(self, filename):
      file = open(filename, 'r')
      lines = file.readlines()
      n     = len(lines)
      file.close()
   
      for line in lines:
         for i in range(self.nFilter):
            if (self.filters[i].search(line)):
               self.results[i] += 1
 
 
 
class FileEditor:
 
   
   def __init__(self):
      self.hasFilter = False
      self.hasNew    = False
      self.hasOld    = False
      self.isReady   = False
      self.isTest    = True
      self.blockSize = 1
 
   
   def setFilter(self, filter):
      self.filter    = re.compile(filter)
      self.hasFilter = True
      if (self.hasOld and self.hasNew):
         self.isReady = True
 
   
   def setOld(self, old):
      self.old    = old
      self.hasOld = True
      if (self.hasFilter and self.hasNew):
         self.isReady = True
 
   
   def setNew(self, new):
      self.new    = new
      self.hasNew = True
      if (self.hasFilter and self.hasOld):
         self.isReady = True
 
   
   def setIsTest(self, isTest):
      self.isTest = isTest
 
   
   def setBlockSize(self, blockSize):
      self.blockSize = blockSize
 
   
   def editFile(self, filename):
     
      if (not self.isReady):
         print('FileEditor is not ready')
         return
 
      oldfile = open(filename, 'r')
      lines   = oldfile.readlines()
      n       = len(lines)
      oldfile.close()
   
      if (not self.isTest): 
         newfile = open (filename, 'w')
      found = False
      for line in lines:
         if (self.filter.search(line)):
            if (not found):
               print(filename)
               found = True
            print(line)
            line = re.sub(self.old, self.new, line);
            print(line)
         if (not self.isTest):
            newfile.write(line)
      if (not self.isTest): 
         newfile.close()
  
   
   def editFileBlocks(self, filename):
     
      if (not self.isReady):
         print('FileEditor is not ready')
         return
 
      oldfile = open(filename, 'r')
      lines   = oldfile.readlines()
      n       = len(lines)
      oldfile.close()
   
      if (not self.isTest): 
         newfile = open (filename, 'w')
 
      found = False
      for i in range(n + 1 - self.blockSize):
         block = join(lines[i:i + self.blockSize], '')
         if (self.filter.search(block)):
            if (not found):
               print(filename)
               found = True
            print(block)
            block = re.sub(self.old, self.new, block)
            print(block)
            if (block[-1] == '\n'):
               block = block[0:-1]
            list  = split(block, '\n')
            for j in range(len(list)):
               list[j] = list[j] + '\n'
            lines[i:i+self.blockSize] = list
      if (not self.isTest):
         for line in lines:
            newfile.write(line)
         newfile.close()
  
   
   def editFiles(self, dirName, pattern):
      if (not self.isReady):
         print('FileEditor is not ready')
         return
      dir = Directory(dirName)
      filenames = dir.filenames(pattern)
      for filename in filenames:
         if isfile(filename): 
            self.editFileBlocks(filename)