From 1875652231ad1ddde215f368a0abd9c6eb08b904 Mon Sep 17 00:00:00 2001
From: Reinier Heeres <rwh@rwh.(none)>
Date: Mon, 20 Oct 2008 08:04:10 +0000
Subject: Find *.txt files in lang/ instead of hardcoding them

---
(limited to 'LanguageModel.py')

diff --git a/LanguageModel.py b/LanguageModel.py
index d3521e0..120bdb9 100644
--- a/LanguageModel.py
+++ b/LanguageModel.py
@@ -16,8 +16,20 @@
 
 from __future__ import with_statement
 import time
+import os
+
+def GetSupportedLanguages():
+    ret = []
+    files = os.listdir("lang/")
+    for name in files:
+        fn, ext = os.path.splitext(name)
+        if ext == ".txt":
+            ret.append(fn)
+    ret.sort()
+    return ret
 
 class LanguageModel():
+
     def SetLanguages(self, lang1, lang2):
         """Take a language pair, prepare the language model."""
         self.lang1_lang2 = {}
--
cgit v0.9.1