forked from igraph/python-igraph
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathpatch_modularized_graph_methods.py
More file actions
90 lines (68 loc) · 2.78 KB
/
Copy pathpatch_modularized_graph_methods.py
File metadata and controls
90 lines (68 loc) · 2.78 KB
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import os
import sys
import inspect
import igraph
# FIXME: there must be a better way to do this
auxiliary_imports = [
('typing', '*'),
('igraph.io.files', '_identify_format'),
('igraph.community', '_optimal_cluster_count_from_merges_and_modularity'),
]
def main():
# Get instance and classmethods
g = igraph.Graph()
methods = inspect.getmembers(g, predicate=inspect.ismethod)
# Get the source code for each method and replace the method name
# in the signature
methodsources = {}
underscore_functions = set()
for mname, method in methods:
# Source code of the function that uses the method
source = inspect.getsourcelines(method)[0]
# Find function name for this modularized method
fname = source[0][source[0].find('def ') + 4: source[0].find('(')]
# FIXME: this also swaps in methods that are already there. While
# that should be fine, we could check
# Make new source code, which is the same but with the name swapped
newsource = [source[0].replace(fname, mname)] + source[1:]
methodsources[mname] = newsource
# Prepare to delete the import for underscore functions
if fname.startswith('_'):
underscore_functions.add(fname)
newmodule = igraph.__file__ + '.new'
with open(newmodule, 'wt') as fout:
# FIXME: whitelisting all cases is not great, try to improve
for (origin, value) in auxiliary_imports:
fout.write(f'from {origin} import {value}\n')
with open(igraph.__file__, 'rt') as f:
# Swap in the method sources
for line in f:
mtype = None
for mname in methodsources:
# Class methods (constructors)
if ' ' + mname + ' = classmethod(_' in line:
mtype = 'class'
break
# Instance methods (excluding factories e.g. 3d layouts)
if (' ' + mname + ' = _' in line) and ('(' not in line):
mtype = 'instance'
break
else:
fout.write(line)
continue
# Method found, substitute and remove from dict
fout.write('\n')
if mtype == 'class':
fout.write(' @classmethod\n')
for mline in methodsources[mname]:
# Correct indentation
fout.write(' ' + mline)
del methodsources[mname]
# Move the new file back
with open(igraph.__file__, 'wt') as fout:
with open(newmodule, 'rt') as f:
fout.write(f.read())
# Delete .new file
os.remove(newmodule)
if __name__ == '__main__':
main()