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 get_url(self, code): query = {'code': code} return url.get_url_by_key(query)