howdy/src/cli.py

#!/usr/bin/env python3
# CLI directly called by running the howdy command

# Import required modules
import sys
import os

# Check if if a command has been given and print help otherwise
if (len(sys.argv) < 2):
	print("Howdy IR face recognition help")
	import cli.help
	sys.exit()

# The command given
cmd = sys.argv[1]

# Call the right files for commands that don't need root
if cmd == "help":
	print("Howdy IR face recognition")
	import cli.help
elif cmd == "test":
	import cli.test
else:
	# Check if the minimum of 3 arugemnts has been met and print help otherwise
	if (len(sys.argv) < 3):
		print("Howdy IR face recognition help")
		import cli.help
		sys.exit()

	# Requre sudo for comamnds that need root rights to read the model files
	if os.getenv("SUDO_USER") is None:
		print("Please run this command with sudo")
		sys.exit(1)

	# Frome here on we require the second argument to be the username,
	# switching the command to the 3rd
	cmd = sys.argv[2]

	if cmd == "list":
		import cli.list
	elif cmd == "add":
		import cli.add
	elif cmd == "remove":
		import cli.remove
	elif cmd == "clear":
		import cli.clear
	else:
		# If the comand is invalid, check if the user hasn't swapped the username and command
		if sys.argv[1] in ["list", "add", "remove", "clear"]:
			print("Usage: howdy <user> <command>")
			sys.exit(1)
		else:
			print('Unknown command "' + cmd + '"')
			import cli.help
			sys.exit(1)