From 3dd367982aed8ee00a01e4a36141d759d3a53863 Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Fri, 27 Sep 2024 11:37:33 -0700
Subject: [PATCH] [nvim] Update UltiSnips `script` snippet for Python

---
 config/nvim/UltiSnips/python.snippets | 48 +++++++++++++++------------
 1 file changed, 26 insertions(+), 22 deletions(-)

diff --git a/config/nvim/UltiSnips/python.snippets b/config/nvim/UltiSnips/python.snippets
index 0fe3d4b..f9b0c22 100644
--- a/config/nvim/UltiSnips/python.snippets
+++ b/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
-	# Eryn Wells <eryn@erynwells.me>
+#!/usr/bin/env python3
+# 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):
-		args = parse_args(argv[1:], prog=argv[0])
-		${1:# TODO}
-	
-	if __name__ == '__main__':
-		import sys
-		result = main(sys.argv)
-		sys.exit(0 if not result else result)
+
+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):
+	args = parse_args(argv[1:], prog=argv[0])
+	${1:# TODO}
+
+
+if __name__ == '__main__':
+	import sys
+	sys.exit(main(sys.argv))
 endsnippet