def main(request, path): ''' Search for a long link matching the `path` and redirect ''' if len(path) and path[-1] == '/': # Removing trailing slash so "/jobs/" and "/jobs" redirect identically path = path[:-1] link = Link.find_by_hash(path) if link is None: # Try to find a matching short link by removing valid "catchall" suffixes path_prefix, redirect_suffix = suffix_catchall.get_hash_from(path) if redirect_suffix is not None: # If we found a suffix, we try to find a link again with the prefix link = Link.find_by_hash(path_prefix) else: redirect_suffix = None # Instrumentation prefix_tag = 'prefix:' + link.prefix if link else 'Http404' statsd.increment('workforus.clicks', tags=[prefix_tag]) statsd.set('workforus.unique_links', link.hash if link else 'Http404', tags=[prefix_tag]) statsd.set('workforus.unique_ips', get_client_ip(request), tags=['browser:' + get_browser(request)]) # 404 if link not found or register a click if the DB is not in readonly mode if link is None: raise Http404 elif not settings.SITE_READ_ONLY: link.click() # Tweak the redirection link based on the query string, redirection suffix, etc. # FIXME: Handle multiple parameters with the same name in the `url` query = request.GET.copy() if redirect_suffix is not None: query[REDIRECT_PARAM_NAME] = redirect_suffix if bool(query) and REF_PARAM_NAME not in query: # If we specify a non empty query, indicate that the shortener tweaked the url query[REF_PARAM_NAME] = REF_PARAM_DEFAULT_VALUE target_url = url_append_parameters( link.long_url, params_to_replace=query, defaults={REF_PARAM_NAME: REF_PARAM_DEFAULT_VALUE}) # Either redirect the user, or load the target page and display it directly return (proxy if link.act_as_proxy else redirect)(target_url)
def test__url_append_parameters(self): self.assertEqual( url_append_parameters('http://workfor.us', dict(), dict()), 'http://workfor.us' ) self.assertEqual( url_append_parameters('http://workfor.us', {'toto': 'tata'}, dict()), 'http://workfor.us?toto=tata' ) self.assertEqual( url_append_parameters('http://www.theuselessweb.com/', {'foo': 'search'}, {'foo': 'bar'}), 'http://www.theuselessweb.com/?foo=search' ) self.assertEqual( url_append_parameters('http://www.theuselessweb.com?a=1&b=2&z=5', {'foo': 'search'}, {'b': '3'}), 'http://www.theuselessweb.com?a=1&b=2&z=5&foo=search' ) self.assertEqual( url_append_parameters('http://www.theuselessweb.com?foo=4', {'foo': 'search'}, {'c': '42'}), 'http://www.theuselessweb.com?c=42&foo=search' )
def main(request, path): ''' Search for a long link matching the `path` and redirect ''' if len(path) and path[-1] == '/': # Removing trailing slash so "/jobs/" and "/jobs" redirect identically path = path[:-1] link = Link.find_by_hash(path) if link is None: # Try to find a matching short link by removing valid "catchall" suffixes path_prefix, redirect_suffix = suffix_catchall.get_hash_from(path) if redirect_suffix is not None: # If we found a suffix, we try to find a link again with the prefix link = Link.find_by_hash(path_prefix) else: redirect_suffix = None # Instrumentation prefix_tag = 'prefix:' + link.prefix if link else 'Http404' statsd.increment('workforus.clicks', tags=[prefix_tag]) statsd.set('workforus.unique_links', link.hash if link else 'Http404', tags=[prefix_tag]) statsd.set('workforus.unique_ips', get_client_ip(request), tags=['browser:' + get_browser(request)]) # 404 if link not found or register a click if the DB is not in readonly mode if link is None: raise Http404 elif mongoengine_is_primary(): link.click() # Tweak the redirection link based on the query string, redirection suffix, etc. # FIXME: Handle multiple parameters with the same name in the `url` query = request.GET.copy() if redirect_suffix is not None: query[REDIRECT_PARAM_NAME] = redirect_suffix if bool(query) and REF_PARAM_NAME not in query: # If we specify a non empty query, indicate that the shortener tweaked the url query[REF_PARAM_NAME] = REF_PARAM_DEFAULT_VALUE target_url = url_append_parameters( link.long_url, params_to_replace=query, defaults={REF_PARAM_NAME: REF_PARAM_DEFAULT_VALUE} ) # Either redirect the user, or load the target page and display it directly return (proxy if link.act_as_proxy else redirect)(target_url)