Burpybot/uptime.py at master · tecknojock/Burpybot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
"""
uptime.py - A simple willie module
Copyright 2013, Tim Dreyer
Licensed under the Eiffel Forum License 2.

http://bitbucket.org/tdreyer/fineline
"""
import time
from datetime import timedelta

from willie.module import commands

try:
    import imp
    import sys
    from permissions import perm_chk
except:
    try:
        ffp, pathname, description = imp.find_module('permissions',['/home/dropbox/Dropbox/WillieBot'])
        permissions = imp.load_source('permissions', pathname, ffp)
        sys.modules['permissions'] = permissions
    finally:
        if ffp:
            ffp.close()
        from permissions import perm_chk


def setup(willie):
    if "uptime" not in willie.memory:
        willie.debug(u"uptime:startup",
                     u"Found no time, adding.",
                     u"verbose"
                     )
        willie.memory["uptime"] = int(time.time())
    else:
        willie.debug(u"uptime:startup",
                     u"Found time.",
                     u"verbose"
                     )


@commands('uptime')
def uptime(willie, trigger):
    if not perm_chk(trigger.hostmask, "Bc", willie):
        return
    now = int(time.time())
    then = willie.memory["uptime"]
    diff = str(timedelta(seconds=now - then))
    willie.debug(u"uptime", diff, u"verbose")
    willie.debug(u"uptime", len(diff), u"verbose")
    if len(diff) < 9:
        h, m, s = diff.split(":")
        d = '0 days'
    else:
        d, m, s = diff.split(":")
        d, h = d.split(", ")
    willie.say((u"I have had %s, %s hours, %s minutes and %s " +
                u"seconds of uptime.") % (d, h, m, s)
               )


if __name__ == "__main__":
    print __doc__.strip()