Merge pull request #59 from maralla/set

support set
Thriftpy · Nov 1, 2014 · bb11d1d · bb11d1d
2 parents 574b911 + f598126
commit bb11d1d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/tests/test_type.py b/tests/test_type.py
@@ -0,0 +1,8 @@
+from thriftpy import load
+from thriftpy.thrift import TType
+
+
+def test_set():
+    s = load("type.thrift")
+
+    assert s.Set.thrift_spec == {1: (TType.SET, "a_set", TType.STRING)}
diff --git a/tests/type.thrift b/tests/type.thrift
@@ -0,0 +1,3 @@
+struct Set {
+    1: required set<string> a_set
+}
diff --git a/thriftpy/parser/__init__.py b/thriftpy/parser/__init__.py
@@ -103,6 +103,8 @@ def _ttype(t, module=None):
                 return getattr(TType, t.upper())
         elif t[0] == "list":
             return TType.LIST, _ttype(t[1])
+        elif t[0] == "set":
+            return TType.SET, _ttype(t[1])
         elif t[0] == "map":
             return TType.MAP, (_ttype(t[1]), _ttype(t[2]))
         else: