Как вы комментируете пользовательские классы? ⇐ Python

[Anonymous]

Предположим, у меня есть класс Shard, и возвращаемое значение метода должно быть списком объектов Shard.

Код: Выделить всё

my_method(args) -> ??:

Pycharm выдает ошибку, когда я импортировал List из набора и указал List[Shard] (ошибка связана с тем, что он не может чтобы найти Шард).
Вот метод, о котором идет речь:

Код: Выделить всё

def _return_shards(self, sql: str = None) -> List[Shard]:
"""
Queries the database to return a sequence of all the database clusters
and shards, and then creates Shard objects with the information.

@type conn_info: dict
:param conn_info: Database information containing the host, port, user,
database, and password.
@type sql: str

:return: A list of Shard items
"""
logging.info('Creating shards.')
cursor = self._return_cursor(self.canvas_conn)

stock_query = '''
SELECT
database_server_id || 'dr' database_server_id
, CASE
WHEN database_server_id LIKE 'cluster1' AND name IS NULL THEN 'canvas'
WHEN database_server_id LIKE 'cluster4' AND name IS NULL THEN 'canvas'
WHEN name IS NULL THEN database_server_id
ELSE name
END AS database_name
,SUBSTRING(database_server_id FROM '\d+$')::INT cluster_id
,CASE
WHEN name IS NULL THEN SUBSTRING(database_server_id FROM '\d+$')::INT
ELSE SUBSTRING(name FROM '\d+$')::INT
END AS shard_id
FROM canvas.switchman_shards
WHERE database_server_id LIKE 'cluster%'
ORDER BY cluster_id, shard_id
'''
if sql:
cursor.execute(sql)
else:
cursor.execute(stock_query)
results = cursor.fetchall()
self._set_reporter_total_shards(count=len(results))
return [Shard(
*result,
job=self.job,
manifest=self.manifest,
reporter=self.reporter.create_slave(db=result[0], schema=result[1])
) for result in results]

Ошибка NameError: имя «Shard» не определено, хотя класс Shard определен в том же модуле.

Подробнее здесь: https://stackoverflow.com/questions/372 ... om-classes