command.py

"""! Module for processing PSCF command scripts. """
 
from pscfpp.text import *
 
 
class Script:
 
   
   def __init__(self, filename = None):
      self._commands = []
      self._indices = {}
      if (filename != None):
         self.read(filename)
 
   
   def read(self, filename):
 
      # Determine type of parameter in
      if isinstance(filename, str):
         file = open(filename, 'r')
      elif isinstance(filename, file):
         file = filename
      else:
         raise Exception("Invalid parameter in")
 
      # Split file into list of line strings
      lines = file.readlines();
 
      # Process lines
      i = 0
      finish = False
      while i < len(lines) and not finish:
         line = lines[i]
         if not isinstance(line, str):
            raise Exception('Command line object is not a string')
         command = Command(line)
 
         # Add command to list self._commands
         self._commands.append(command)
 
         # Add index i to dictionary self._indices
         label = command.label
         if label in self._indices:
            index = self._indices[label]
            if isinstance(index, int):
               self._indices[label] = [index]
            elif not instance(index, list):
               raise Exception('Values of _indices must be int or list')
            self._indices[label].append(i)
         else:
            self._indices[label] = i
 
         if (command.label == 'FINISH'):
            finish = True
         i += 1
 
   
   def __str__(self):
      list = []
      for command in self._commands:
         list.append(str(command))
      return '\n'.join(list)
 
   
   def __getitem__(self, key):
      if isinstance(key, int):
         return self._commands[key]
      elif isinstance(key, str):
         if key in self._indices:
            index = self._indices[key]
            if isinstance(index, int):
               return self._commands[index]
            elif isinstance(index, list):
               clist = []
               for j in index:
                  clist.append(self._commands[j])
               return clist
      else:
         raise Exception("Key must an int or a command string")
 
   
   def __len__(self):
      return len(self._commands)
 
 
class Command(Record):
 
   
   def __init__(self, line):
      Record.__init__(self, line)
      self.label = self.fields[0]
 
   
   def nParam(self):
      return (len(self.fields) - 1)
 
   
   def param(self, i = 0):
      return self.fields[i+1]
 
   
   def setParam(self, value, i = 0):
      self.fields[i+1] = str(value)