Ydl class creation

main.py

@@ -0,0 +1,90 @@
+from __future__ import unicode_literals
+import youtube_dl
+from feedgen.feed import FeedGenerator
+import os
+import sys
+
+class ydl_object:
+    def __init__(self,args):
+        self.args = args
+        self.infos = self.fetch_info()
+        print(self.args)
+
+    def fetch_info(self):
+        print('Fetching Youtube Link informations...')
+        return youtube_dl.YoutubeDL({'quiet':True,'ignoreerrors':True}).extract_info(self.args.youtube_url, False)
+
+    def create_feed(self,results):
+        """
+        replace results by info
+        #results keys
+        _type entries id title uploader uploader_id uploader_url extractor webpage_url webpage_url_basename extractor_key
+        """
+        fg = FeedGenerator()
+        fg.load_extension('podcast')
+        fg.podcast.itunes_category('Podcasting')
+
+        fg.title(results['title'])
+        fg.description('none')
+        fg.link(href=self.args.url,rel='self')
+
+        for item in results['entries']:
+            """
+            #results['entries'] keys
+            id uploader uploader_id uploader_url channel_id channel_url upload_date license creator title alt_title thumbnail description categories tags subtitles automatic_captions duration
+            """
+            fe = fg.add_entry()
+            fe.id(item['id'])
+            fe.title(item['title'])
+            fe.description(item['description'])
+            item_full_path = self.args.url +'/'+results['title']+'/'+item['title']+'.mp3'
+            fe.enclosure(item_full_path,str(item['duration']),'audio/mpeg')
+
+        fg.rss_str(pretty=True)
+        # create folder of feed if it doesn't exists
+        os.makedirs(self.args.dir+'/'+results['title'], exist_ok=True)
+        fg.rss_file(self.args.dir+'/'+results['title']+'/podcast.xml')
+
+        return True
+
+    def my_hook(d):
+        if d['status'] == 'finished':
+            print('Done downloading, now converting ...')
+
+    def yt_download(ydl_opts):
+        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+            print('Downloading Videos...')
+            print('Depending on number of files, it can take a while...')
+            results = ydl.extract_info(self.args.youtube_url, not self.args.simulate)
+            print('Done')
+
+    def create_opts(infos,type):
+        if type=='audio':
+            ydl_opts = {
+                'quiet':self.args.quiet,
+                'format': 'bestaudio/best',
+                'ignoreerrors': True,
+                'postprocessors': [{
+                    'key': 'FFmpegExtractAudio',
+                    'preferredcodec': 'mp3',
+                    'preferredquality': '192',
+                }],
+                'progress_hooks': [my_hook],
+            }
+        elif type=='video':
+            ydl_opts= {}
+
+
+        # if youtube url  is a list, then download archive 
+        print('Fetching Url properties...')
+        if 'entries' in infos:
+            ydl_opts['download_archive']= self.args.dir+'/archive.txt'
+            ydl_opts['outtmpl']= self.args.dir+'/%(playlist_title)s/%(title)s.%(ext)s'
+        else:
+            ydl_opts['download_archive']= self.args.dir+'/archive.txt'
+            ydl_opts['outtmpl']= self.args.dir+'/%(title)s.%(ext)s'
+        print('Done')
+
+        return ydl_opts
+
+

ydl.py

@@ -4,6 +4,7 @@ from feedgen.feed import FeedGenerator
 import os
 import sys
 import argparse
+import main
 
 #TODO:
 # offer download video option
@@ -21,6 +22,7 @@ parser.add_argument("-url", help="Define base url for podcasts, default value:'h
 parser.add_argument("-yturl","--youtube_url", help="Define youtube url to fetch", default ="https://www.youtube.com/watch?v=xJO5GstqTSY&list=PLxzM9a5lhAumFRpcigmGY1QLDYxb4-P2B")
 args = parser.parse_args()
 
+ydlo = main.ydl_object(args)
 
 def fetch_info(url_yt):
     print('Fetching Youtube Link informations...')
@@ -102,7 +104,7 @@ def create_opts(infos,type):
 
 if __name__ == "__main__":
     print('Downloads folder name: {}'.format(args.dir))
-    infos = fetch_info(args.youtube_url)
+    infos = ydlo.infos
 
     # if youtube url linked to a list
     if 'entries' in infos: