.gitignore

@@ -1,4 +1,2 @@
 venv/
 .vscode
-notes.txt
-

README.md

@@ -1,7 +1 @@
-# IRC DOSE MONITORING and MANAGEMENT UTILITY
+Log and annotate your doses
-
-Built around the Limnoria Bot Framework. 
-
-It's a tool to log and retrieve doses. 
-
-

config.py

@@ -1,5 +1,5 @@
 ###
-# Copyright (c) 2020, mogad0n
+# Copyright (c) 2021, Pratyush Desai
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -29,11 +29,9 @@
 ###
 
 from supybot import conf, registry
-
 try:
     from supybot.i18n import PluginInternationalization
-
+    _ = PluginInternationalization('DoseLogs')
-    _ = PluginInternationalization("Tripsit")
 except:
     # Placeholder that allows to run the plugin on a bot
     # without the i18n module
@@ -46,13 +44,12 @@ def configure(advanced):
     # user or not.  You should effect your configuration by manipulating the
     # registry as appropriate.
     from supybot.questions import expect, anything, something, yn
-
+    conf.registerPlugin('DoseLogs', True)
-    conf.registerPlugin("Tripsit", True)
 
 
-Tripsit = conf.registerPlugin("Tripsit")
+DoseLogs = conf.registerPlugin('DoseLogs')
 # This is where your configuration variables (if any) should go.  For example:
-# conf.registerGlobalValue(Tripsit, 'someConfigVariableName',
+# conf.registerGlobalValue(DoseLogs, 'someConfigVariableName',
 #     registry.Boolean(False, _("""Help for someConfigVariableName.""")))
 
 

plugin.py

@@ -1,8 +1,8 @@
 ###
-# Copyright (c) 2020, mogad0n
+# Copyright (c) 2021, Pratyush Desai
 # All rights reserved.
 #
-# Redistribution and use in source and binary forms, with or wthout
+# 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,
@@ -28,344 +28,23 @@
 
 ###
 
-from humanize import ordinal
+from supybot import utils, plugins, ircutils, callbacks
-from supybot import utils, plugins, ircutils, callbacks, world, conf, log
 from supybot.commands import *
-
-
-from num2words import num2words
-import dateutil.parser
-import json
-import requests
-import pickle
-import sys
-import datetime
-import time
-import pytz
-
 try:
     from supybot.i18n import PluginInternationalization
-
+    _ = PluginInternationalization('DoseLogs')
-    _ = PluginInternationalization("DoseLogs")
 except ImportError:
     # Placeholder that allows to run the plugin on a bot
     # without the i18n module
     _ = lambda x: x
 
-filename = conf.supybot.directories.data.dirize("DoseLogs.db")
-
 
 class DoseLogs(callbacks.Plugin):
-    """Tools for tracking and controlling substance use"""
+    """Log and annotate your doses"""
-
     threaded = True
 
-    def __init__(self, irc):
-        self.__parent = super(DoseLogs, self)
-        self.__parent.__init__(irc)
-        self.db = {}
-        self._loadDb()
-        world.flushers.append(self._flushDb)
 
