]> git.bitcoin.ninja Git - youtube-dl/commitdiff
Added support for custom user agents.
authorhdclark <hdclark@ualberta.ca>
Thu, 23 Aug 2012 06:40:35 +0000 (23:40 -0700)
committerhdclark <hdclark@ualberta.ca>
Thu, 23 Aug 2012 06:40:35 +0000 (23:40 -0700)
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

index 86951840dedf44c61298224fb6d789a5739ed91d..d457d6b8226b4ae6ee9f91ed4fd31d5f32cc696d 100644 (file)
@@ -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,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']