terminator/terminatorlib/configfile.py

#!/usr/local/bin/python

import re
from terminatorlib.config import dbg, debug

def group(*choices): return '(' + '|'.join(choices) + ')'
def any(*choices): return group(*choices) + '*'
def maybe(*choices): return group(*choices) + '?'

Newline = re.compile(r'[\r\n]+')
Whitespace = r'[ \f\t]*'
Comment = r'#[^\r\n]*'
Ignore  = re.compile(Whitespace + maybe(Comment) + maybe(r'[\r\n]+') + '$')

WhitespaceRE = re.compile(Whitespace)
CommentRE = re.compile(Comment)

QuotedStrings = {"'": re.compile(r"'([^'\r\n]*)'"), '"': re.compile(r'"([^"\r\n]*)"')}

Section = re.compile(r"\[([^\r\n\]]+)\][ \f\t]*")
Setting = re.compile(r"(\w+)\s*=\s*")

PaletteColours = '(?:#[0-9a-fA-F]{12}:){15}#[0-9a-fA-F]{12}'
SingleColour   = '#[0-9a-fA-F]{6}|#[0-9a-fA-F]{3}'

Colourvalue = re.compile(group(PaletteColours, SingleColour))
Barevalue = re.compile(r'((?:[^\r\n# \f\t]+|[^\r\n#]+(?!' + Ignore.pattern +'))+)')

Tabsize = 8

class ConfigSyntaxError(Exception):
  def __init__(self, message, cf):
    self.single_error = cf.errors_are_fatal
    self.message = message
    self.file = cf.filename
    self.lnum = cf._lnum
    self.pos  = cf._pos
    self.line = cf._line

  def __str__(self):
    if self.single_error:
      fmt = "File %(file)s line %(lnum)d:\n    %(line)s\n    %(pad)s^\n%(message)s"
    else:
      fmt = " * %(message)s, line %(lnum)d:\n    %(line)s\n    %(pad)s^\n"
    return fmt % {'message': self.message, 'file': self.file, 'lnum': self.lnum,
                  'line': self.line.rstrip(), 'pad': '-' * self.pos}

class ParsedWithErrors(Exception):
  def __init__(self, filename, errors):
    self.file = filename
    self.errors = errors

  def __str__(self):
    return """Errors were encountered while parsing configuration file:

  %s

Some lines have been ignored.

%s
""" % (repr(self.file), "\n".join(map(lambda error: str(error), self.errors)))


class ConfigFile:
  def __init__(self, filename = None, errors_are_fatal = False):
    self.errors_are_fatal = errors_are_fatal
    self.filename = filename
    self.settings = {}
    self.errors = []

  def _call_if_match(self, re, callable, group = 0):
    if self._pos == self._max:
      return False
    mo = re.match(self._line, self._pos)
    if mo:
      if callable:
        callable(mo.group(group))
      self._pos = mo.end()
      return True
    else:
      return False

  def _call_if_quoted_string(self, callable):
    if self._pos == self._max:
      return False
    chr = self._line[self._pos]
    if chr in '"\'':
      string = ''
      while True:
        mo = QuotedStrings[chr].match(self._line, self._pos)
        if mo is None:
          raise ConfigSyntaxError("Unterminated quoted string", self)
        self._pos = mo.end()
        if self._line[self._pos - 2] == '\\':
          string += mo.group(1)[0:-1] + chr
          self._pos -= 1
        else:
          string += mo.group(1)
          break
      callable(string)
      return True
    else:
      return False

  def parse_error(self, error):
    e = ConfigSyntaxError(error, self)
    if self.errors_are_fatal:
      raise e
    else:
      self.errors.append(e)

  def parse(self):
    file = open(self.filename)
    rc = file.readlines()
    file.close()

    self._indents = [0]
    self._pos = 0
    self._max = 0
    self._lnum = 0
    self._line = ''

    self._currsection = None
    self._cursetting = None
    self.errors = []

    for self._line in rc:
      self._lnum += 1
      self._pos = 0
      self._max = len(self._line)
      dbg("Line %d: %s" % (self._lnum, repr(self._line)))

      self._call_if_match(WhitespaceRE, None)

      # [Section]
      self._call_if_match(Section, self._section, 1)
      # setting =
      if self._call_if_match(Setting, self._setting, 1):
        # "quoted value"
        if not self._call_if_quoted_string(self._value):
          # #000000 # colour that would otherwise be a comment
          if not self._call_if_match(Colourvalue, self._value, 1):
            # bare value
            if not self._call_if_match(Barevalue, self._value, 1):
              self.parse_error("Setting without a value")
              continue

      self._call_if_match(Ignore, lambda junk: dbg("Ignoring: %s" % junk))

      if self._line[self._pos:] != '':
        self.parse_error("Unexpected token")

    if self.errors:
      raise ParsedWithErrors(self.filename, self.errors)

  def _section(self, section):
    dbg("Section %s" % repr(section))
    self._currsection = section.lower()

  def _setting(self, setting):
    dbg("Setting %s" % repr(setting))
    self._currsetting = setting.lower()

  def _value(self, value):
    dbg("Value %s" % repr(value))
    if self._currsection is not None:
      self.settings.setdefault(self._currsection, {})[self._currsetting] = value
    else:
      self.settings[self._currsetting] = value