from sys import argv, exit
from cbgetter import CoffeebotGetter
+SS = {
+ "covfefe": "espresso;espresso lungo;cappuccino;latte macchiato",
+ "coffee": "espresso;espresso lungo;cappuccino;latte macchiato",
+}
+
if __name__ == "__main__":
if len(argv) != 4:
- print("Usage: python3 drunk_sum.py {} {} {}".format(
+ print("Usage: PYTHONPATH=. python3 run/drunk_sum.py {} {} {}".format(
"\"espresso lungo\"",
"\"2010-10-20\"",
"\"2020-10-20\""
))
+ print("Or: PYTHONPATH=. python3 run/drunk_sum.py {} {} {}".format(
+ "\"espresso lungo;espresso\"",
+ "\"2010-10-20\"",
+ "\"2020-10-20\""
+ ))
+ print("Or: PYTHONPATH=. python3 run/drunk_sum.py {} {} {}".format(
+ "coffee",
+ "2010-10-20",
+ "2020-10-20"
+ ))
exit(1)
g = CoffeebotGetter()
- g.loadDrunkSum(argv[1], argv[2], argv[3])
+ if ";" in argv[1]:
+ g.loadDrunkList(argv[1], argv[2], argv[3])
+ elif argv[1] in SS:
+ g.loadDrunkSum(SS[argv[1]], argv[2], argv[3])
+ else:
+ g.loadDrunkSum(argv[1], argv[2], argv[3])
print(g.getMsg())