Moveing CLI commands to seperate files

This commit is contained in:
boltgolt 2018-02-10 15:09:49 +01:00
parent 92a8d0a259
commit 22c72432a7
3 changed files with 33 additions and 29 deletions

30
cli.py
View file

@ -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

0
cli/__init__.py Normal file
View file

32
cli/list.py Normal file
View file

@ -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()