Commit 63e4396a authored by Dom Sekotill's avatar Dom Sekotill
Browse files

Use a KeyboardInterrupt handler for the CLI script

parent 81485924

list_files/__init__.py

+7 −0
Original line number Diff line number Diff line
@@ -15,3 +15,10 @@ def main(): 
	options = argp.parse_args()

	for f in walk.walk_files(options.start_dir):

		print(f)





def mainwrap():

	try:

		main()

	except KeyboardInterrupt:

		pass

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -12,6 +12,6 @@ setup( 
	packages = find_packages(exclude=['util']),



	entry_points = {

		'console_scripts': ['list-files=list_files:main'],

		'console_scripts': ['list-files=list_files:mainwrap'],

	},

)