You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
146 lines
3.9 KiB
146 lines
3.9 KiB
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import json
|
|
import argparse
|
|
import requests
|
|
|
|
__author__ = "Meliurwen"
|
|
__version__ = "0.1.1"
|
|
__license__ = "GPL3"
|
|
|
|
def get_link_m3u(source, channel, chan_dict):
|
|
|
|
headers = {
|
|
"User-Agent": ("Mozilla/5.0 (Windows NT 10.0; rv:75.0) "
|
|
"Gecko/20100101 Firefox/75.0"),
|
|
"Accept-Language": "en-US;q=0.9,en;q=0.8,en-GB;q=0.7",
|
|
"X-Requested-With": "XMLHttpRequest",
|
|
"Connection": "keep-alive",
|
|
"Referer": str(chan_dict[source][channel]["data"]["referer"]),
|
|
"cb-enabled": "enabled",
|
|
"DNT": "1",
|
|
"Pragma": "no-cache",
|
|
"Cache-Control": "no-cache"
|
|
}
|
|
|
|
content = requests.get(
|
|
chan_dict[source][channel]["url"],
|
|
headers=headers,
|
|
timeout=(3, 300)
|
|
)
|
|
|
|
content = {
|
|
"status_code": content.status_code,
|
|
"content": content.content,
|
|
"url": content.url
|
|
}
|
|
|
|
if content["status_code"] == 200:
|
|
ugly_payload = json.loads(content["content"].decode('unicode_escape')[1:-1])
|
|
print(
|
|
ugly_payload["data"]["attributes"]["streaming"]["hls"]["url"],
|
|
sep='',
|
|
end='\n',
|
|
file=sys.stdout,
|
|
flush=False
|
|
)
|
|
sys.exit(0)
|
|
else:
|
|
print(
|
|
"[ERROR] Http status code: {stat_code}".format(stat_code=content["status_code"]),
|
|
file=sys.stderr
|
|
)
|
|
sys.exit(1)
|
|
|
|
def is_supp_source(chan_dict, source):
|
|
|
|
if source not in chan_dict.keys():
|
|
print(
|
|
"[ERROR] Source not supported: {source}".format(source=source),
|
|
file=sys.stderr
|
|
)
|
|
sys.exit(1)
|
|
|
|
def is_supp_channel(chan_dict, source, channel):
|
|
|
|
if channel not in chan_dict[source].keys():
|
|
print(
|
|
"[ERROR] Channel not supported: {channel}".format(channel=channel),
|
|
file=sys.stderr
|
|
)
|
|
sys.exit(1)
|
|
|
|
def get_list(source, chan_dict):
|
|
|
|
if source:
|
|
is_supp_source(chan_dict, source)
|
|
print(*chan_dict[source].keys(), sep = "\n")
|
|
else:
|
|
print(*chan_dict.keys(), sep = "\n")
|
|
|
|
sys.exit(1)
|
|
|
|
def main(arguments):
|
|
|
|
source = arguments.source[0] if args.source else None
|
|
channel = arguments.channel[0] if args.channel else None
|
|
is_list = arguments.list if args.list else False
|
|
|
|
with open('channels.json', 'r') as myfile:
|
|
chan_dict = json.loads(myfile.read())
|
|
|
|
if source and channel:
|
|
is_supp_source(chan_dict, source)
|
|
is_supp_channel(chan_dict, source, channel)
|
|
get_link_m3u(source, channel, chan_dict)
|
|
|
|
if is_list:
|
|
get_list(source, chan_dict)
|
|
|
|
if __name__ == "__main__":
|
|
|
|
parser = argparse.ArgumentParser(
|
|
prog='IPTV Network Extract',
|
|
usage='%(prog)s [-h] [[-s SOURCE -c CHANNEL] | [-s SOURCE -l]] [-l] [--version]',
|
|
description='Extracts original m3u8 links from multiple IPTV networks.'
|
|
)
|
|
parser.add_argument(
|
|
"-s", "--source",
|
|
nargs=1,
|
|
required=False,
|
|
help="source where getting the channel"
|
|
)
|
|
parser.add_argument(
|
|
"-c", "--channel",
|
|
nargs=1,
|
|
required=False,
|
|
help="TV channel name"
|
|
)
|
|
parser.add_argument(
|
|
"-l", "--list",
|
|
required=False,
|
|
action='store_true',
|
|
help="List supported networks and channels"
|
|
)
|
|
parser.add_argument(
|
|
"--version",
|
|
action="version",
|
|
version="%(prog)s (version {version})".format(version=__version__)
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if len(sys.argv) <= 1:
|
|
parser.error("at least 1 argument is required")
|
|
|
|
if len(sys.argv) == 3 and args.channel:
|
|
parser.error("--source is required for this parameter")
|
|
|
|
if args.source and not bool(args.channel) ^ bool(args.list):
|
|
parser.error("--source requires either --channel CHANNEL or --list")
|
|
|
|
if args.list and args.channel:
|
|
parser.error("--list should either be used singularly or with --source")
|
|
|
|
main(args)
|
|
|