### # Copyright (c) 2020, mogad0n # All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions are met: # # * Redistributions of source code must retain the above copyright notice, # this list of conditions, and the following disclaimer. # * Redistributions in binary form must reproduce the above copyright notice, # this list of conditions, and the following disclaimer in the # documentation and/or other materials provided with the distribution. # * Neither the name of the author of this software nor the name of # contributors to this software may be used to endorse or promote products # derived from this software without specific prior written consent. # # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. ### 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') 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" URL_WIKI = "http://drugs.tripsit.me/%s" INSUFFLATED = ["Insufflated", "Insufflated-IR", "Insufflated-XR"] METHODS = { "iv": ["IV"], "shot": ["IV"], "im": ["IM"], "oral": ["Oral", "Oral-IR", "Oral-XR"], "insufflated": INSUFFLATED, "snorted": INSUFFLATED, "smoked": ["Smoked"] } class Tripsit(callbacks.Plugin): """Harm-Reduction tools from tripsit's tripbot and the tripsitwiki""" threaded = True @wrap(['something', optional('something')]) def drug(self, irc, msg, args, name, category): """ [] fetches data on drug from tripsit wiki """ category_list = [] r = requests.get(URL_DRUG, params={f"name": name}).json() # r = json.loads(utils.web.getUrlContent(URL_DRUG, data={f"name": name})) if not r['err']: drug = r["data"][0]["pretty_name"] properties = r["data"][0]["properties"] if category == None: # category_list = list(properties) for key in properties: category_list.append(key) re = ", ".join(category_list) # re = category_list irc.reply(re) else: if category in properties.keys(): re = properties[category] irc.reply(re) else: irc.error("Unknown category") else: irc.error("unknown drug") def combo(self, irc, msg, args, drugA, drugB): """ fetches known interaction between the substances provided. """ r = requests.get(URL_COMBO, params={f"drugA": drugA, f"drugB": drugB}).json() if not r["err"] and r["data"][0]: interaction = r["data"][0] drug_a = interaction["interactionCategoryA"] drug_b = interaction["interactionCategoryB"] interaction_status = interaction["status"] note = interaction["note"] re = f"{drug_a} and {drug_b}: {interaction_status}, Note: {note}" irc.reply(re) else: irc.error("Unknown Combo") combo = wrap(combo, [("something"), ("something")]) def idose(self, irc, msg, args, dose, name, 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 # vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: