Added support for custom user agents.
Added a few simple lines to add support for the flag "--user-agent" to pass a custom string to std_header['User-Agent'].
This commit is contained in:
parent
9bb8dc8e42
commit
ea46fe2dd4
1 changed files with 6 additions and 1 deletions
|
@ -190,6 +190,8 @@ def parseOpts():
|
|||
general.add_option('--dump-user-agent',
|
||||
action='store_true', dest='dump_user_agent',
|
||||
help='display the current browser identification', default=False)
|
||||
general.add_option('--user-agent',
|
||||
action='store', dest='useragent', help='specify a custom user agent')
|
||||
general.add_option('--list-extractors',
|
||||
action='store_true', dest='list_extractors',
|
||||
help='List all supported extractors and the URLs they would handle', default=False)
|
||||
|
@ -368,6 +370,9 @@ def _real_main():
|
|||
jar.load()
|
||||
except (IOError, OSError), err:
|
||||
sys.exit(u'ERROR: unable to open cookie file')
|
||||
# Set user agent
|
||||
if opts.useragent is not None:
|
||||
std_headers['User-Agent'] = opts.useragent
|
||||
|
||||
# Dump user agent
|
||||
if opts.dump_user_agent:
|
||||
|
|
Loading…
Reference in a new issue