Write playlist description to file (Closes #36)

:ci skip dl

diff --git a/youtube_dlc/YoutubeDL.py b/youtube_dlc/YoutubeDL.py
index bf57d4765c52d7367bcc777a5272b24769d5b6a6..0dd7374710ad74d2eb19784607db1730cf7d5ba9 100644 (file)
--- a/youtube_dlc/YoutubeDL.py
+++ b/youtube_dlc/YoutubeDL.py
@@ -1101,10 +1101,15 @@ def __process_playlist(self, ie_result, download):
         playlist = ie_result.get('title') or ie_result.get('id')
         self.to_screen('[download] Downloading playlist: %s' % playlist)
 
+        def ensure_dir_exists(path):
+            return make_dir(path, self.report_error)
+
         if self.params.get('writeinfojson', False):
             infofn = replace_extension(
                 self.prepare_filepath(self.prepare_filename(ie_result), 'infojson'),
                 'info.json', ie_result.get('ext'))
+            if not ensure_dir_exists(encodeFilename(infofn)):
+                return
             if self.params.get('overwrites', True) and os.path.exists(encodeFilename(infofn)):
                 self.to_screen('[info] Playlist description metadata is already present')
             else:
@@ -1116,6 +1121,25 @@ def __process_playlist(self, ie_result, download):
                 except (OSError, IOError):
                     self.report_error('Cannot write playlist description metadata to JSON file ' + infofn)
 
+        if self.params.get('writedescription', False):
+            descfn = replace_extension(
+                self.prepare_filepath(self.prepare_filename(ie_result), 'description'),
+                'description', ie_result.get('ext'))
+            if not ensure_dir_exists(encodeFilename(descfn)):
+                return
+            if not self.params.get('overwrites', True) and os.path.exists(encodeFilename(descfn)):
+                self.to_screen('[info] Playlist description is already present')
+            elif ie_result.get('description') is None:
+                self.report_warning('There\'s no playlist description to write.')
+            else:
+                try:
+                    self.to_screen('[info] Writing playlist description to: ' + descfn)
+                    with io.open(encodeFilename(descfn), 'w', encoding='utf-8') as descfile:
+                        descfile.write(ie_result['description'])
+                except (OSError, IOError):
+                    self.report_error('Cannot write playlist description file ' + descfn)
+                    return
+
         playlist_results = []
 
         playliststart = self.params.get('playliststart', 1) - 1