PyNet/inspectNodes.py at main · leecafe/PyNet · GitHub

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
import inspect
import open3d

modules = []
mnetjson = {'nodes': '', 'links': ''}
nodes = []
links = []


def get_modules(arg, str):
    print(arg.__name__, str)
    if arg.__name__ == filename:
        modules.append(arg.__name__)
        nodes.append({'name': arg.__name__, 'file': arg.__file__})
    mmembers = inspect.getmembers(arg, inspect.ismodule)
    # print(modules)
    # print(nodes)
    # print(mmembers[0:3])
    # print('--------------------------')
    for (name, moduleurl) in mmembers:
        print("name=" + name)
        print(moduleurl.__name__)
        print("===============")
        if str in moduleurl.__name__ and not (moduleurl.__name__ in modules):  # ÅÐ¶ÏÃ»ÓÐ±»É¨Ãèµ½µÄModule
            modules.append(moduleurl.__name__)
            # print(modules)
            __import__(moduleurl.__name__)
            # print(moduleurl.__name__)
            # print('__file__' in dir(moduleurl))
            if '__file__' in dir(moduleurl):
                nodes.append({'name': moduleurl.__name__, 'file': eval(moduleurl.__name__).__file__})
                # print(nodes)
                # print("------------3--------------")
                # print(moduleurl.__name__, filename)
                get_modules(eval(moduleurl.__name__), filename)
            else:
                # print("----------4-------------")
                if len(inspect.getmembers(arg, inspect.ismodule)) > 0:
                    nodes.append({'name': moduleurl.__name__, 'file': 'null'})
                    get_modules(eval(moduleurl.__name__), filename)

    return modules, nodes


filename = 'open3d'
get_modules(eval(filename), filename)