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')
