From ea46fe2dd441a659e9748abc5727e568583bc6e7 Mon Sep 17 00:00:00 2001 From: hdclark Date: Wed, 22 Aug 2012 23:40:35 -0700 Subject: [PATCH] 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']. --- youtube_dl/__init__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/youtube_dl/__init__.py b/youtube_dl/__init__.py index 86951840de..d457d6b822 100644 --- a/youtube_dl/__init__.py +++ b/youtube_dl/__init__.py @@ -190,6 +190,8 @@ def _find_term_columns(): 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,7 +370,10 @@ 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: print std_headers['User-Agent']