argv.py

#!/usr/bin/env python3
import sys


def argv_absent(arg):
	'''Check for command line args and run func if not found

	>>> sys.argv = ['main.py', '--meow']
	>>> @argv_absent('--meow')
	... def meow():
	...    print("meow")
	...
	'''
	def wrapper(func):
		if not arg in sys.argv[1:]:
			return func()
	return wrapper


def argv_required(arg):
	'''Check for command line args and run func if found

	>>> sys.argv = ['main.py', '--meow']
	>>> @argv_required('--meow')
	... def meow():
	...    print("meow")
	...
	meow
	'''
	def wrapper(func):
		if arg in sys.argv[1:]:
			return func()
	return wrapper


@argv_required('--doctest')
def meowtime():
	import doctest
	print(doctest.testmod(report=True))