Python exec e __name__

Question

quando ho eseguito:

exec "print __name__" 

la stampa __main__.

Ma quando ho eseguito:

exec "print __name__" in {} 

la stampa __builtin__.

Come eseguire il secondo esempio per stampare anche __main__?

Quello che cerco di ottenere è di eseguire un pezzo di codice con exec in modo che dal punto di vista di esso sembra essere stato eseguito dalla riga di comando.

Vorrei sintonizzare il codice con ambito pulito ma il secondo esempio interrompe il codice basandosi su if __name__ == "__main__". Come risolvere questo?

Answer 1

Si potrebbe utilizzare imp.load_module invece:

import imp 

with open(mainfile) as src: 
    imp.load_module('__main__', src, mainfile, (".py", "r", imp.PY_SOURCE)) 

Questo importa il file del modulo __main__, eseguirlo.

Nota che è necessario un oggetto file effettivo quando il tipo è impostato su imp.PY_SOURCE, quindi è necessario creare un file temporaneo affinché funzioni correttamente se il codice sorgente proviene da un luogo diverso da un file.

In caso contrario, si può sempre impostare manualmente __name__:

>>> src = '''\ 
... if __name__ == '__main__': print 'Main!' 
... else: print 'Damn', __name__ 
... ''' 
>>> exec src 
Main! 
>>> exec src in {} 
Damn __builtin__ 
>>> exec src in {'__name__':'__main__'} 
Main! 

Answer 2

Una soluzione è quella di fornire il __name__ esplicito nel dict exec ution:

exec "print __name__" in {'__name__': '__main__'}