Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

API to get/set user agent

Browse Source
This commit is contained in:
Kovid Goyal 2017-02-28 23:33:36 +05:30
parent 5d3c99df59
commit 1dfbd333a3
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/calibre/utils/browser.py
View File

@ -53,6 +53,20 @@ class Browser(B):
def https_handler(self): def https_handler(self):
return self._ua_handlers['https'] return self._ua_handlers['https']
def current_user_agent(self):
for k, v in self.addheaders:
if k.lower() == 'user-agent':
return v
def change_user_agent(self, newval):
found = False
for i, (k, v) in enumerate(tuple(self.addheaders)):
if k.lower() == 'user-agent':
self.addheaders[i] = newval
found = True
if not found:
self.addheaders.append(('User-agent', newval))
def set_handle_refresh(self, *args, **kwargs): def set_handle_refresh(self, *args, **kwargs):
B.set_handle_refresh(self, *args, **kwargs) B.set_handle_refresh(self, *args, **kwargs)
self._clone_actions['set_handle_refresh'] = ('set_handle_refresh', self._clone_actions['set_handle_refresh'] = ('set_handle_refresh',
@ -131,6 +145,7 @@ class Browser(B):
func(*args, **kwargs) func(*args, **kwargs)
return clone return clone
if __name__ == '__main__': if __name__ == '__main__':
from calibre import browser from calibre import browser
from pprint import pprint from pprint import pprint
@ -142,5 +157,3 @@ if __name__ == '__main__':
assert orig._ua_handlers['_cookies'].cookiejar is \ assert orig._ua_handlers['_cookies'].cookiejar is \
clone._ua_handlers['_cookies'].cookiejar clone._ua_handlers['_cookies'].cookiejar
assert orig.addheaders == clone.addheaders assert orig.addheaders == clone.addheaders