Merge pull request #19 from joakim-hove/enum-from-name

Added functionality to get an enum from the name.

cwrap/basecenum.py

@@ -49,6 +49,14 @@ def from_param(cls, c_class_object):
             raise ValueError("c_class_object must be an BaseCEnum instance!")
         return c_class_object.value
 
+    @classmethod
+    def from_string(cls, name):
+        for enum in cls.enum_namespace[cls]:
+            if enum.name == name:
+                return enum
+
+        raise ValueError("No such enum:%s" % name)
+
     @classmethod
     def addEnum(cls, name, value):
         name = str(name)

tests/test_basecenum.py

@@ -90,3 +90,17 @@ class MyLonelyEnum(BaseCEnum):
         self.assertEqual(str(tri), 'THREE')
         self.assertEqual(tri.name, 'THREE')
         self.assertEqual(tri.value, 3)
+
+
+    def test_from_name(self):
+        class EnumName(BaseCEnum):
+            pass
+
+        EnumName.addEnum("ONE", 1)
+        EnumName.addEnum("TWO", 2)
+
+        with self.assertRaises(ValueError):
+            enum = EnumName.from_string("No-not-this")
+
+        one = EnumName.from_string("ONE")
+        self.assertEqual(one, EnumName.ONE)