[nvim] Update UltiSnips script snippet for Python

This commit is contained in:
Eryn Wells 2024-09-27 11:37:33 -07:00
parent 2882bf085e
commit 3dd367982a

View file

@ -11,11 +11,13 @@ def parse_args():
endsnippet endsnippet
snippet ifmain snippet ifmain
def main(): def main(argv):
${1:pass} ${1:pass}
if __name__ == '__main__': if __name__ == '__main__':
main() import sys
sys.exit(main(sys.argv))
endsnippet endsnippet
snippet script snippet script
@ -28,18 +30,20 @@ snippet script
import argparse import argparse
def parse_args(argv, *a, **kw): def parse_args(argv, *a, **kw):
parser = argparse.ArgumentParser(*a, **kw) parser = argparse.ArgumentParser(*a, **kw)
# TODO: Configure arguments here. # TODO: Configure arguments here.
args = parser.parse_args(argv) args = parser.parse_args(argv)
return args return args
def main(argv): def main(argv):
args = parse_args(argv[1:], prog=argv[0]) args = parse_args(argv[1:], prog=argv[0])
${1:# TODO} ${1:# TODO}
if __name__ == '__main__': if __name__ == '__main__':
import sys import sys
result = main(sys.argv) sys.exit(main(sys.argv))
sys.exit(0 if not result else result)
endsnippet endsnippet