views:

321

answers:

3

Is it possible to break the execution of a Python script called with the execfile function without using an if/else statement? I've tried exit(), but it doesn't allow main.py to finish.

# main.py
print "Main starting"
execfile("script.py")
print "This should print"

# script.py
print "Script starting"
a = False

if a == False:
    # Sanity checks. Script should break here
    # <insert magic command>    

# I'd prefer not to put an "else" here and have to indent the rest of the code
print "this should not print"
# lots of lines bellow
+2  A: 
# script.py
def main():
    print "Script starting"
    a = False

    if a == False:
        # Sanity checks. Script should break here
        # <insert magic command>    
        return;
        # I'd prefer not to put an "else" here and have to indent the rest of the code
    print "this should not print"
    # lots of lines bellow

if __name__ ==  "__main__":
    main();

I find this aspect of Python (the __name__ == "__main__", etc.) irritating.

Matthew Flaschen
There are two small errors in the code you posted: "print "this should not print"" is indented one too much and return does not need a semicolon. Thanks, I think I'll use this.
JcMaco
But then the lines are indented, which is what the OP wants to avoid. I don't understand why that indenting is to be avoided, though.
Vinay Sajip
I fixed the print. I'm keeping the semi-colon though. ;) Ah, how I hate significant whitespace. :)
Matthew Flaschen
+3  A: 

main can wrap the execfile into a try/except block: sys.exit raises a SystemExit exception which main can catch in the except clause in order to continue its execution normally, if desired. I.e., in main.py:

try:
  execfile('whatever.py')
except SystemExit:
  print "sys.exit was called but I'm proceeding anyway (so there!-)."
print "so I'll print this, etc, etc"

and whatever.py can use sys.exit(0) or whatever to terminate its own execution only. Any other exception will work as well as long as it's agreed between the source to be execfiled and the source doing the execfile call -- but SystemExit is particularly suitable as its meaning is pretty clear!

Alex Martelli
That seems like a hack to me. Though Python is an exception-friendly language...
Matthew Flaschen
Exiting early from linear code requires one of return, continue, break, or raise (sys.exit() being a special case of raise). raise is the *least* hackish in this context, IMO.
Rick Copeland
@Matthew, I agree with @Rick: in Python, exceptions aren't that exceptional (there's one, StopIteration, at the normal end of _every_ for loop, for example!), so using one to terminate the exec'd-file execution seems acceptable to me.
Alex Martelli
+1  A: 

What's wrong with plain old exception handling?

scriptexit.py

class ScriptExit( Exception ): pass

main.py

from scriptexit import ScriptExit
print "Main Starting"
try:
    execfile( "script.py" )
except ScriptExit:
    pass
print "This should print"

script.py

from scriptexit import ScriptExit
print "Script starting"
a = False

if a == False:
    # Sanity checks. Script should break here
    raise ScriptExit( "A Good Reason" )

# I'd prefer not to put an "else" here and have to indent the rest of the code
print "this should not print"
# lots of lines below
S.Lott