Update code generator to parse CSV files

telethon_generator/parsers/methods.py

@@ -0,0 +1,29 @@
+import csv
+
+
+class MethodInfo:
+    def __init__(self, name, usability, errors):
+        self.name = name
+        self.usability = usability
+        self.errors = errors
+
+
+def parse_methods(csv_file, errors_dict):
+    """
+    Parses the input CSV file with columns (method, usability, errors)
+    and yields `MethodInfo` instances as a result.
+    """
+    with open(csv_file, newline='') as f:
+        f = csv.reader(f)
+        next(f, None)  # header
+        for line, (method, usability, errors) in enumerate(f, start=2):
+            if usability not in ('user', 'bot', 'both', 'unknown'):
+                raise ValueError('Usability must be either user, bot, '
+                                 'both or unknown, not {}'.format(usability))
+            try:
+                errors = [errors_dict[x] for x in errors.split()]
+            except KeyError:
+                raise ValueError('Method {} references unknown errors {}'
+                                 .format(method, errors)) from None
+
+            yield MethodInfo(method, usability, errors)