From 658f00f6d6314f6bb99f8a29f753c63ad734f80a Mon Sep 17 00:00:00 2001 From: Jiri Vlasak Date: Mon, 7 Nov 2022 16:19:09 +0100 Subject: [PATCH] Update drunk sum runnable --- run/drunk_sum.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/run/drunk_sum.py b/run/drunk_sum.py index eeec196..4abc0ea 100755 --- a/run/drunk_sum.py +++ b/run/drunk_sum.py @@ -1,8 +1,6 @@ #!/usr/bin/env python3 -# -*- coding: utf-8 -*- -"""Print drunk drink in time period.""" from sys import argv, exit -from cbgetter import CoffeebotGetter +from coffee_getter.bot import Bot SS = { "covfefe": "espresso;espresso lungo;cappuccino;latte macchiato", @@ -32,11 +30,11 @@ if __name__ == "__main__": print("- covfefe (equal to coffee)") print("- mate (for all Club-mates)") exit(1) - g = CoffeebotGetter() + b = Bot() if ";" in argv[1]: - g.loadDrunkList(argv[1], argv[2], argv[3]) + b.load_drunk_list(argv[1], argv[2], argv[3]) elif argv[1] in SS: - g.loadDrunkSum(SS[argv[1]], argv[2], argv[3]) + b.load_drunk_sum(SS[argv[1]], argv[2], argv[3]) else: - g.loadDrunkSum(argv[1], argv[2], argv[3]) - print(g.getMsg()) + b.load_drunk_sum(argv[1], argv[2], argv[3]) + print(b.get_msg()) -- 2.39.2