-    def _loadDb(self):
+Class = DoseLogs
-        """Loads the (flatfile) database mapping nicks to doses."""
-
-        try:
-            with open(filename, "rb") as f:
-                self.db = pickle.load(f)
-        except Exception as e:
-            self.log.debug("DoseLogs: Unable to load pickled database: %s", e)
-
-    def _flushDb(self):
-        """Flushes the (flatfile) database mapping nicks to doses."""
-
-        try:
-            with open(filename, "wb") as f:
-                pickle.dump(self.db, f, 2)
-        except Exception as e:
-            self.log.warning("DoseLogs: Unable to write pickled database: %s", e)
-
-    def die(self):
-        self._flushDb()
-        world.flushers.remove(self._flushDb)
-        self.__parent.die()
-
-    def set(self, irc, msg, args, timezone):
-        """<IANA-timezone>
-
-        Sets location for your current nick to <timezone>
-        for eg. America/Chicago
-        """
-        nick = msg.nick
-        try:
-            timezone = pytz.timezone(timezone)
-            if nick in self.db:
-                self.db[nick]["timezone"] = timezone
-            else:
-                self.db[nick] = {"timezone": timezone}
-            irc.replySuccess()
-        except pytz.UnknownTimeZoneError:
-            irc.error(_("Unknown timezone"), Raise=True)
-
-    set = wrap(set, ["something"])
-
-    @wrap(
-        [getopts({"ago": "something"}), "something", "something", optional("something")]
-    )
-    def idose(self, irc, msg, args, opts, dose, name, method):
-        """[--ago <HHMM>] <amount> <drug> [<method/ROA>]
-
-        Logs a dose for your nick, Ex. @idose --ago 0100 20mg mph oral
-        would log that dose as if it was taken an hour ago.
-        [--ago] and [ROA] fields are optional
-        """
-        opts = dict(opts)
-        found_method = False
-        onset = None
-        methods = []
-        if method:
-            methods = [method.lower()]
-            methods = METHODS.get(methods[0], methods)
-
-        drug_and_method = name
-        if method:
-            if not found_method:
-                method = method.title()
-            drug_and_method = "%s via %s" % (drug_and_method, method)
-        else:
-            method = "Undefined"
-
-        nick = msg.nick
-        if nick in self.db:
-            timezone = self.db[nick].get("timezone", "UTC")
-            tz = pytz.timezone(str(timezone))
-            time = datetime.datetime.now(tz=tz)
-            dose_td = 0
-            if "ago" in opts and len(opts["ago"]) == 4:
-                ago = opts["ago"]
-                dose_td = datetime.timedelta(hours=int(ago[0:2]), minutes=int(ago[2:4]))
-                dose_td_s = dose_td.total_seconds()
-                time = time - dose_td
-            doseLog = {"time": time, "dose": dose, "drug": name, "method": method}
-            doses = self.db[nick].get("doses")
-            if doses:
-                doses.append(doseLog)
-            else:
-                doses = [doseLog]
-            self.db[nick]["doses"] = doses
-        else:
-            timezone = "UTC"
-            tz = pytz.timezone(timezone)
-            time = datetime.datetime.now(tz=tz)
-            dose_td = 0
-            if "ago" in opts and len(opts["ago"]) == 4:
-                ago = opts["ago"]
-                dose_td = datetime.timedelta(hours=int(ago[0:2]), minutes=int(ago[2:4]))
-                dose_td_s = dose_td.total_seconds()
-                time = time - dose_td
-            doseLog = {"time": time, "dose": dose, "drug": name, "method": method}
-            doses = [doseLog]
-            self.db[nick] = {"timezone": timezone, "doses": doses}
-
-        if dose_td == 0:
-            re = utils.str.format(
-                "You dosed %s of %s at %s, %s",
-                dose,
-                drug_and_method,
-                time.strftime("%c"),
-                timezone,
-            )
-            if onset is not None:
-                re += utils.str.format(
-                    ". You should start feeling effects %s from now", onset
-                )
-        else:
-            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("positiveInt")])
-    def undose(self, irc, msg, args, entry):
-        """<n>
-
-        removes your last dose entry, if <n> is provided then
-        deletes the nth last dose
-        """
-        nick = msg.nick
-        if nick in self.db:
-            nick_dose_log = self.db[nick]["doses"]
-            if entry:
-                try:
-                    del nick_dose_log[-int(entry)]
-                    entry = num2words(entry, to="ordinal")
-                    irc.replySuccess(
-                        f"Deleted the {entry} last dose logged for {nick} "
-                    )
-                except IndexError:
-                    irc.error("The dose entry doesn't exist")
-                return
-            else:
-                del nick_dose_log[-1]
-            irc.replySuccess(f"Deleted the last dose logged for {nick} ")
-        else:
-            irc.error(f"No doses saved for {nick}")
-
-    def doseslogged(self, irc, msg, args):
-        """
-        This command takes no arguments.
-
-        Retrieves the number of doses logged for a given nick
-        """
-        nick = msg.nick
-        if nick in self.db:
-            try:
-                nick_dose_log_count = len(self.db[nick]["doses"])
-                nick_dose_log_since = self.db[nick]["doses"][0]["time"]
-                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("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):
-        """<n>
-
-        retrieves your <n>th last logged dose
-        """
-        nick = msg.nick
-        if nick in self.db:
-            if history:
-                try:
-                    lastdose = self.db[nick]["doses"][-int(history)]
-                except IndexError:
-                    irc.error("You haven't logged that many doses")
-                    return
-            else:
-                lastdose = self.db[nick]["doses"][-1]
-            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 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,
-                    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,
-                    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
 
 
 # vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: