def batchProcess(request):
inputs = batchProcessorForm(request.POST)
if(inputs.is_valid()):
url = inputs.cleaned_data['processingURL']
unnecArgs = inputs.cleaned_data['choices']
remainingLinks = inputs.cleaned_data['remainingLinks']
#DB Connect to add unnecArgs.
if(unnecArgs != ""):
db = MySQLdb.connect(host='kozmo.cis.upenn.edu', user='******', passwd='2U1ZWVmZ', db='twitter03')
print "db connected"
cursor=db.cursor()
try:
cursor.execute("update twitter_links set unnecArgs="+unnecArgs+" where expanded_link="+url)
print "Successfully added unnec args to db: "+unnecArgs
except:
print "oh shiiiittttt. error."
db.close()
#Loop back to run with remaining links.
c=Client()
return c.post("/batchSubmit/", {'links':remainingLinks})
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))
