0

moved nice_exit() back into util.py (Issue #647)

This commit is contained in:
Aaron Griffith
2012-03-15 04:03:46 -04:00
parent 3bbab24a5b
commit ab22724fc7
3 changed files with 31 additions and 31 deletions

View File

@@ -45,33 +45,6 @@ helptext = """
%prog [--rendermodes=...] [options] <World> <Output Dir> %prog [--rendermodes=...] [options] <World> <Output Dir>
%prog --config=<config file> [options]""" %prog --config=<config file> [options]"""
def is_bare_console():
"""Returns true if Overviewer is running in a bare console in
Windows, that is, if overviewer wasn't started in a cmd.exe
session.
"""
if platform.system() == 'Windows':
try:
import ctypes
GetConsoleProcessList = ctypes.windll.kernel32.GetConsoleProcessList
num = GetConsoleProcessList(ctypes.byref(ctypes.c_int(0)), ctypes.c_int(1))
if (num == 1):
return True
except Exception:
pass
return False
def nice_exit(ret=0):
"""Drop-in replacement for sys.exit that will automatically detect
bare consoles and wait for user input before closing.
"""
if ret and is_bare_console():
print
print "Press [Enter] to close this window."
raw_input()
sys.exit(ret)
def main(): def main():
# bootstrap the logger with defaults # bootstrap the logger with defaults
logger.configure() logger.configure()
@@ -156,7 +129,7 @@ def main():
if len(args) == 0 and not options.config: if len(args) == 0 and not options.config:
# first provide an appropriate error for bare-console users # first provide an appropriate error for bare-console users
# that don't provide any options # that don't provide any options
if is_bare_console(): if util.is_bare_console():
print "\n" print "\n"
print "The Overviewer is a console program. Please open a Windows command prompt" print "The Overviewer is a console program. Please open a Windows command prompt"
print "first and run Overviewer from there. Further documentation is available at" print "first and run Overviewer from there. Further documentation is available at"
@@ -487,10 +460,10 @@ if __name__ == "__main__":
multiprocessing.freeze_support() multiprocessing.freeze_support()
try: try:
ret = main() ret = main()
nice_exit(ret) util.nice_exit(ret)
except Exception, e: except Exception, e:
logging.exception("""An error has occurred. This may be a bug. Please let us know! logging.exception("""An error has occurred. This may be a bug. Please let us know!
See http://docs.overviewer.org/en/latest/index.html#help See http://docs.overviewer.org/en/latest/index.html#help
This is the error that occurred:""") This is the error that occurred:""")
nice_exit(1) util.nice_exit(1)

View File

@@ -68,4 +68,4 @@ def check_c_overviewer():
if not sys.argv[0].endswith("setup.py"): if not sys.argv[0].endswith("setup.py"):
ret = check_c_overviewer() ret = check_c_overviewer()
if ret > 0: if ret > 0:
util.exit(ret) util.nice_exit(ret)

View File

@@ -82,6 +82,33 @@ def findGitVersion():
except Exception: except Exception:
return "unknown" return "unknown"
def is_bare_console():
"""Returns true if Overviewer is running in a bare console in
Windows, that is, if overviewer wasn't started in a cmd.exe
session.
"""
if platform.system() == 'Windows':
try:
import ctypes
GetConsoleProcessList = ctypes.windll.kernel32.GetConsoleProcessList
num = GetConsoleProcessList(ctypes.byref(ctypes.c_int(0)), ctypes.c_int(1))
if (num == 1):
return True
except Exception:
pass
return False
def nice_exit(ret=0):
"""Drop-in replacement for sys.exit that will automatically detect
bare consoles and wait for user input before closing.
"""
if ret and is_bare_console():
print
print "Press [Enter] to close this window."
raw_input()
sys.exit(ret)
# http://docs.python.org/library/itertools.html # http://docs.python.org/library/itertools.html
def roundrobin(iterables): def roundrobin(iterables):
"roundrobin('ABC', 'D', 'EF') --> A D E B F C" "roundrobin('ABC', 'D', 'EF') --> A D E B F C"