Changed ModeDict to a Modes enum-like class

Signed-off-by: Juanjo Alvarez <juanjo@sourced.tech>

Author: Juanjo Alvarez

bblfsh/aliases.py

@@ -21,10 +21,13 @@
 Mode = protocol_v2_module.Mode
 ModeType = google.protobuf.internal.enum_type_wrapper.EnumTypeWrapper
 
+
+class Modes:
+    pass
+
 # Current values: {'DEFAULT_MODE': 0, 'NATIVE': 1, 'PREPROCESSED': 2, 'ANNOTATED': 4, 'SEMANTIC': 8}
-ModeDict = {}
 for k, v in Mode.DESCRIPTOR.values_by_name.items():
-    ModeDict[k] = v.number
+    setattr(Modes, k, v.number)
 
 DriverStub = protocol_grpc_v2_module.DriverStub
 DriverServicer = protocol_grpc_v2_module.DriverServicer

bblfsh/test.py

@@ -5,7 +5,7 @@
 import docker
 
 from bblfsh import (BblfshClient, iterator, TreeOrder,
-                    ModeDict, role_id, role_name)
+                    Modes, role_id, role_name)
 from bblfsh.launcher import ensure_bblfsh_is_running
 from bblfsh.client import NonUTF8ContentException
 from bblfsh.result_context import (Node, NodeIterator,
@@ -45,7 +45,7 @@ def testVersion(self) -> None:
         self.assertTrue(version.build)
 
     def testNativeParse(self) -> None:
-        ctx = self.client.parse(self.fixtures_file, mode=ModeDict["NATIVE"])
+        ctx = self.client.parse(self.fixtures_file, mode=Modes.NATIVE)
         self._validate_ctx(ctx)
         self.assertIsNotNone(ctx)