def submit(request): #Allow link input to create shortened link. link_form = LinkSubmitForm(request.POST) print(request.POST); if link_form.is_valid(): url = link_form.cleaned_data['submitForm'] #Arg Parse urlPart = url.partition('?') args = urlPart[2].split('&') combinations = [] for i in range(0,len(args)+1): for perm in itertools.combinations(args,i): combinations.append(perm) print combinations count = 0 #Image Gen for c in combinations: urlCombination = urlPart[0]+"?" for arg in c: urlCombination = urlCombination + arg +"&" urlCombination = urlCombination[0:len(urlCombination)-1] webkit2png.generate_image(urlCombination, os.path.join(settings.PROJECT_ROOT, "media/ss_%d.png"%count)) count = count + 1 #Gets the shortened link if this url has been shortened already. If not, it makes a new one. link = None try: link = Link.objects.get(url = url) except Link.DoesNotExist: pass if link == None: new_link = Link(url = url) new_link.save() link = new_link return render_to_response('results.html', {"encodedLink": link.shortUrl(), "perm": simplejson.dumps(combinations)}, context_instance=RequestContext(request))
def batchSubmit(request): inputs = batchLinkSubmitForm(request.POST) print(request.POST); if(inputs.is_valid()): links = inputs.cleaned_data['links'].split("|!|") print links url = links[0] print url updatedLinks = inputs.cleaned_data['links'].replace(url+"|!|","") print updatedLinks invBatchProcessor = batchProcessorForm() #Arg Parse urlPart = url.partition('?') args = urlPart[2].split('&') combinations = [] for i in range(0,len(args)+1): for perm in itertools.combinations(args,i): combinations.append(perm) print combinations count = 0 #Image Gen for c in combinations: urlCombination = urlPart[0]+"?" for arg in c: urlCombination = urlCombination + arg +"&" urlCombination = urlCombination[0:len(urlCombination)-1] webkit2png.generate_image(urlCombination, os.path.join(settings.PROJECT_ROOT, "media/ss_%d.png"%count)) count = count + 1 return render_to_response('batchResults.html', {"url" : url, "invBatchProcessor" : invBatchProcessor, "perm": simplejson.dumps(combinations), "links": updatedLinks}, context_instance=RequestContext(request))
def main(url): generate_image(url, 'dummy')