Adiciona verificação da variável de ambiente 'TRANSLATE_DATA' para a leitura dos arquivos

Erickson Silva
1 parent 49730dbc
Showing 3 changed files with 23 additions and 15 deletions Show diff stats
src/AplicaRegras.py
src/LerDicionarios.py
src/alexp.py
@@ -41,6 +41,8 @@ class AplicaRegras(object):
       	'''
 		if platform.system() == 'Windows':
 			return ET.parse(environ.get("HOMEDRIVE")+'\\vlibras-libs\\vlibras-translate\data\\regras.xml').getroot()
+		elif "TRANSLATE_DATA" in environ:
+			return ET.parse('regras.xml').getroot()
 		return ET.parse(expanduser("~")+'/vlibras-translate/data/regras.xml').getroot()
 	# Aplica regras morfológicas apartir do arquivo regras.xml
@@ -45,7 +45,9 @@ class LerDicionarios(Singleton):
       '''
       if platform.system() == 'Windows':
          return environ.get("HOMEDRIVE") + "\\vlibras-libs\\vlibras-translate\data\\"
-      return expanduser("~") + "/vlibras-translate/data/"
+      elif "TRANSLATE_DATA" in environ:
+         return environ("TRANSLATE_DATA")
+      return expanduser("~") + "/vlibras-translate/data"
    def carregar_dicionarios(self):
       '''Realiza a leitura dos arquivos e atribui à estruturas de dicionários e sets.
@@ -62,11 +64,14 @@ class LerDicionarios(Singleton):
       self.carregar_verbos_ligacao()     
       self.carregar_verbos_muda_negacao  
+   def montar_diretorio(self, arquivo):
+      return os.path.join(self.path, arquivo)
+
    def carregar_excecoes_plural(self):
       '''Carrega arquivo de exceções de plural.
       '''
       try:
-         self.file = csv.reader(open(self.path+"excecoesPlural.csv"))
+         self.file = csv.reader(open(montar_diretorio("excecoesPlural.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_excecoes_plural"
@@ -80,7 +85,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de adverbios de intensidade.
       '''
       try:
-         self.file = csv.reader(open(self.path+"adverbiosIntensidade.csv"), delimiter=";")
+         self.file = csv.reader(open(montar_diretorio("adverbiosIntensidade.csv"), delimiter=";"))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_adverbios_intensidade"
@@ -93,7 +98,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de advérbios de tempo.
       '''
       try:
-         self.file = csv.reader(open(self.path+"adverbiosTempo.csv"))
+         self.file = csv.reader(open(montar_diretorio("adverbiosTempo.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_adverbios_tempo"
@@ -107,7 +112,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de artigos a serem removidos.
       '''
       try:
-         self.file = csv.reader(open(self.path+"artigos.csv"))
+         self.file = csv.reader(open(montar_diretorio("artigos.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_artigos"
@@ -121,7 +126,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de preposições a serem removidas.
       '''
       try:
-         self.file = csv.reader(open(self.path+"preposicoes.csv"))
+         self.file = csv.reader(open(montar_diretorio("preposicoes.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_preposicoes"
@@ -135,7 +140,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de sinônimos.
       '''
       try:
-         self.file = csv.reader(open(self.path+"sinonimos.csv"), delimiter=";")
+         self.file = csv.reader(open(montar_diretorio("sinonimos.csv"), delimiter=";"))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_sinonimos"
@@ -151,7 +156,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo dos substantivos comuns de 2 generos.
       '''
       try:
-         self.file = csv.reader(open(self.path+"subs2Generos.csv"))
+         self.file = csv.reader(open(montar_diretorio("subs2Generos.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_subs_2_generos" 
@@ -165,7 +170,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de verbos no infinitivo.
       '''
       try:
-         self.file = csv.reader(open(self.path+"verbosInfinitivo.csv"), delimiter=";")
+         self.file = csv.reader(open(montar_diretorio("verbosInfinitivo.csv"), delimiter=";"))
       except IOError, (errno, strerror): 
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_verbos_infinitivo"
@@ -181,7 +186,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de verbos de ligação.
       '''
       try:
-         self.file = csv.reader(open(self.path+"verbosLigacao.csv"))
+         self.file = csv.reader(open(montar_diretorio("verbosLigacao.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_verbos_ligacao"
@@ -195,7 +200,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de pronomes de tratamento.
       '''
       try:
-         self.file = csv.reader(open(self.path+"pronomesTratamento.csv"))
+         self.file = csv.reader(open(montar_diretorio("pronomesTratamento.csv")))
       except IOError, (errno, strerror):
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_pronomes_tratamento"
@@ -209,7 +214,7 @@ class LerDicionarios(Singleton):
       '''Carrega arquivo de verbos que mudam a negação.
       '''
       try:
-         self.file = csv.reader(open(self.path+"verbosMudaNegacao.csv"), delimiter=";")
+         self.file = csv.reader(open(montar_diretorio("verbosMudaNegacao.csv"), delimiter=";"))
       except IOError, (errno, strerror): 
          print "I/O error(%s): %s" % (errno, strerror)
          print "carregar_verbos_muda_negacao"
@@ -29,7 +29,7 @@
 """
 import re,nltk,platform, time, random
 from os.path import expanduser
-from os import environ
+from os import environ, path
 from Aelius.Extras import carrega
 from Aelius import AnotaCorpus
 from unicodedata import normalize
@@ -88,8 +88,9 @@ def encontraArquivo():
 	so = platform.system()
 	if so == 'Windows':
 		return environ.get("HOMEDRIVE") + "\\vlibras-libs\\vlibras-translate\data\cfg.syn.nltk"
-	else:
-		return expanduser("~") + "/vlibras-translate/data/cfg.syn.nltk"
+	elif "TRANSLATE_DATA" in environ:
+		return os.path.join(environ("TRANSLATE_DATA"), "cfg.syn.nltk")
+	return expanduser("~") + "/vlibras-translate/data/cfg.syn.nltk"
 def extraiSintaxe():
 	"""Extrai gramática armazenada em arquivo cujo caminho é definido relativamente ao diretório nltk_data.