def short_it(self, url_in):
new_url = {'url' : url_in}
code = helper._id_generator()
size = 5
tries = 0
query = {'code': code}
while url.get_url_by_key(query):
code = helper._id_generator(size)
tries += 1
if tries > 100000:
tries = 0
size += 1
new_url['code'] = code
url.save_url(new_url)
return code
def gen():
for i in range(5,10):
str = helper._id_generator(i)
assert i == len(str)
def gen():
for i in range(5, 10):
str = helper._id_generator(i)
assert i == len(str)
