[nvim] Update UltiSnips script snippet for Python

config/nvim/UltiSnips/python.snippets

@@ -11,35 +11,39 @@ def parse_args():
 endsnippet
 
 snippet ifmain
-def main():
+def main(argv):
 	${1:pass}
 
+
 if __name__ == '__main__':
-	main()
+	import sys
+	sys.exit(main(sys.argv))
 endsnippet
 
 snippet script
-	#!/usr/bin/env python3
+#!/usr/bin/env python3
-	# Eryn Wells <eryn@erynwells.me>
+# Eryn Wells <eryn@erynwells.me>
 
-	'''
+'''
-	New script.
+New script.
-	'''
+'''
 
-	import argparse
+import argparse
 
-	def parse_args(argv, *a, **kw):
-		parser = argparse.ArgumentParser(*a, **kw)
-		# TODO: Configure arguments here.
-		args = parser.parse_args(argv)
-		return args
 
-	def main(argv):
+def parse_args(argv, *a, **kw):
-		args = parse_args(argv[1:], prog=argv[0])
+	parser = argparse.ArgumentParser(*a, **kw)
-		${1:# TODO}
+	# TODO: Configure arguments here.
+	args = parser.parse_args(argv)
+	return args
 
-	if __name__ == '__main__':
+
-		import sys
+def main(argv):
-		result = main(sys.argv)
+	args = parse_args(argv[1:], prog=argv[0])
-		sys.exit(0 if not result else result)
+	${1:# TODO}
+
+
+if __name__ == '__main__':
+	import sys
+	sys.exit(main(sys.argv))
 endsnippet