blob: c4d515b68fd2bd72fa6e5de8ec99318cf3120489 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
from __future__ import absolute_import
from contextlib import contextmanager
__all__ = [
'reversed',
]
@contextmanager
def reversed(G):
"""A context manager for temporarily reversing a directed graph in place.
This is a no-op for undirected graphs.
Parameters
----------
G : graph
A NetworkX graph.
"""
directed = G.is_directed()
if directed:
G.reverse(copy=False)
try:
yield
finally:
if directed:
# Reverse the reverse.
G.reverse(copy=False)
|
