Skip to content

Commit

Permalink
added validator option to prox so that you can't get anything from js…
Browse files Browse the repository at this point in the history
…-coverage
  • Loading branch information
delfick committed Dec 30, 2010
1 parent 8580ad3 commit 01d2aa5
Showing 1 changed file with 16 additions and 11 deletions.
27 changes: 16 additions & 11 deletions support/web/urls.py
@@ -1,10 +1,13 @@
from django.conf.urls.defaults import *
from django.http import HttpResponse
from django.http import HttpResponse, Http404
from django.conf import settings

import httplib

def proxy(request, domain, port, debugOnly=False, add="", path=""):
def validInstrumented(url):
return url.endswith('.js') and url.startswith('/gma')

def proxy(request, domain, port, debugOnly=False, add="", path="", validator=None):
if path:
querystring = path
else:
Expand All @@ -13,15 +16,17 @@ def proxy(request, domain, port, debugOnly=False, add="", path=""):
if add:
querystring = "%s%s" % (add, querystring)

t = httplib.HTTPConnection(domain, port)
t.request("GET", querystring)
s = t.getresponse()
if not validator or validator(querystring):
t = httplib.HTTPConnection(domain, port)
t.request("GET", querystring)
s = t.getresponse()

response = HttpResponse(s.read(), status=s.status)
for key, value in s.getheaders():
response[key] = value

response = HttpResponse(s.read(), status=s.status)
for key, value in s.getheaders():
response[key] = value

return response
return response
raise Http404

urlpatterns = patterns('',
(r'^wglemedia/(?P<path>.*)$', 'django.views.static.serve',
Expand Down Expand Up @@ -52,7 +57,7 @@ def proxy(request, domain, port, debugOnly=False, add="", path=""):
{'document_root': settings.GMA_MEDIA}),

(r'^instrumented/(?P<path>.*)$', proxy,
{'domain' : '0.0.0.0', 'port' : 8081, 'add' : '/'}
{'domain' : '0.0.0.0', 'port' : 8081, 'add' : '/', 'validator':validInstrumented}
),

(r'^', include('main.urls')),
Expand Down

0 comments on commit 01d2aa5

Please sign in to comment.