Merge pull request #59 from maralla/set

support set

Author: Chao Wang

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)}

tests/type.thrift

@@ -0,0 +1,3 @@
+struct Set {
+    1: required set<string> a_set
+}

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: