Esempio n. 1
0
 def choose(self, server, msg, query):
     if "," not in query: return
     context = server.lower(msg.context)
     self.decision[context] = time.time()
     self.results[context] = {
         strip(i): 0
         for i in re.split(r",|\bor\b", query)
     }
     schedule_after(1, self.report, args=(server, msg.context))
Esempio n. 2
0
 def report(self, server, channel):
     if time.time() - self.decision[server.lower(channel)] < 2:
         schedule_after(1, self.report, args=(server, channel))
         return
     results = self.results[server.lower(channel)]
     highest = max(results.values())
     choice = random.choice([i for i in results if results[i] == highest])
     results[choice] += 1
     server.message("\x0309│\x03 " + choice, channel)
     data = sorted(results.items(), key=lambda x: -x[1])
     if len(data) < 2: 
         return
     output = " · ".join("%s%s (%d)\x0f" % ("\x02" if i[0] == choice else "", i[0], i[1]) for i in data)
     server.message("\x039│\x03 " + output, channel)
Esempio n. 3
0
    def __init__(self, server):
        self.settingsf = server.get_config_dir(self.SETTINGS_FILE)
        try:
            self.settings = json.load(open(self.settingsf))
        except FileNotFoundError:
            self.settings = {}
        
        self.deletionf = server.get_config_dir(self.DELETION_FILE)
        try:
            self.deletion = json.load(open(self.deletionf))
        except FileNotFoundError:
            self.deletion = {}

        self.usersf = server.get_config_dir(self.USERS_FILE)
        try:
            self.users = json.load(open(self.usersf))
        except FileNotFoundError:
            self.users = {}

        self.unverified = {}

        self.accounts = {i: pysnap.Snapchat() for i in self.settings}
        for i in self.settings:
            self.accounts[i].login(self.settings[i]["username"], self.settings[i]["password"])
        self.cache = {}
        self.checker = scheduler.schedule_after(45, self.checksnaps, args=(server,), stop_after=None)
        self.server = server

        super().__init__(server)
Esempio n. 4
0
 def report(self, server, channel):
     if time.time() - self.decision[server.lower(channel)] < 2:
         schedule_after(1, self.report, args=(server, channel))
         return
     results = self.results[server.lower(channel)]
     highest = max(results.values())
     choice = random.choice([i for i in results if results[i] == highest])
     results[choice] += 1
     server.message("\x0309│\x03 " + choice, channel)
     data = sorted(results.items(), key=lambda x: -x[1])
     if len(data) < 2 or sum(results[i] for i in results) == 1:
         return
     output = " · ".join("%s%s (%d)\x0f" %
                         ("\x02" if i[0] == choice else "", i[0], i[1])
                         for i in data)
     server.message("\x039│\x03 " + output, channel)
Esempio n. 5
0
    def reminder(self, server, msg, user, after, text, after2, method, repeat, cancel):
        # TODO: handle inactive people
        # TODO: print reminders in last-spoke channel
        after = parse_time(after or after2)
        repeat = parse_time(repeat)
        method = (method or "channel message").lower()
        text = re.sub("\bs?he's\b", "you're", text, flags=re.IGNORECASE)
        text = re.sub("\bs?he|they\b\s+(\S+)s", r"you \1", text, flags=re.IGNORECASE)
        if user.lower() in ["me", "self"]:
            user = msg.address.nick
        if method == "snapchat":
            return "Snapchat not yet implemented."
        if re.match(r"what\s+to\s+\S+|(again\s+)later", text, re.IGNORECASE):
            return "Not yet implemented"

        jobid = uuid.uuid4().hex
        job = {"id": jobid, "sender": msg.address.nick, "message": text, "method": method, "time": time.time(), "after": after + time.time(), "channel": msg.context}

        def setreminder(job):
            self.waiting.setdefault(server.lower(user), {}).pop(jobid, None)
            comchans = sorted([i for i in server.channels if server.isIn(user, server.channels[i])], key=lambda x:not server.eq(x, msg.context))
            if comchans:
                self.send_messages([i for i in server.channels[comchans[0]] if server.lower(i) == server.lower(user)][0], comchans[0])
        
        self.reminders.setdefault(server.lower(user), []).append(job)

        if after:
            self.waiting.setdefault(server.lower(user), {})[jobid] = {"job": scheduler.schedule_after(after, setreminder, args=(job,)), "args": job}

        with open(self.server.get_config_dir(self.REMINDERF), "w") as f:
            json.dump(self.reminders, f)

        return "user=%(user)s, after=%(after)s, text=%(text)s, method=%(method)s, repeat=%(repeat)s, cancel=%(cancel)s" % locals()
Esempio n. 6
0
 def __init__(self, server):
     self.server = server
     self.last = time.time()
     self.lastcheck = time.time()
     self.watchdog = scheduler.schedule_after(90, self.check, stop_after=None)
     super().__init__(server)    
Esempio n. 7
0
 def __init__(self, server):
     self.server = server
     self.last = time.time()
     self.lastcheck = time.time()
     self.watchdog = schedule_after(90, self.check, stop_after=None)
     super().__init__(server)
Esempio n. 8
0
 def choose(self, server, msg, query):
     if "," not in query: return
     context = server.lower(msg.context)
     self.decision[context] = time.time()
     self.results[context] = {strip(i): 0 for i in re.split(r",|\bor\b", query)}
     schedule_after(1, self.report, args=(server, msg.context))
Esempio n. 9
0
    def reminder(self, server, msg, user, after, text, after2, method, repeat,
                 cancel):
        # TODO: handle inactive people
        # TODO: print reminders in last-spoke channel
        after = parse_time(after or after2)
        repeat = parse_time(repeat)
        if msg.context.startswith("#"):
            method = (method or "channel message").lower()
        else:
            method = (method or "private message").lower()
        text = re.sub("\bs?he's\b", "you're", text, flags=re.IGNORECASE)
        text = re.sub("\bs?he|they\b\s+(\S+)s",
                      r"you \1",
                      text,
                      flags=re.IGNORECASE)
        if user.lower() in ["me", "self"]:
            user = msg.address.nick
        user = user.lstrip("@")
        if method == "snapchat":
            return "Snapchat not yet implemented."
        if re.match(r"what\s+to\s+\S+|(again\s+)later", text, re.IGNORECASE):
            return "Not yet implemented"

        jobid = uuid.uuid4().hex
        job = {
            "id": jobid,
            "sender": msg.address.nick,
            "message": text,
            "method": method,
            "time": time.time(),
            "after": after + time.time(),
            "channel": msg.context
        }

        def setreminder(job):
            self.waiting.setdefault(server.lower(user), {}).pop(jobid, None)
            comchans = sorted([
                i for i in server.channels
                if server.isIn(user, server.channels[i])
            ],
                              key=lambda x: not server.eq(x, msg.context))
            if comchans:
                self.send_messages([
                    i for i in server.channels[comchans[0]]
                    if server.lower(i) == server.lower(user)
                ][0], comchans[0])

        self.reminders.setdefault(server.lower(user), []).append(job)

        if after:
            self.waiting.setdefault(server.lower(user), {})[jobid] = {
                "job": scheduler.schedule_after(after,
                                                setreminder,
                                                args=(job, )),
                "args": job
            }

        with open(self.server.get_config_dir(self.REMINDERF), "w") as f:
            json.dump(self.reminders, f)

        when = "in %s" % from_now(after) if after else "next time I see them"
        return "03│ I'll tell %s that %s." % (user, when)