Bagaimana saya bisa mendapatkan kode sumber fungsi Python?

Misalkan saya memiliki fungsi Python seperti yang didefinisikan di bawah ini:

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

Saya bisa mendapatkan nama fungsi menggunakan foo.func_name. Bagaimana saya bisa mendapatkan kode sumber secara terprogram, seperti yang saya ketikkan di atas?

Jika fungsinya dari file sumber yang tersedia di sistem file, maka inspect.getsource(foo) mungkin bisa membantu:

Jika foodidefinisikan sebagai:

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

Kemudian:

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

Pengembalian:

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

Tetapi saya percaya bahwa jika fungsi dikompilasi dari string, stream atau diimpor dari file yang dikompilasi, maka Anda tidak dapat mengambil kode sumbernya.

The memeriksa modul memiliki metode untuk mengambil kode sumber dari objek python. Tampaknya itu hanya berfungsi jika sumbernya berada dalam file sekalipun. Jika Anda punya itu saya kira Anda tidak perlu mendapatkan sumber dari objek.

dis adalah teman Anda jika kode sumber tidak tersedia:

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

Jika Anda menggunakan IPython, maka Anda perlu mengetik "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

Meskipun saya umumnya setuju bahwa itu inspectadalah jawaban yang baik, saya tidak setuju bahwa Anda tidak bisa mendapatkan kode sumber objek yang didefinisikan dalam interpreter. Jika Anda menggunakan dill.source.getsourcedari dill, Anda bisa mendapatkan sumber fungsi dan lambda, bahkan jika mereka didefinisikan secara interaktif. Ia juga bisa mendapatkan kode dari metode dan fungsi kelas yang diikat atau tidak terikat yang didefinisikan dalam kari ... namun, Anda mungkin tidak dapat mengkompilasi kode itu tanpa kode objek yang dilampirkan.

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

>>> 

Untuk memperluas jawaban 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: Seperti yang ditunjukkan oleh @ 0sh contoh ini berfungsi menggunakan ipythontetapi tidak sederhana python. Namun, keduanya harus baik ketika mengimpor kode dari file sumber.

Anda dapat menggunakan inspectmodul untuk mendapatkan kode sumber lengkap untuk itu. Anda harus menggunakan getsource()metode untuk itu dari inspectmodul. Sebagai contoh:

import inspect

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

print(inspect.getsource(get_my_code))

Anda dapat memeriksanya lebih banyak opsi pada tautan di bawah ini. ambil kode python Anda

Karena pos ini ditandai sebagai duplikat dari pos lain ini , saya jawab di sini untuk kasus "lambda", meskipun OP bukan tentang lambda.

Jadi, untuk fungsi lambda yang tidak didefinisikan dalam baris mereka sendiri: selain jawaban marko.ristin , Anda mungkin ingin menggunakan mini-lambda atau menggunakan SymPy seperti yang disarankan dalam jawaban ini .

• mini-lambda lebih ringan dan mendukung segala jenis operasi, tetapi hanya berfungsi untuk satu variabel
• SymPylebih berat tetapi jauh lebih dilengkapi dengan operasi matematika / kalkulus. Secara khusus itu dapat menyederhanakan ekspresi Anda. Ini juga mendukung beberapa variabel dalam ekspresi yang sama.

Berikut ini cara melakukannya 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))

Ini menghasilkan dengan benar

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

x ** 2

Lihat mini-lambda dokumentasi untuk detailnya. Aku pengarangnya;)

Harap diingat bahwa jawaban yang diterima hanya berfungsi jika lambda diberikan pada baris terpisah. Jika Anda meneruskannya sebagai argumen ke fungsi dan ingin mengambil kode lambda sebagai objek, masalahnya menjadi sedikit rumit karena inspectakan memberi Anda seluruh baris.

Misalnya, pertimbangkan sebuah file test.py:

import inspect

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

if __name__ == "__main__":
    main()

Melaksanakannya memberi Anda (ingatlah indensi!):

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

Untuk mengambil kode sumber lambda, taruhan terbaik Anda, menurut saya, adalah mengurai kembali seluruh file sumber (dengan menggunakan f.__code__.co_filename) dan mencocokkan node lambda AST dengan nomor baris dan konteksnya.

Kami harus melakukan tepat bahwa di perpustakaan kami desain-by-kontrak icontract karena kami harus mengurai fungsi lambda kita lulus sebagai argumen untuk dekorator. Terlalu banyak kode untuk disisipkan di sini, jadi lihat implementasi dari fungsi ini .

Jika Anda benar-benar mendefinisikan fungsi sendiri dan itu adalah definisi yang relatif singkat, solusi tanpa dependensi adalah mendefinisikan fungsi dalam sebuah string dan menetapkan eval () ekspresi untuk fungsi Anda.

Misalnya

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

kemudian secara opsional untuk melampirkan kode asli ke fungsi:

func.source = funcstring

untuk meringkas :

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

Saya percaya bahwa nama variabel tidak disimpan dalam file pyc / pyd / pyo, jadi Anda tidak dapat mengambil baris kode yang tepat jika Anda tidak memiliki file sumber.