Descargar vídeos de YouTube y otros sitios

Código fuente Python para la descarga de vídeos de YouTube y otros sitios soportados. Para esto es necesaria la librería externa youtube-dl

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
3. 
   
4. 
   
5. import os
6. import sys
7. import youtube_dl
8. from urllib2 import urlopen
9. import StringIO
10. import cPickle as pickle
11. 
    
12. 
    
13. FILE_DESCARGAS = "descargas.dat" #archivo donde se guardaran los links
14. FORMAT_ASP = ["best", "worst", "bestaudio", "worstaudio", "bestvideo", "worstvideo"]
15. EXTENSIONS = ["mp4", "flv", "webm", "3gp", "m4a", "mp3", "ogg", "aac", "wav"]
16. 
    
17. 
    
18. 
    
19. class VideoDownloader(object):
20.     """
21.     Clase para la descarga de videos.
22.     """
23.     
24. 
    
25.     def Downloader(self, url, format_asp=None, continuedl=False):
26.         """
27.         Descarga el video de la url indicada, en el outpath indicado.
28.         format_asp es opcional.
29.         --format_asp = best, worst, bestaudio, worstaudio, bestvideo, worstvideo
30.         --extension = mp4, flv, webm, 3gp, m4a, mp3, ogg, aac, wav
31.         --continuedl = True|False (para continuar la descarga en donde la dejamos.
32.         """
33.         parameters = {"outtmpl": "%(title)s_%(format_id)s.%(ext)s"}
34.      
35.         if format_asp:
36.             parameters["format"] = format_asp
37.     
38.         if continuedl:
39.             parameters["continuedl"] = True
40. 
    
41.         ydl = youtube_dl.YoutubeDL( parameters )
42.         ydl.extract_info(url, True)
43.  
44.  
45.     def ContinueDownloader(self, url, outpath, format_asp=None):
46.         """
47.         Continua la descarga del video que no termino de descargarse.
48.         """
49.         return self.Downloader(url, outpath, format_asp, True)
50.      
51.  
52.     def ExtractInfo(self, url):
53.         """
54.         Obtiene la informacion de la url. retorna info
55.         """
56.         ydl = youtube_dl.YoutubeDL()
57.         return ydl.extract_info(url, False)
58.  
59.  
60.     def GetFormats(self, info):
61.         """
62.         Obtiene los distintos formatos disponibles del video 
63.         con sus size. [ (format_id, format, size) ]
64.         """
65.         try:
66.             lista = info["formats"]
67.         except KeyError:
68.             return [ 
69.                     (info.get("format_id", ""), 
70.                      info.get("format", ""), 
71.                      info.get("filesize", 0), 
72.                      info.get("ext", "") )
73.                      ]
74.         formats = []
75.         for dic in lista:
76.             formats.append( (
77.                              dic.get("format_id", ""), 
78.                              dic.get("format", ""), 
79.                              dic.get("filesize", 0), 
80.                              dic.get("ext", "") 
81.                              ) )
82.         return formats
83.  
84.  
85.     def GetTitle(self, info):
86.         """
87.         Obtiene el titulo del video
88.         """
89.         return info.get("title", "Sin_titulo")
90.  
91.  
92.     def GetDescription(self, info):
93.         """
94.         Obtiene la descripcion del video
95.         """
96.         return info.get("description", "")
97.          
98.          
99.     def Thumbnails(self, info):
100.         """Obtiene el link de la imagen que se muestra en el video.
101.         """
102.         return info.get("thumbnail", None)
103.       
104. 
     
105.     def WriteFile(self, data, filename):
106.         """Escribe en un archivo en disco.
107.         """
108.         f = file(filename, "wb")
109.         pickle.dump(data, f)
110.         f.close()
111. 
     
112.   
113.     def ReadFile(self, filename):
114.         """Lee un archivo del disco.
115.         """
116.         try:
117.             f = file(filename, "rb")
118.             data = pickle.load(f)
119.             f.close()
120.         except BaseException as e:
121.             self.WriteFile({}, filename)
122.             data = {}
123.             print(e)
124.         return data
125.      
126.      
127. 
     
128. vd = VideoDownloader()