expanded subcommands

1 files changed, 192 insertions, 0 deletions

diff --git a/plugin.py b/plugin.py
new file mode 100644
index 0000000..be17684
--- /dev/null
+++ b/plugin.py
@@ -0,0 +1,192 @@
+###
+# 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 *
+import requests
+
+try:
+    from supybot.i18n import PluginInternationalization
+    _ = PluginInternationalization('PsychonautWiki')
+except ImportError:
+    # Placeholder that allows to run the plugin on a bot
+    # without the i18n module
+    _ = lambda x: x
+
+url = "https://api.psychonautwiki.org/graphql"
+
+
+class PsychonautWiki(callbacks.Plugin):
+    """Queries the Psychonautwiki api"""
+    threaded = True
+
+
+    @wrap(['something', optional('something')])
+    def psywiki(self, irc, msg, args, name, category):
+        """<drug> <category>
+        fetches data on drug from psychonautwiki. Categories currently supported are name, dose, duration
+        """
+        categories = ['name', 'summary', 'dose', 'duration', 'effects', 'addictionPotential', 'crossTolerances', 'toxicity']
+        query = { 'query' : 'query getdruginfo($name: String){substances(query: $name) {name summary toxicity addictionPotential crossTolerances effects { name } roas {name dose {units threshold heavy common { min max } light { min max } strong { min max }} duration {afterglow { min max units } comeup { min max units } duration { min max units } offset { min max units } onset { min max units }peak { min max units }total { min max units }}}}}', 'variables': {'name': name}}
+        r = requests.post(url=url, json=query).json()
+
+        def format_dose(dose, unit):
+            if dose == 'N/A':
+                return 'N/A'
+            else:
+                return f"{dose['min']}-{dose['max']} {unit}"
+
+        def format_duration(duration, unit):
+            if duration == 'N/A':
+                return "N/A"
+            else:
+                return f"{duration['min']}-{duration['max']} {unit}"
+
+        if r['data']:
+            formatted_doses = []
+            formatted_duration = []
+            formatted_effects = []
+            substance = r['data']['substances'][0]
+            drug_name = substance['name']
+            summary = substance['summary']
+            toxicity = substance['toxicity']
+            addictionPotential = substance['addictionPotential']
+            crossTolerances = substance['crossTolerances']
+            effects = substance['effects']
+            for effect in effects:
+                formatted_effects.append(effect['name'])
+            roas = substance['roas']
+            for roa in roas:
+                if roa['dose']['threshold'] is not None:
+                    threshold = roa['dose']['threshold']
+                else:
+                    threshold = 'N/A'
+                if roa['dose']['heavy'] is not None:
+                    heavy = roa['dose']['heavy']
+                else:
+                    heavy = 'N/A'
+                if roa['dose']['light'] is not None:
+                    light = roa['dose']['light']
+                else:
+                    light = 'N/A'
+                if roa['dose']['common'] is not None:
+                    common = roa['dose']['common']
+                else:
+                    common = 'N/A'
+                if roa['dose']['strong'] is not None:
+                    strong = roa['dose']['strong']
+                else:
+                    strong = 'N/A'
+                unit_dose = roa['dose']['units']
+                formatted_doses.append(f"{roa['name']} threshold: {threshold}{unit_dose} light: {format_dose(light, unit_dose)} common: {format_dose(common, unit_dose)} strong: {format_dose(strong, unit_dose)} heavy: {heavy}{unit_dose}")
+                if roa['duration']['afterglow'] is not None:
+                    afterglow = roa['duration']['afterglow']
+                    unit_afterglow = roa['duration']['afterglow']['units']
+                else:
+                    afterglow = 'N/A'
+                    unit_afterglow = 'N/A'
+                if roa['duration']['comeup'] is not None:
+                    comeup = roa['duration']['comeup']
+                    unit_comeup = roa['duration']['comeup']['units']
+                else:
+                    comeup = 'N/A'
+                    unit_comeup = 'N/A'
+
+                if roa['duration']['offset'] is not None:
+                    offset = roa['duration']['offset']
+                    unit_offset = roa['duration']['offset']['units']
+                else:
+                    offset = 'N/A'
+                    unit_offset = 'N/A'
+
+                if roa['duration']['peak'] is not None:
+                    peak = roa['duration']['peak']
+                    unit_peak = roa['duration']['peak']['units']
+                else:
+                    peak = 'N/A'
+                    unit_peak = 'N/A'
+
+                if roa['duration']['onset'] is not None:
+                    onset = roa['duration']['onset']
+                    unit_onset = roa['duration']['onset']['units']
+                else:
+                    onset = 'N/A'
+                    unit_onset = 'N/A'
+
+                if roa['duration']['total'] is not None:
+                    total = roa['duration']['total']
+                    unit_total = roa['duration']['total']['units']
+                else:
+                    total = 'N/A'
+                    unit_total = 'N/A'
+
+                # if roa['duration']['duration'] in not None:
+                #     duration = roa['duration']['duration']
+                #     unit_duration = roa['duration']['duration']['units']
+                # else:
+                #     duration = 'N/A'
+                #     unit_duration = 'N/A'
+
+                formatted_duration.append(f"{roa['name']} onset: {format_duration(onset, unit_onset)} comeup: {format_duration(comeup, unit_comeup)} peak: {format_duration(peak, unit_peak)} offset: {format_duration(offset, unit_offset)} afterglow: {format_duration(afterglow, unit_afterglow)} total: {format_duration(total, unit_total)}")
+            if category in categories and category == 'name':
+                re = drug_name
+            elif category in categories and category == 'summary':
+                re = drug_name + ' ' + summary
+            elif category in categories and category == 'addictionPotential':
+                re = drug_name + ' ' + addictionPotential
+            elif category in categories and category == 'toxicity':
+                if toxicity is None:
+                    re = 'N/A'
+                else:
+                    re = drug_name + ' ' + ', '.join(toxicity)
+            elif category in categories and category == 'crossTolerances':
+                if crossTolerances is None:
+                    re = 'N/A'
+                else:
+                    re = drug_name + ' ' + ', '.join(crossTolerances)
+            elif category in categories and category == 'effects':
+                re = drug_name + ' ' + ', '.join(formatted_effects)
+            elif category in categories and category == 'dose':
+                re = drug_name + ' ' + ' | '.join(formatted_doses)
+            elif category in categories and category == 'duration':
+                re = drug_name + ' ' +  ' | '.join(formatted_duration)
+            else:
+                re = drug_name + ' ' + ', '.join(categories)
+        else:
+            re = "Unknown Substance"
+        irc.reply(re)
+
+
+
+Class = PsychonautWiki
+
+
+# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79:
+