mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2025-06-18 02:56:39 +00:00
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
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)
|