Як я можу отримати вихідний код функції Python?

Припустимо, у мене функція Python, як визначено нижче:

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

Я можу отримати назву функції за допомогою foo.func_name. Як я можу програмно отримати його вихідний код, як я вводив вище?

Якщо функція є з вихідного файлу, доступного у файловій системі, то, inspect.getsource(foo)можливо, допоможе:

Якщо fooвизначено як:

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

Тоді:

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

Повернення:

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

Але я вважаю, що якщо функція компілюється з рядка, потоку або імпортується із зібраного файлу, ви не можете отримати її вихідний код.

Модуль перевірки має методи для отримання вихідного коду з об'єктів python. Здається, він працює лише в тому випадку, якщо джерело знаходиться у файлі. Якби у вас це було, я думаю, вам не потрібно було б отримувати джерело від об'єкта.

dis є вашим другом, якщо вихідний код недоступний:

>>> 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

Якщо ви використовуєте IPython, вам потрібно набрати "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

Хоча я, як правило, погоджуюся, що inspectце хороша відповідь, я не погоджуюся, що ви не можете отримати вихідний код об'єктів, визначений в інтерпретаторі. Якщо ви користуєтесь dill.source.getsourceз dill, ви можете отримати джерело функцій та лямбда, навіть якщо вони визначені інтерактивно. Він також може отримати код для зв'язаних чи незв'язаних методів та функцій класу, визначених у curries ... однак, ви, можливо, не зможете скласти цей код без коду об'єкта, що додається.

>>> 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

>>> 

Щоб розгорнути відповідь на runeh:

>>> 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

EDIT: Як зазначає @ 0sh, цей приклад працює з використанням, ipythonале не з простих python. Однак це повинно бути добре в обох випадках, коли імпортується код з вихідних файлів.

Ви можете використовувати inspectмодуль, щоб отримати повний вихідний код для цього. Ви повинні використовувати getsource()метод для цього з inspectмодуля. Наприклад:

import inspect

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

print(inspect.getsource(get_my_code))

Ви можете ознайомитись з іншими параметрами за посиланням нижче. отримати свій код python

Оскільки ця посада позначена як дублікат цієї іншої публікації , я відповідаю тут за випадок "лямбда", хоча ОП не стосується лямбдів.

Отже, для функцій лямбда, які не визначені у власних рядках: крім відповіді marko.ristin , ви можете скористатись міні-лямбда або використовувати SymPy, як пропонується у цій відповіді .

• mini-lambda легше і підтримує будь-яку операцію, але працює лише для однієї змінної
• SymPyважче, але набагато більше оснащене операціями з математики / обчислення. Зокрема, це може спростити ваші вирази. Він також підтримує кілька змінних в одному виразі.

Ось як це можна зробити за допомогою 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))

Це правильно дає урожай

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

x ** 2

Детальну інформацію див. У mini-lambda документації . Я, до речі, автор;)

Зауважте, що прийняті відповіді працюють лише в тому випадку, якщо лямбда надається в окремому рядку. Якщо ви передаєте це як аргумент функції і хочете отримати код лямбда як об'єкт, проблема стає трохи хитрою, оскількиinspect дасть вам весь рядок.

Наприклад, розгляньте файл test.py:

import inspect

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

if __name__ == "__main__":
    main()

Виконання цього дає вам (пам’ятайте про відступи!):

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

Щоб отримати вихідний код лямбда, найкраще, на мою думку, - переаналізувати весь вихідний файл (використовуючи f.__code__.co_filename ) та зіставити вузол AST лямбда за номером рядка та його контекстом.

Ми повинні були зробити це саме в нашій іконтрактній програмі бібліотеки за контрактом, оскільки нам довелося проаналізувати функції лямбда, які ми передаємо як аргументи декораторам. Тут занадто багато коду для вставки, тому подивіться на реалізацію цієї функції .

Якщо ви чітко визначаєте функцію самостійно, і це відносно коротке визначення, рішенням без залежностей було б визначити функцію в рядку та призначити вираз eval () для вашої функції.

Напр

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

потім необов'язково приєднати до функції вихідний код:

func.source = funcstring

узагальнити :

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

Я вважаю, що імена змінних не зберігаються у файлах pyc / pyd / pyo, тому ви не можете отримати точні рядки коду, якщо у вас немає вихідних файлів.