ts.py

from tag import Tag
from token import Token

class TS:
   '''
   Classe para a tabela de simbolos representada por um dicionario: {'chave' : 'valor'}
   '''
   def __init__(self):
      '''
      Repare que as palavras reservadas sao todas cadastradas
      a principio com linha e coluna em zero
      '''
      self.ts = {}
      
      self.ts['program'] = Token(Tag.KW_PROGRAM, 'program', 0, 0)
      self.ts['if'] = Token(Tag.KW_IF, 'if', 0, 0)
      self.ts['else'] = Token(Tag.KW_ELSE, 'else', 0, 0,)
      self.ts['while'] = Token(Tag.KW_WHILE, 'while', 0, 0)
      self.ts['write'] = Token(Tag.KW_WRITE, 'write', 0, 0)
      self.ts['read'] = Token(Tag.KW_READ, 'read', 0, 0)
      self.ts['num'] = Token(Tag.KW_NUM, 'num', 0, 0)
      self.ts['char'] = Token(Tag.KW_CHAR, 'char', 0, 0)
      self.ts['not'] = Token(Tag.KW_NOT, 'not', 0, 0)
      self.ts['or'] = Token(Tag.KW_OR, 'or', 0, 0)
      self.ts['and'] = Token(Tag.KW_AND, 'and', 0, 0)
      

   def getToken(self, lexema):
      token = self.ts.get(lexema.lower())
      return token

   def addToken(self, lexema, token):
      self.ts[lexema] = token

   def removeToken(self, lexema):
      self.ts.pop(lexema)

   def printTS(self):
      for k, t in (self.ts.items()):
         print(k, ":", t.toString())