diff options
| -rw-r--r-- | plugin.py | 72 |
1 files changed, 72 insertions, 0 deletions
@@ -30,8 +30,12 @@ from supybot import utils, plugins, ircutils, callbacks from supybot.commands import * +from pathlib import Path +from os import path +import dateutil.parser import json import requests +from datetime import datetime try: from supybot.i18n import PluginInternationalization _ = PluginInternationalization('Tripsit') @@ -39,6 +43,11 @@ except ImportError: # Placeholder that allows to run the plugin on a bot # without the i18n module _ = lambda x: x +dose_filename = 'dose.json' +dose_db = Path(dose_filename) +if not path.isfile(dose_filename): + dose_db.write_text('{}') +dose_data = json.loads(dose_db.read_text()) URL_DRUG = "http://tripbot.tripsit.me/api/tripsit/getDrug" URL_COMBO = "http://tripbot.tripsit.me/api/tripsit/getInteraction" @@ -107,6 +116,69 @@ class Tripsit(callbacks.Plugin): irc.error("Unknown Combo") combo = wrap(combo, [("something"), ("something")]) + def idose(self, irc, msg, args, dose, name, method): + """<amt> <drug> <method> logs a dose to keep track of + """ + r = requests.get(URL_DRUG, params={f"name": name}).json() + found_method = False + onset = None + if not r['err']: + drug = r['data'][0] + drug_name = r['data'][0]['pretty_name'] + method_keys = ['value'] + methods = [] + if method: + methods = [method.lower()] + methods = METHODS.get(methods[0], methods) + method_keys += methods + + if 'formatted_onset' in drug: + match = list(set(method_keys)& + set(drug["formatted_onset"].keys())) + if match: + onset = drug["formatted_onset"][match[0]] + found_method = True + if match[0] in methods: + method = (match or [method])[0] + + if onset and "_unit" in drug["formatted_onset"]: + onset = "%s %s" % ( + onset, drug["formatted_onset"]["_unit"]) + drug_and_method = drug_name + if method: + if not found_method: + method = method.title() + + drug_and_method = "%s via %s" % (drug_and_method, method) + + time = datetime.utcnow() + dose_data[str(msg.nick)] = { 'time': str(time), 'dose': dose, 'drug': drug_name, 'method': method } + re = f"Dosed {dose} of {drug_and_method} at {str(time)}" + + if not onset == None: + re += f". You should start feeling effects {onset} from now" + irc.reply(re) + idose = wrap(idose, [("something"), ("something"), ("something")]) + + def lastdose(self, irc, msg, args): + """ retrieves saved dose + """ + if str(msg.nick) in dose_data: + lastdose = dose_data[str(msg.nick)] + time = dateutil.parser.isoparse(lastdose['time']) + re = f"You last dosed {lastdose['dose']} of {lastdose['drug']} via {lastdose['method']} at {time}" + irc.reply(re) + else: + irc.error('No last dose saved for you') + + lastdose = wrap(lastdose) + + + + + + + Class = Tripsit |