Elenca tutti i pacchetti di terze parti e le loro stesse funzioni utilizzate in un file Python

Question

Ho molti pacchetti python, scritti dai miei colleghi e voglio scrivere uno strumento per verificare su quali terzi si affidano.

Ti piace questa

#it is my package, need to check,call it example.py 
#We have more than one way to import a package, It is a problem need to consider too 

from third_party_packages import third_party_function 

def my_function(arg): 
    return third_party_function(arg) 

e lo strumento dovrebbe funzionare come questo

result = tool(example.py) 
#this result should be a dict like this structure 
#{"third_party_function":["my_function",]} 
#Means "my_function" relies on "third_party_function" 

non ho idea di come fare, tutto quello che può venire l'attuazione di questo strumento è leggere un file Python linea per riga come stringa, e usa regex per confrontarla. Potresti darmi qualche consiglio?

Se non si sa cosa intendo, si prega di commentare la tua domanda, lo aggiusterò il prima possibile. Grazie!

Answer 1

È possibile analizzare i file con il modulo ast e controllare tutte le dichiarazioni Import e ImportFrom.

Per darvi un'idea, ecco un esempio:

>>> import ast 
>>> tree = ast.parse('import a; from b import c') 
>>> tree.body 
[<_ast.Import object at 0x7f3041263860>, <_ast.ImportFrom object at 0x7f3041262c18>] 
>>> tree.body[0].names[0].name 
'a' 
>>> tree.body[1].module 
'b' 
>>> tree.body[1].names[0].name 
'c' 

Lo script potrebbe funzionare in questo modo:

1. analizzare il file sorgente tramite ast.parse
2. visita ogni nodo utilizzando ast.walk()
3. Se un nodo è un oggetto Import o ImportFrom, quindi ispezionare i nomi e fare ciò che si ha fare.

L'utilizzo di ast è molto più semplice e robusto delle espressioni regolari o di un parser personalizzato.