mirror of
				https://github.com/ytdl-org/youtube-dl
				synced 2025-11-04 03:33:46 +00:00 
			
		
		
		
	Set process title to youtube-dl
This allows killing all youtube-dl processes with killall youtube-dl, and shows up nicer in some programs.
This commit is contained in:
		
							parent
							
								
									8863d0de91
								
							
						
					
					
						commit
						e3946f989e
					
				@ -62,6 +62,7 @@ from .utils import (
 | 
			
		||||
    MaxDownloadsReached,
 | 
			
		||||
    preferredencoding,
 | 
			
		||||
    SameFileError,
 | 
			
		||||
    setproctitle,
 | 
			
		||||
    std_headers,
 | 
			
		||||
    write_string,
 | 
			
		||||
)
 | 
			
		||||
@ -471,12 +472,15 @@ def parseOpts(overrideArguments=None):
 | 
			
		||||
 | 
			
		||||
    return parser, opts, args
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _real_main(argv=None):
 | 
			
		||||
    # Compatibility fixes for Windows
 | 
			
		||||
    if sys.platform == 'win32':
 | 
			
		||||
        # https://github.com/rg3/youtube-dl/issues/820
 | 
			
		||||
        codecs.register(lambda name: codecs.lookup('utf-8') if name == 'cp65001' else None)
 | 
			
		||||
 | 
			
		||||
    setproctitle(u'youtube-dl')
 | 
			
		||||
 | 
			
		||||
    parser, opts, args = parseOpts(argv)
 | 
			
		||||
 | 
			
		||||
    # Set user agent
 | 
			
		||||
 | 
			
		||||
@ -1,6 +1,7 @@
 | 
			
		||||
#!/usr/bin/env python
 | 
			
		||||
# -*- coding: utf-8 -*-
 | 
			
		||||
 | 
			
		||||
import ctypes
 | 
			
		||||
import datetime
 | 
			
		||||
import email.utils
 | 
			
		||||
import errno
 | 
			
		||||
@ -1062,3 +1063,17 @@ def month_by_name(name):
 | 
			
		||||
def fix_xml_all_ampersand(xml_str):
 | 
			
		||||
    """Replace all the '&' by '&' in XML"""
 | 
			
		||||
    return xml_str.replace(u'&', u'&')
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def setproctitle(title):
 | 
			
		||||
    try:
 | 
			
		||||
        libc = ctypes.cdll.LoadLibrary("libc.so.6")
 | 
			
		||||
    except OSError:
 | 
			
		||||
        return
 | 
			
		||||
    title = title
 | 
			
		||||
    buf = ctypes.create_string_buffer(len(title) + 1)
 | 
			
		||||
    buf.value = title
 | 
			
		||||
    try:
 | 
			
		||||
        libc.prctl(15, ctypes.byref(buf), 0, 0, 0)
 | 
			
		||||
    except AttributeError:
 | 
			
		||||
        return  # Strange libc, just skip this
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user