From d58646628c61ffe0540982264178addb46fe261b Mon Sep 17 00:00:00 2001 From: Fletcher Boyd Date: Wed, 4 Sep 2013 22:49:55 +0800 Subject: [PATCH] Add random puu.sh fetcher. --- plugins/puush.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 plugins/puush.py diff --git a/plugins/puush.py b/plugins/puush.py new file mode 100644 index 0000000..202aa89 --- /dev/null +++ b/plugins/puush.py @@ -0,0 +1,47 @@ +import urllib2 +import random +from util import hook + + +def make_string(): + stuff = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890" + string = random.choice("123") + for x in range(4): + string += random.choice(stuff) + return string + + +def check_url(code): + try: + x = urllib2.urlopen(make_url(code)) + return True + except: + return False # sorry <3 + + +def make_url(code): + return "http://puu.sh/" + code + + +@hook.command(autohelp=False) +def puush(inp): + """puush -- Returns a random puush entry.""" + out = "" + num = 0 + if not inp: + inp = "1" + if inp[0] not in "123456789": + out += "Defaulting to one: " + num = 1 + elif int(inp[0]) > 5: + out += "Five images max: " + num = 5 + else: + num = int(inp[0]) + images = [] + for x in range(num): + ran = make_string() + while not check_url(ran): + ran = make_string() + images.append(make_url(ran)) + return out + " ".join(images)