diff options
| -rw-r--r-- | libbe/storage/http.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/libbe/storage/http.py b/libbe/storage/http.py index 877fc96..cfed7b7 100644 --- a/libbe/storage/http.py +++ b/libbe/storage/http.py @@ -79,7 +79,7 @@ class InvalidURL (Exception): return self.error.__str__() return self.msg -def get_post_url(url, get=True, data_dict=None, headers=[]): +def get_post_url(url, get=True, data_dict=None, headers=[], agent=None): """Execute a GET or POST transaction. Parameters @@ -95,6 +95,8 @@ def get_post_url(url, get=True, data_dict=None, headers=[]): """ if data_dict == None: data_dict = {} + if agent is None: + agent = USER_AGENT if get == True: if data_dict != {}: # encode get parameters in the url @@ -104,7 +106,7 @@ def get_post_url(url, get=True, data_dict=None, headers=[]): else: data = urllib.urlencode(data_dict) headers = dict(headers) - headers['User-Agent'] = USER_AGENT + headers['User-Agent'] = agent req = urllib2.Request(url, data=data, headers=headers) try: response = urllib2.urlopen(req) |