Получение списка локально определенных функций в Python

Question

Если у меня есть сценарий вроде этого:

import sys 

def square(x): 
    return x*x 

def cube(x): 
    return x**3 

Как я могу вернуть список всех функций, определенных локально в программе ['square', 'cube'], а не те, импортирован ,

Они включены, когда я пытаюсь использовать dir(), но все переменные и другие импортированные модули. Я не знаю, что положить в dir, чтобы обратиться к локально исполняемому файлу.

Answer 1

l = [] 
for key, value in locals().items(): 
    if callable(value) and value.__module__ == __name__: 
     l.append(key) 
print l 

Так файл с содержанием:

from os.path import join 

def square(x): 
    return x*x 

def cube(x): 
    return x**3 

l = [] 
for key, value in locals().items(): 
    if callable(value) and value.__module__ == __name__: 
     l.append(key) 
print l 

Печать:

['square', 'cube'] 

Местные прицелы также работают:

def square(x): 
    return x*x 

def encapsulated(): 
    from os.path import join 

    def cube(x): 
     return x**3 

    l = [] 
    for key, value in locals().items(): 
     if callable(value) and value.__module__ == __name__: 
      l.append(key) 
    print l 

encapsulated() 

Печатает только:

['cube'] 

Answer 2

Применение inspect Модуль:

def is_function_local(object): 
    return isinstance(object, types.FunctionType) and object.__module__ == __name__ 

import sys 
print inspect.getmembers(sys.modules[__name__], predicate=is_function_local) 

Пример:

import inspect 
import types 
from os.path import join 

def square(x): 
    return x*x 

def cube(x): 
    return x**3 

def is_local(object): 
    return isinstance(object, types.FunctionType) and object.__module__ == __name__ 

import sys 
print [name for name, value in inspect.getmembers(sys.modules[__name__], predicate=is_local)] 

печатает:

['cube', 'is_local', 'square'] 

См: нет join функция импортируется из os.path.

is_local здесь, так как это функция - текущий модуль. Вы можете перенести его на другой модуль или исключить его вручную или определить вместо него lambda (как предлагал @BartoszKP).

Answer 3

import sys 
import inspect 
from os.path import join 

def square(x): 
    return x*x 

def cube(x): 
    return x**3 

print inspect.getmembers(sys.modules[__name__], \ 
     predicate = lambda f: inspect.isfunction(f) and f.__module__ == __name__) 

Печать:

[('cube', <function cube at 0x027BAC70>), ('square', <function square at 0x0272BAB0>)]