1 files changed, 36 insertions, 0 deletions
diff --git a/wlp.py b/wlp.py
new file mode 100644
index 0000000..64d7361
--- /dev/null
+++ b/wlp.py
@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+import wlp_parser
+
+
+__current_file = None
+
+
+def setfilebyname(name):
+    global __current_file
+    __current_file = open(name, 'r')
+
+
+def setfilebyfd(fd):
+    global __current_file
+    __current_file = fd
+
+
+def mkdict():
+    dict = {}
+    if __current_file is None:
+        raise ValueError('current file has not been set.')
+
+    tree = wlp_parser.parser.parse(
+        wlp_parser.lexer.lex(__current_file.read()))
+
+    for subtree in tree:
+        current_key = subtree[0].getstr().strip('<>')
+        if current_key not in dict:
+            dict[current_key] = {}
+
+        for key, value in subtree[1]:
+            key = key.getstr()
+            value = value.getstr().strip("'\"")
+            dict[current_key][key] = value
+
+    return dict