From 7a45a1590b2c75904830c994b68c71275fdf4ba0 Mon Sep 17 00:00:00 2001
From: Ashish <39122144+Ashish0804@users.noreply.github.com>
Date: Wed, 25 Aug 2021 19:33:32 +0530
Subject: [PATCH] [Epicon] Add extractors (#789)

Authored by: Ashish0804
---
 yt_dlp/extractor/epicon.py     | 119 +++++++++++++++++++++++++++++++++
 yt_dlp/extractor/extractors.py |   4 ++
 2 files changed, 123 insertions(+)
 create mode 100644 yt_dlp/extractor/epicon.py

diff --git a/yt_dlp/extractor/epicon.py b/yt_dlp/extractor/epicon.py
new file mode 100644
index 000000000..b4e544d4f
--- /dev/null
+++ b/yt_dlp/extractor/epicon.py
@@ -0,0 +1,119 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+import re
+
+from .common import InfoExtractor
+from ..utils import ExtractorError
+
+
+class EpiconIE(InfoExtractor):
+    _VALID_URL = r'(?:https?://)(?:www\.)?epicon\.in/(?:documentaries|movies|tv-shows/[^/?#]+/[^/?#]+)/(?P<id>[^/?#]+)'
+    _TESTS = [{
+        'url': 'https://www.epicon.in/documentaries/air-battle-of-srinagar',
+        'info_dict': {
+            'id': 'air-battle-of-srinagar',
+            'ext': 'mp4',
+            'title': 'Air Battle of Srinagar',
+            'description': 'md5:c4de2013af9bc05ae4392e4115d518d7',
+            'thumbnail': r're:^https?://.*\.jpg$',
+        }
+    }, {
+        'url': 'https://www.epicon.in/movies/krit',
+        'info_dict': {
+            'id': 'krit',
+            'ext': 'mp4',
+            'title': 'Krit',
+            'description': 'md5:c12b35dad915d48ccff7f013c79bab4a',
+            'thumbnail': r're:^https?://.*\.jpg$',
+        }
+    }, {
+        'url': 'https://www.epicon.in/tv-shows/paapnaashini-ganga/season-1/vardaan',
+        'info_dict': {
+            'id': 'vardaan',
+            'ext': 'mp4',
+            'title': 'Paapnaashini Ganga - Season 1 - Ep 1 - VARDAAN',
+            'description': 'md5:f517058c3d0402398eefa6242f4dd6ae',
+            'thumbnail': r're:^https?://.*\.jpg$',
+        }
+    }, {
+        'url': 'https://www.epicon.in/movies/jayadev',
+        'info_dict': {
+            'id': 'jayadev',
+            'ext': 'mp4',
+            'title': 'Jayadev',
+            'description': 'md5:09e349eecd8e585a3b6466904f19df6c',
+            'thumbnail': r're:^https?://.*\.jpg$',
+        }
+    }]
+
+    def _real_extract(self, url):
+        id = self._match_id(url)
+        webpage = self._download_webpage(url, id)
+        cid = self._search_regex(r'class=\"mylist-icon\ iconclick\"\ id=\"(\d+)', webpage, 'cid')
+        headers = {'content-type': 'application/x-www-form-urlencoded; charset=UTF-8'}
+        data = f'cid={cid}&action=st&type=video'.encode()
+        data_json = self._parse_json(self._download_json('https://www.epicon.in/ajaxplayer/', id, headers=headers, data=data), id)
+
+        if not data_json['success']:
+            raise ExtractorError(data_json['message'], expected=True)
+
+        title = self._search_regex(r'setplaytitle=\"([^\"]+)', webpage, 'title')
+        description = self._og_search_description(webpage) or None
+        thumbnail = self._og_search_thumbnail(webpage) or None
+        formats = self._extract_m3u8_formats(data_json['url']['video_url'], id)
+        self._sort_formats(formats)
+
+        subtitles = {}
+        for subtitle in data_json.get('subtitles', []):
+            sub_url = subtitle.get('file')
+            if not sub_url:
+                continue
+            subtitles.setdefault(subtitle.get('lang', 'English'), []).append({
+                'url': self._proto_relative_url(sub_url),
+            })
+
+        return {
+            'id': id,
+            'formats': formats,
+            'title': title,
+            'description': description,
+            'thumbnail': thumbnail,
+            'subtitles': subtitles,
+        }
+
+
+class EpiconSeriesIE(InfoExtractor):
+    _VALID_URL = r'(?!.*season)(?:https?://)(?:www\.)?epicon\.in/tv-shows/(?P<id>[^/?#]+)'
+    _TESTS = [{
+        'url': 'https://www.epicon.in/tv-shows/1-of-something',
+        'playlist_mincount': 5,
+        'info_dict': {
+            'id': '1-of-something',
+        },
+    }, {
+        'url': 'https://www.epicon.in/tv-shows/eco-india-english',
+        'playlist_mincount': 76,
+        'info_dict': {
+            'id': 'eco-india-english',
+        },
+    }, {
+        'url': 'https://www.epicon.in/tv-shows/s/',
+        'playlist_mincount': 25,
+        'info_dict': {
+            'id': 's',
+        },
+    }, {
+        'url': 'https://www.epicon.in/tv-shows/ekaant',
+        'playlist_mincount': 38,
+        'info_dict': {
+            'id': 'ekaant',
+        },
+    }]
+
+    def _real_extract(self, url):
+        id = self._match_id(url)
+        webpage = self._download_webpage(url, id)
+        episodes = re.findall(r'ct-tray-url=\"(tv-shows/%s/[^\"]+)' % id, webpage)
+        entries = [self.url_result('https://www.epicon.in/%s' % episode, ie=EpiconIE.ie_key()) for episode in episodes]
+        return self.playlist_result(entries, playlist_id=id)
diff --git a/yt_dlp/extractor/extractors.py b/yt_dlp/extractor/extractors.py
index 99deebbcc..a1be9bdfc 100644
--- a/yt_dlp/extractor/extractors.py
+++ b/yt_dlp/extractor/extractors.py
@@ -390,6 +390,10 @@
 from .elpais import ElPaisIE
 from .embedly import EmbedlyIE
 from .engadget import EngadgetIE
+from .epicon import (
+    EpiconIE,
+    EpiconSeriesIE,
+)
 from .eporner import EpornerIE
 from .eroprofile import (
     EroProfileIE,