Jak mogę uzyskać kod źródłowy funkcji Python?

Załóżmy, że mam funkcję Python, jak zdefiniowano poniżej:

def foo(arg1,arg2):
    #do something with args
    a = arg1 + arg2
    return a

Mogę uzyskać nazwę funkcji za pomocą foo.func_name . Jak mogę programowo uzyskać kod źródłowy, jak napisałem powyżej?

Jeśli funkcja pochodzi z pliku źródłowego dostępnego w systemie plików, inspect.getsource(foo)może być pomocna:

Jeśli foojest zdefiniowany jako:

def foo(arg1,arg2):         
    #do something with args 
    a = arg1 + arg2         
    return a  

Następnie:

import inspect
lines = inspect.getsource(foo)
print(lines)

Zwroty:

def foo(arg1,arg2):         
    #do something with args 
    a = arg1 + arg2         
    return a                

Uważam jednak, że jeśli funkcja jest skompilowana z ciągu, strumienia lub zaimportowana ze skompilowanego pliku, nie można pobrać jej kodu źródłowego.

Sprawdzić moduł ma metody pobierania kodu źródłowego z obiektów Pythona. Pozornie działa to tylko wtedy, gdy źródło znajduje się w pliku. Gdybyś to miał, myślę, że nie musiałbyś pobierać źródła z obiektu.

dis jest twoim przyjacielem, jeśli kod źródłowy nie jest dostępny:

>>> import dis
>>> def foo(arg1,arg2):
...     #do something with args
...     a = arg1 + arg2
...     return a
...
>>> dis.dis(foo)
  3           0 LOAD_FAST                0 (arg1)
              3 LOAD_FAST                1 (arg2)
              6 BINARY_ADD
              7 STORE_FAST               2 (a)

  4          10 LOAD_FAST                2 (a)
             13 RETURN_VALUE

Jeśli używasz IPython, musisz wpisać „foo ??”

In [19]: foo??
Signature: foo(arg1, arg2)
Source:
def foo(arg1,arg2):
    #do something with args
    a = arg1 + arg2
    return a

File:      ~/Desktop/<ipython-input-18-3174e3126506>
Type:      function

Chociaż ogólnie zgadzam się, że inspectto dobra odpowiedź, nie zgadzam się z tym, że nie można uzyskać kodu źródłowego obiektów zdefiniowanych w tłumaczu. Jeśli używasz dill.source.getsourcez dill, możesz uzyskać źródło funkcji i lambd, nawet jeśli są one zdefiniowane interaktywnie. Może również uzyskać kod z metod klasy i funkcji powiązanych lub niezwiązanych zdefiniowanych w curry ... jednak kompilacja tego kodu może nie być możliwa bez kodu obiektu otaczającego.

>>> from dill.source import getsource
>>> 
>>> def add(x,y):
...   return x+y
... 
>>> squared = lambda x:x**2
>>> 
>>> print getsource(add)
def add(x,y):
  return x+y

>>> print getsource(squared)
squared = lambda x:x**2

>>> 
>>> class Foo(object):
...   def bar(self, x):
...     return x*x+x
... 
>>> f = Foo()
>>> 
>>> print getsource(f.bar)
def bar(self, x):
    return x*x+x

>>> 

Aby rozwinąć odpowiedź Runeha:

>>> def foo(a):
...    x = 2
...    return x + a

>>> import inspect

>>> inspect.getsource(foo)
u'def foo(a):\n    x = 2\n    return x + a\n'

print inspect.getsource(foo)
def foo(a):
   x = 2
   return x + a

EDYCJA: Jak wskazano przez @ 0sh, ten przykład działa przy użyciu, ipythonale nie zwykły python. Jednak w przypadku importowania kodu z plików źródłowych powinno być dobrze w obu przypadkach.

Możesz użyć inspectmodułu, aby uzyskać pełny kod źródłowy. Musisz użyć getsource()do tego metody z inspectmodułu. Na przykład:

import inspect

def get_my_code():
    x = "abcd"
    return x

print(inspect.getsource(get_my_code))

Możesz sprawdzić więcej opcji pod poniższym linkiem. pobierz swój kod python

Ponieważ ten post jest oznaczony jako duplikat tego drugiego postu , odpowiadam tutaj w sprawie „lambda”, chociaż OP nie dotyczy lambdas.

Tak więc, dla funkcji lambda, które nie są zdefiniowane we własnych liniach: oprócz odpowiedzi marko.ristin , możesz chcieć użyć mini-lambda lub użyć SymPy, jak sugerowano w tej odpowiedzi .

• mini-lambda jest lżejszy i obsługuje wszelkie operacje, ale działa tylko dla jednej zmiennej
• SymPyjest cięższy, ale o wiele bardziej wyposażony w operacje matematyczne / rachunku różniczkowego. W szczególności może uprościć twoje wyrażenia. Obsługuje również kilka zmiennych w tym samym wyrażeniu.

Oto jak możesz to zrobić za pomocą mini-lambda:

from mini_lambda import x, is_mini_lambda_expr
import inspect

def get_source_code_str(f):
    if is_mini_lambda_expr(f):
        return f.to_string()
    else:
        return inspect.getsource(f)

# test it

def foo(arg1, arg2):
    # do something with args
    a = arg1 + arg2
    return a

print(get_source_code_str(foo))
print(get_source_code_str(x ** 2))

Prawidłowo daje

def foo(arg1, arg2):
    # do something with args
    a = arg1 + arg2
    return a

x ** 2

Szczegółowe informacje można znaleźć w mini-lambda dokumentacji . Nawiasem mówiąc, jestem autorem;)

Pamiętaj, że zaakceptowane odpowiedzi działają tylko wtedy, gdy lambda jest podana w osobnym wierszu. Jeśli przekażesz go jako argument funkcji i chcesz pobrać kod lambda jako obiekt, problem stanie się nieco trudny, ponieważ inspectda ci całą linię.

Rozważmy na przykład plik test.py:

import inspect

def main():
    x, f = 3, lambda a: a + 1
    print(inspect.getsource(f))

if __name__ == "__main__":
    main()

Wykonanie go daje (uwaga na wcięcie!):

    x, f = 3, lambda a: a + 1

Aby pobrać kod źródłowy lambda, moim zdaniem najlepiej jest ponownie przeanalizować cały plik źródłowy (przy użyciu f.__code__.co_filename) i dopasować węzeł AST lambda według numeru linii i jego kontekstu.

Musieliśmy zrobić dokładnie, że w naszym design-by-umowy biblioteki icontract ponieważ mieliśmy do analizowania funkcje lambda Mijamy się jako argumenty dekoratorów. Jest tu za dużo kodu do wklejenia, więc spójrz na implementację tej funkcji .

Jeśli ściśle definiujesz funkcję samodzielnie i jest to stosunkowo krótka definicja, rozwiązaniem bez zależności byłoby zdefiniowanie funkcji w ciągu i przypisanie funkcji eval () wyrażeniu do funkcji.

Na przykład

funcstring = 'lambda x: x> 5'
func = eval(funcstring)

następnie opcjonalnie, aby dołączyć oryginalny kod do funkcji:

func.source = funcstring

podsumowując:

import inspect
print( "".join(inspect.getsourcelines(foo)[0]))

Uważam , że nazwy zmiennych nie są przechowywane w plikach pyc / pyd / pyo, więc nie możesz pobrać dokładnych wierszy kodu, jeśli nie masz plików źródłowych.