From 3f5a264cd9cfafa478272d5d5d41928dc69ee39b Mon Sep 17 00:00:00 2001 From: Roger Meier Date: Fri, 13 Apr 2012 14:20:08 +0000 Subject: [PATCH] THRIFT-1568 Thrift: Add User-Agent header to Python THttpClient Patch: Dave Watson git-svn-id: https://svn.apache.org/repos/asf/thrift/trunk@1325781 13f79535-47bb-0310-9956-ffa450edef68 --- lib/py/src/transport/THttpClient.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/lib/py/src/transport/THttpClient.py b/lib/py/src/transport/THttpClient.py index 293d9304..ea80a1ae 100644 --- a/lib/py/src/transport/THttpClient.py +++ b/lib/py/src/transport/THttpClient.py @@ -18,7 +18,10 @@ # import httplib +import os import socket +import sys +import urllib import urlparse import warnings @@ -122,6 +125,13 @@ class THttpClient(TTransportBase): self.__http.putheader('Content-Type', 'application/x-thrift') self.__http.putheader('Content-Length', str(len(data))) + if not self.__custom_headers or 'User-Agent' not in self.__custom_headers: + user_agent = 'Python/THttpClient' + script = os.path.basename(sys.argv[0]) + if script: + user_agent = '%s (%s)' % (user_agent, urllib.quote(script)) + self.__http.putheader('User-Agent', user_agent) + if self.__custom_headers: for key, val in self.__custom_headers.iteritems(): self.__http.putheader(key, val) -- 2.17.1