This repository has been archived on 2023-04-13. You can view files and clone it, but cannot push or open issues or pull requests.
CloudBot/plugins/puush.py
2013-09-11 02:28:43 +12:00

49 lines
1 KiB
Python

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:
urllib2.urlopen(make_url(code))
return True
except:
return False # sorry <3
def make_url(code):
return "http://puu.sh/{}".format(code)
@hook.command(autohelp=False)
def puush(inp):
"""puush [1-5] -- Returns a number of random puu.sh entries."""
out = ""
num = 0
if not inp:
inp = "1"
if not inp.isdigit():
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 xrange(num):
ran = make_string()
while not check_url(ran):
ran = make_string()
images.append(make_url(ran))
return out + " ".join(images)