Merge pull request 'Commit manual changes' (#9) from production into master
Reviewed-on: #9
This commit is contained in:
commit
844491972e
204
plugin.py
204
plugin.py
@ -2,7 +2,7 @@
|
||||
# Copyright (c) 2020, mogad0n
|
||||
# All rights reserved.
|
||||
#
|
||||
# Redistribution and use in source and binary forms, with or without
|
||||
# Redistribution and use in source and binary forms, with or wthout
|
||||
# modification, are permitted provided that the following conditions are met:
|
||||
#
|
||||
# * Redistributions of source code must retain the above copyright notice,
|
||||
@ -54,8 +54,10 @@ except ImportError:
|
||||
filename = conf.supybot.directories.data.dirize("Tripsit.db")
|
||||
|
||||
|
||||
url_drug = "http://tripbot.tripsit.me/api/tripsit/getDrug"
|
||||
url_combo = "http://tripbot.tripsit.me/api/tripsit/getInteraction"
|
||||
#url_drug = "http://tripbot.tripsit.me/api/tripsit/getDrug"
|
||||
#url_combo = "http://tripbot.tripsit.me/api/tripsit/getInteraction"
|
||||
url_drug = "http://172.16.24.2:1999/api/tripsit/getDrug"
|
||||
url_combo = "http://172.16.24.2:1999/api/tripsit/getInteraction"
|
||||
|
||||
insufflated = ["Insufflation", "Insufflation-IR", "Insufflation-XR"]
|
||||
|
||||
@ -113,49 +115,49 @@ class Tripsit(callbacks.Plugin):
|
||||
|
||||
fetches data on drug from tripsit wiki
|
||||
"""
|
||||
category_list = []
|
||||
r = requests.get(url_drug, params={"name": name}).json()
|
||||
if not r['err']:
|
||||
drug = r["data"][0]["pretty_name"]
|
||||
properties = r["data"][0]["properties"]
|
||||
for key in properties:
|
||||
category_list.append(key)
|
||||
if category is None:
|
||||
re = drug + " Available categories are: " + ", ".join(category_list)
|
||||
irc.reply(re)
|
||||
else:
|
||||
if category in properties.keys():
|
||||
re = drug + " " + properties[category]
|
||||
irc.reply(re)
|
||||
else:
|
||||
irc.error(f"Unknown category {drug} Available categories are: " + ", ".join(category_list))
|
||||
else:
|
||||
irc.error("unknown drug")
|
||||
# category_list = []
|
||||
# r = requests.get(url_drug, params={"name": name}).json()
|
||||
# if not r['err']:
|
||||
# drug = r["data"][0]["pretty_name"]
|
||||
# properties = r["data"][0]["properties"]
|
||||
# for key in properties:
|
||||
# category_list.append(key)
|
||||
# if category is None:
|
||||
# re = drug + " Available categories are: " + ", ".join(category_list)
|
||||
# irc.reply(re)
|
||||
# else:
|
||||
# if category in properties.keys():
|
||||
# re = drug + " " + properties[category]
|
||||
# irc.reply(re)
|
||||
# else:
|
||||
# irc.error(f"Unknown category {drug} Available categories are: " + ", ".join(category_list))
|
||||
# else:
|
||||
# irc.error("unknown drug")
|
||||
irc.reply("This no longer works, try @psywiki " + name + " " + category)
|
||||
|
||||
def combo(self, irc, msg, args, drugA, drugB):
|
||||
"""<drugA> <drugB>
|
||||
# def combo(self, irc, msg, args, drugA, drugB):
|
||||
# """<drugA> <drugB>
|
||||
#
|
||||
# fetches known interactions 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"]
|
||||
# re = f"{drug_a} and {drug_b}: {interaction_status}"
|
||||
# if 'note' in interaction:
|
||||
# note = interaction["note"]
|
||||
# re += f'. Note: {note}'
|
||||
# irc.reply(re)
|
||||
# else:
|
||||
# irc.reply(re)
|
||||
# else:
|
||||
# irc.reply("Unknown combo (that doesn't mean it's safe). Known combos: lsd, mushrooms, dmt, mescaline, dox, nbomes, 2c-x, 2c-t-x, amt, 5-meo-xxt, cannabis, ketamine, mxe, dxm, pcp, nitrous, amphetamines, mdma, cocaine, caffeine, alcohol, ghb/gbl, opioids, tramadol, benzodiazepines, maois, ssris.")
|
||||
#
|
||||
# combo = wrap(combo, [("something"), ("something")])
|
||||
|
||||
fetches known interactions 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"]
|
||||
re = f"{drug_a} and {drug_b}: {interaction_status}"
|
||||
if 'note' in interaction:
|
||||
note = interaction["note"]
|
||||
re += f'. Note: {note}'
|
||||
irc.reply(re)
|
||||
else:
|
||||
irc.reply(re)
|
||||
else:
|
||||
irc.reply("Unknown combo (that doesn't mean it's safe). Known combos: lsd, mushrooms, dmt, mescaline, dox, nbomes, 2c-x, 2c-t-x, amt, 5-meo-xxt, cannabis, ketamine, mxe, dxm, pcp, nitrous, amphetamines, mdma, cocaine, caffeine, alcohol, ghb/gbl, opioids, tramadol, benzodiazepines, maois, ssris.")
|
||||
|
||||
combo = wrap(combo, [("something"), ("something")])
|
||||
|
||||
# TitleCase for lowercase inputs
|
||||
def set(self, irc, msg, args, timezone):
|
||||
"""<timezone>
|
||||
|
||||
@ -183,35 +185,35 @@ class Tripsit(callbacks.Plugin):
|
||||
[--ago] and [ROA] fields are optional
|
||||
"""
|
||||
opts = dict(opts)
|
||||
r = requests.get(url_drug, params={"name": name}).json()
|
||||
# r = requests.get(url_drug, params={"name": name}).json()
|
||||
found_method = False
|
||||
onset = None
|
||||
if not r['err']:
|
||||
drug = r['data'][0]
|
||||
drug_name = drug['pretty_name']
|
||||
method_keys = ['value']
|
||||
# if not r['err']:
|
||||
# drug = r['data'][0]
|
||||
# drug_name = drug['pretty_name']
|
||||
# method_keys = ['value']
|
||||
methods = []
|
||||
if method:
|
||||
methods = [method.lower()]
|
||||
methods = METHODS.get(methods[0], methods)
|
||||
method_keys += 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"])
|
||||
# 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 = name
|
||||
if method:
|
||||
if not found_method:
|
||||
method = method.title()
|
||||
method = method
|
||||
drug_and_method = "%s via %s" % (drug_and_method, method)
|
||||
else:
|
||||
method = 'Undefined'
|
||||
@ -256,9 +258,11 @@ class Tripsit(callbacks.Plugin):
|
||||
re = utils.str.format("You dosed %s of %s at %s, %s ; %T ago", dose, drug_and_method, time.strftime("%c"), timezone, dose_td.total_seconds())
|
||||
if onset is not None:
|
||||
re += utils.str.format(". You should have/will start feeling effects %s from/after dosing", onset)
|
||||
# re=":-( This is currently not available, sorry. Exception ID T0T4LLYFCK3D."
|
||||
irc.reply(re)
|
||||
|
||||
@wrap([optional('postiveInt')])
|
||||
|
||||
@wrap([optional('positiveInt')])
|
||||
def undose(self, irc, msg, args, entry):
|
||||
"""<n>
|
||||
|
||||
@ -282,8 +286,7 @@ class Tripsit(callbacks.Plugin):
|
||||
else:
|
||||
irc.error(f'No doses saved for {nick}')
|
||||
|
||||
@wrap()
|
||||
def doseslogged(self, irc, msg, args, history):
|
||||
def doseslogged(self, irc, msg, args):
|
||||
"""
|
||||
This command takes no arguments.
|
||||
|
||||
@ -297,10 +300,11 @@ class Tripsit(callbacks.Plugin):
|
||||
nick_dose_log_since_string = nick_dose_log_since.strftime("%c")
|
||||
irc.reply(f"{nick} has logged {nick_dose_log_count} doses since {nick_dose_log_since_string}")
|
||||
except IndexError:
|
||||
irc.error(f"Can't seem to do math, check logs")
|
||||
irc.error("Can't seem to do math, check logs")
|
||||
else:
|
||||
irc.error(f"No doses saved for {nick}")
|
||||
|
||||
doseslogged = wrap(doseslogged)
|
||||
|
||||
@wrap([optional('positiveInt')])
|
||||
def lastdose(self, irc, msg, args, history):
|
||||
@ -329,13 +333,75 @@ class Tripsit(callbacks.Plugin):
|
||||
since_dose_seconds = since_dose.total_seconds()
|
||||
if history:
|
||||
history = num2words(history, to='ordinal')
|
||||
re = utils.str.format("Your %i last dose was %s of %s via %s at %s %s, %T ago", history, dose, drug, method, str(dose_time), timezone, since_dose_seconds)
|
||||
re = utils.str.format("Your %i last dose was %s of %s via %s at %s %s, %T ago", history, dose, drug, method, dose_time.strftime("%c"), timezone, since_dose_seconds)
|
||||
else:
|
||||
re = utils.str.format("You last dosed %s of %s via %s at %s %s, %T ago", dose, drug, method, str(dose_time), timezone, since_dose_seconds)
|
||||
re = utils.str.format("You last dosed %s of %s via %s at %s %s, %T ago", dose, drug, method, dose_time.strftime("%c"), timezone, since_dose_seconds)
|
||||
irc.reply(re)
|
||||
else:
|
||||
irc.error(f'No doses saved for {nick}')
|
||||
|
||||
@wrap(['positiveInt'])
|
||||
def listdose(self, irc, msg, args, history):
|
||||
"""<n>
|
||||
|
||||
retrieves your <n> last logged doses
|
||||
|
||||
"""
|
||||
if history > 20:
|
||||
irc.error("you can't retrieve more than 20 doses")
|
||||
return
|
||||
nick = msg.nick
|
||||
if nick in self.db:
|
||||
try:
|
||||
rangecheck = self.db[nick]['doses'][-int(history)]
|
||||
irc.reply(f"Your last {history} doses logged are:", private=True)
|
||||
for number in range(history,0,-1):
|
||||
lastdose = self.db[nick]['doses'][-int(number)]
|
||||
dose = lastdose['dose']
|
||||
drug = lastdose['drug']
|
||||
method = lastdose['method']
|
||||
dose_time = lastdose['time']
|
||||
timezone = self.db[nick]['timezone']
|
||||
tz = pytz.timezone(str(timezone))
|
||||
time = datetime.datetime.now(tz=tz)
|
||||
since_dose = time - dose_time
|
||||
since_dose_seconds = since_dose.total_seconds()
|
||||
if number == 1:
|
||||
number = "The"
|
||||
else:
|
||||
number = num2words(number, to='ordinal')
|
||||
re = utils.str.format("::> %i last dose: Amount: %s of \x02%s\x0F via %s | datetime: %s %s | timedelta %T ", number, dose, drug, method, dose_time.strftime("%c"), timezone, since_dose_seconds)
|
||||
irc.reply(re, private=True)
|
||||
except IndexError:
|
||||
irc.error("You haven't logged that many doses")
|
||||
return
|
||||
else:
|
||||
irc.error(f'No doses saved for {nick}')
|
||||
|
||||
|
||||
|
||||
@wrap(["something"])
|
||||
def amountdosed(self, irc, msg, args, drug):
|
||||
"""<drug>
|
||||
|
||||
shows Aggregate amount in "mg" for <drug> ever logged
|
||||
"""
|
||||
num = 0
|
||||
unit = ""
|
||||
nick = msg.nick
|
||||
if nick in self.db:
|
||||
doselogs = self.db[nick]['doses']
|
||||
for doselog in doselogs:
|
||||
if doselog["drug"] == drug:
|
||||
for i,c in enumerate(doselog["dose"]):
|
||||
if not c.isdigit():
|
||||
break
|
||||
num += int(doselog["dose"][:i])
|
||||
unit = doselog["dose"][i:].lstrip()
|
||||
irc.reply(f"You have dosed a total of {num}{unit} amount of {drug}")
|
||||
else:
|
||||
irc.error(f"No doses saved for {nick}")
|
||||
|
||||
Class = Tripsit
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user