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(
))
exit(1)
g = CoffeebotGetter()
- g.loadDrunkSum(argv[1], argv[2], argv[3])
+ if 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())