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

DRYer

Browse Source
This commit is contained in:
Kovid Goyal 2017-02-28 23:56:24 +05:30
parent 1dfbd333a3
commit 0ac9ac2e15
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -53,19 +53,23 @@ class Browser(B):
def https_handler(self):
return self._ua_handlers['https']
def set_current_header(self, header, value):
found = False
q = header.lower()
for i, (k, v) in enumerate(tuple(self.addheaders)):
if k.lower() == q:
self.addheaders[i] = value
found = True
if not found:
self.addheaders.append((header, value))
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_user_agent(self, newval):
self.set_current_header('User-agent', newval)
def set_handle_refresh(self, *args, **kwargs):
B.set_handle_refresh(self, *args, **kwargs)