49 lines
1 KiB
Python
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)
|