LimnoriaPlugins/TripSit
6
0
Fork 0
Code Issues 5 Pull Requests Projects Releases Wiki Activity

Merge pull request 'Commit manual changes' (#9) from production into master

Browse Source 
Reviewed-on: #9
This commit is contained in:
Pratyush Desai 2024-10-29 22:33:05 +01:00
commit 844491972e
1 changed files with 139 additions and 73 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

212
plugin.py
View File

@ -2,7 +2,7 @@
# Copyright (c) 2020, mogad0n # Copyright (c) 2020, mogad0n
# All rights reserved. # 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: # modification, are permitted provided that the following conditions are met:
# #
# * Redistributions of source code must retain the above copyright notice, # * Redistributions of source code must retain the above copyright notice,
@ -54,8 +54,10 @@ except ImportError:
filename = conf.supybot.directories.data.dirize("Tripsit.db") filename = conf.supybot.directories.data.dirize("Tripsit.db")
url_drug = "http://tripbot.tripsit.me/api/tripsit/getDrug" #url_drug = "http://tripbot.tripsit.me/api/tripsit/getDrug"
url_combo = "http://tripbot.tripsit.me/api/tripsit/getInteraction" #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"] insufflated = ["Insufflation", "Insufflation-IR", "Insufflation-XR"]
@ -113,49 +115,49 @@ class Tripsit(callbacks.Plugin):
fetches data on drug from tripsit wiki fetches data on drug from tripsit wiki
""" """
category_list = [] # category_list = []
r = requests.get(url_drug, params={"name": name}).json() # r = requests.get(url_drug, params={"name": name}).json()
if not r['err']: # if not r['err']:
drug = r["data"][0]["pretty_name"] # drug = r["data"][0]["pretty_name"]
properties = r["data"][0]["properties"] # properties = r["data"][0]["properties"]
for key in properties: # for key in properties:
category_list.append(key) # category_list.append(key)
if category is None: # if category is None:
re = drug + " Available categories are: " + ", ".join(category_list) # re = drug + " Available categories are: " + ", ".join(category_list)
irc.reply(re) # irc.reply(re)
else: # else:
if category in properties.keys(): # if category in properties.keys():
re = drug + " " + properties[category] # re = drug + " " + properties[category]
irc.reply(re) # irc.reply(re)
else: # else:
irc.error(f"Unknown category {drug} Available categories are: " + ", ".join(category_list)) # irc.error(f"Unknown category {drug} Available categories are: " + ", ".join(category_list))
else: # else:
irc.error("unknown drug") # irc.error("unknown drug")
irc.reply("This no longer works, try @psywiki " + name + " " + category)
def combo(self, irc, msg, args, drugA, drugB): # def combo(self, irc, msg, args, drugA, drugB):
"""<drugA> <drugB> # """<drugA> <drugB>
#
fetches known interactions between the substances provided. # fetches known interactions between the substances provided.
""" # """
r = requests.get(url_combo, params={f"drugA": drugA, f"drugB": drugB}).json() # r = requests.get(url_combo, params={f"drugA": drugA, f"drugB": drugB}).json()
if not r["err"] and r["data"][0]: # if not r["err"] and r["data"][0]:
interaction = r["data"][0] # interaction = r["data"][0]
drug_a = interaction["interactionCategoryA"] # drug_a = interaction["interactionCategoryA"]
drug_b = interaction["interactionCategoryB"] # drug_b = interaction["interactionCategoryB"]
interaction_status = interaction["status"] # interaction_status = interaction["status"]
re = f"{drug_a} and {drug_b}: {interaction_status}" # re = f"{drug_a} and {drug_b}: {interaction_status}"
if 'note' in interaction: # if 'note' in interaction:
note = interaction["note"] # note = interaction["note"]
re += f'. Note: {note}' # re += f'. Note: {note}'
irc.reply(re) # irc.reply(re)
else: # else:
irc.reply(re) # irc.reply(re)
else: # 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.") # 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")])
combo = wrap(combo, [("something"), ("something")])
# TitleCase for lowercase inputs
def set(self, irc, msg, args, timezone): def set(self, irc, msg, args, timezone):
"""<timezone> """<timezone>
@ -183,35 +185,35 @@ class Tripsit(callbacks.Plugin):
[--ago] and [ROA] fields are optional [--ago] and [ROA] fields are optional
""" """
opts = dict(opts) opts = dict(opts)
r = requests.get(url_drug, params={"name": name}).json() # r = requests.get(url_drug, params={"name": name}).json()
found_method = False found_method = False
onset = None onset = None
if not r['err']: # if not r['err']:
drug = r['data'][0] # drug = r['data'][0]
drug_name = drug['pretty_name'] # drug_name = drug['pretty_name']
method_keys = ['value'] # method_keys = ['value']
methods = [] methods = []
if method: if method:
methods = [method.lower()] methods = [method.lower()]
methods = METHODS.get(methods[0], methods) methods = METHODS.get(methods[0], methods)
method_keys += methods # method_keys += methods
if 'formatted_onset' in drug: # if 'formatted_onset' in drug:
match = list(set(method_keys)& # match = list(set(method_keys)&
set(drug["formatted_onset"].keys())) # set(drug["formatted_onset"].keys()))
if match: # if match:
onset = drug["formatted_onset"][match[0]] # onset = drug["formatted_onset"][match[0]]
found_method = True # found_method = True
if match[0] in methods: # if match[0] in methods:
method = (match or [method])[0] # method = (match or [method])[0]
#
if onset and "_unit" in drug["formatted_onset"]: # if onset and "_unit" in drug["formatted_onset"]:
onset = "%s %s" % ( # onset = "%s %s" % (
onset, drug["formatted_onset"]["_unit"]) # onset, drug["formatted_onset"]["_unit"])
drug_and_method = name drug_and_method = name
if method: if method:
if not found_method: if not found_method:
method = method.title() method = method
drug_and_method = "%s via %s" % (drug_and_method, method) drug_and_method = "%s via %s" % (drug_and_method, method)
else: else:
method = 'Undefined' 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()) 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: if onset is not None:
re += utils.str.format(". You should have/will start feeling effects %s from/after dosing", onset) 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) irc.reply(re)
@wrap([optional('postiveInt')])
@wrap([optional('positiveInt')])
def undose(self, irc, msg, args, entry): def undose(self, irc, msg, args, entry):
"""<n> """<n>
@ -282,8 +286,7 @@ class Tripsit(callbacks.Plugin):
else: else:
irc.error(f'No doses saved for {nick}') irc.error(f'No doses saved for {nick}')
@wrap() def doseslogged(self, irc, msg, args):
def doseslogged(self, irc, msg, args, history):
""" """
This command takes no arguments. This command takes no arguments.
@ -297,10 +300,11 @@ class Tripsit(callbacks.Plugin):
nick_dose_log_since_string = nick_dose_log_since.strftime("%c") 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}") irc.reply(f"{nick} has logged {nick_dose_log_count} doses since {nick_dose_log_since_string}")
except IndexError: except IndexError:
irc.error(f"Can't seem to do math, check logs") irc.error("Can't seem to do math, check logs")
else: else:
irc.error(f"No doses saved for {nick}") irc.error(f"No doses saved for {nick}")
doseslogged = wrap(doseslogged)
@wrap([optional('positiveInt')]) @wrap([optional('positiveInt')])
def lastdose(self, irc, msg, args, history): def lastdose(self, irc, msg, args, history):
@ -329,13 +333,75 @@ class Tripsit(callbacks.Plugin):
since_dose_seconds = since_dose.total_seconds() since_dose_seconds = since_dose.total_seconds()
if history: if history:
history = num2words(history, to='ordinal') 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: 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) irc.reply(re)
else: else:
irc.error(f'No doses saved for {nick}') 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 Class = Tripsit