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)