From: Sergey M․ <dstftw@gmail.com>
Date: Sat, 26 Dec 2020 15:58:27 +0000 (+0700)
Subject: [instagram] Fix comment count extraction
X-Git-Url: http://git.oshgnacknak.de/?a=commitdiff_plain;h=12053450dceda14ec63e3e721c72731f2cc40b34;p=youtube-dl

[instagram] Fix comment count extraction
---

diff --git a/youtube_dl/extractor/instagram.py b/youtube_dl/extractor/instagram.py
index e15508fca..07c8df7a6 100644
--- a/youtube_dl/extractor/instagram.py
+++ b/youtube_dl/extractor/instagram.py
@@ -164,12 +164,18 @@ class InstagramIE(InfoExtractor):
             uploader = media.get('owner', {}).get('full_name')
             uploader_id = media.get('owner', {}).get('username')
 
-            def get_count(key, kind):
-                return int_or_none(try_get(
-                    media, (lambda x: x['edge_media_%s' % key]['count'],
-                            lambda x: x['%ss' % kind]['count'])))
+            def get_count(keys, kind):
+                if not isinstance(keys, (list, tuple)):
+                    keys = [keys]
+                for key in keys:
+                    count = int_or_none(try_get(
+                        media, (lambda x: x['edge_media_%s' % key]['count'],
+                                lambda x: x['%ss' % kind]['count'])))
+                    if count is not None:
+                        return count
             like_count = get_count('preview_like', 'like')
-            comment_count = get_count('to_comment', 'comment')
+            comment_count = get_count(
+                ('preview_comment', 'to_comment', 'to_parent_comment'), 'comment')
 
             comments = [{
                 'author': comment.get('user', {}).get('username'),