|
5 | 5 |
|
6 | 6 | def tree(cls, level=0, last_sibling=True): |
7 | 7 | yield cls, level, last_sibling |
8 | | - subclasses = [c for c in cls.__subclasses__() |
9 | | - if c.__module__ == 'httpx' or c is RuntimeError] |
| 8 | + # get RuntimeError and exceptions defined in httpx |
| 9 | + subclasses = [sub for sub in cls.__subclasses__() |
| 10 | + if sub is RuntimeError or sub.__module__ == 'httpx'] |
10 | 11 | if subclasses: |
11 | 12 | last = subclasses[-1] |
12 | | - for sub_cls in subclasses: |
13 | | - yield from tree(sub_cls, level+1, sub_cls is last) |
| 13 | + for sub in subclasses: |
| 14 | + yield from tree(sub, level+1, sub is last) |
14 | 15 |
|
15 | 16 |
|
16 | 17 | def display(cls): |
|
0 commit comments