From: Roger Meier <roger@apache.org>
Date: Fri, 13 Apr 2012 14:20:08 +0000 (+0000)
Subject: THRIFT-1568 Thrift: Add User-Agent header to Python THttpClient
X-Git-Tag: 0.9.1~405
X-Git-Url: https://source.supwisdom.com/gerrit/gitweb?a=commitdiff_plain;h=3f5a264cd9cfafa478272d5d5d41928dc69ee39b;p=common%2Fthrift.git

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

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)