From 22c72432a7090bc03accf2937d72d5ac3e54a4ed Mon Sep 17 00:00:00 2001 From: boltgolt Date: Sat, 10 Feb 2018 15:09:49 +0100 Subject: [PATCH] Moveing CLI commands to seperate files --- cli.py | 30 +----------------------------- cli/__init__.py | 0 cli/list.py | 32 ++++++++++++++++++++++++++++++++ 3 files changed, 33 insertions(+), 29 deletions(-) create mode 100644 cli/__init__.py create mode 100644 cli/list.py diff --git a/cli.py b/cli.py index 2f1bb96..2c645f1 100755 --- a/cli.py +++ b/cli.py @@ -1,35 +1,7 @@ #!/usr/bin/env python3 import sys -import os -import json -import time -path = os.path.dirname(os.path.realpath(__file__)) cmd = sys.argv[1] if cmd == "list": - if not os.path.exists(path + "/models"): - print("Face models have not been initialized yet, please run:") - print("\n\thowdy add\n") - sys.exit(1) - - user = os.environ.get("USER") - enc_file = path + "/models/" + user + ".dat" - - try: - encodings = json.load(open(enc_file)) - except FileNotFoundError: - print("No face model known for the current user (" + user + "), please run:") - print("\n\thowdy add\n") - sys.exit(1) - - print("Known face models for " + user + ":") - print("\n\t\033[1;29mID Date Label\033[0m") - - for enc in encodings: - print("\t" + str(enc["id"]), end="") - print((4 - len(str(enc["id"]))) * " ", end="") - print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(enc["time"])), end="") - print(" " + enc["label"]) - - print() + import cli.list diff --git a/cli/__init__.py b/cli/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/cli/list.py b/cli/list.py new file mode 100644 index 0000000..78a62aa --- /dev/null +++ b/cli/list.py @@ -0,0 +1,32 @@ +import sys +import os +import json +import time + +path = os.path.dirname(os.path.realpath(__file__)) + "/.." + +if not os.path.exists(path + "/models"): + print("Face models have not been initialized yet, please run:") + print("\n\thowdy add\n") + sys.exit(1) + +user = os.environ.get("USER") +enc_file = path + "/models/" + user + ".dat" + +try: + encodings = json.load(open(enc_file)) +except FileNotFoundError: + print("No face model known for the current user (" + user + "), please run:") + print("\n\thowdy add\n") + sys.exit(1) + +print("Known face models for " + user + ":") +print("\n\t\033[1;29mID Date Label\033[0m") + +for enc in encodings: + print("\t" + str(enc["id"]), end="") + print((4 - len(str(enc["id"]))) * " ", end="") + print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(enc["time"])), end="") + print(" " + enc["label"]) + +print()