gen_operator_list.py

#!/usr/bin/python
# encoding: utf-8
# Copyright (C) 2019-2020 ArvVoid (https://github.com/arvvoid)
# SPDX-License-Identifier: GPL-2.0-or-later

from __future__ import absolute_import, division

#  simple utility script to generate provider list for all Hbo Go Eu
#  usefull for wiki and documentation and get the
#  "big picture" and have a panorama of all operators.
#  will generate marked-up list for documentation

import requests
import os.path
import codecs
from datetime import date

from hbogolib.constants import HbogoConstants


def replacetextbetween(original, delimeter_a, delimter_b, newtext):
    leadingtext = original.split(delimeter_a)[0]
    trailingtext = original.split(delimter_b)[1]

    return leadingtext + delimeter_a + newtext + delimter_b + trailingtext


info_string = "|COUNTRY|OPERATOR|LOGIN TYPE|API" + os.linesep
info_string = info_string + "|-------|--------|----------|---|" + os.linesep
print('HBO Go EU Operator list generator V2.0: ')
print('')
print('')
for countrie in HbogoConstants.countries:
    if countrie[6] == HbogoConstants.HANDLER_EU:
        url_basic = 'https://api.ugw.hbogo.eu/v3.0/Operators/' + str(countrie[3]) + '/JSON/' + str(countrie[4]) + '/COMP'
        url_operators = 'https://' + str(countrie[2]) + 'gwapi.hbogo.eu/v2.1/Operators/json/' + str(countrie[3]) + '/COMP'
        print('Processing operators for: ' + str(countrie[0]) + '...')
        json_web_operators = requests.get(url_basic).json()
        for operator in json_web_operators['Items']:
            print('DIRECT ' + operator['Type'] + ': ' + operator['Name'])
            info_string = info_string + "|" + str(countrie[0]) + "|" + operator['Name'] + "|" + 'DIRECT (' + operator['Type'] + ")|EU|" + os.linesep
        json_operators = requests.get(url_operators).json()
        for operator in json_operators['Items']:
            strtype = "AFFILIATE GATEWAY"
            if len(operator['RedirectionUrl']) > 0:
                strtype = "AFFILIATE OAuth External (Redirect)"
            print(strtype + ': ' + operator['Name'])
            info_string = info_string + "|" + str(countrie[0]) + "|" + operator['Name'] + "|" + strtype + "|EU|" + os.linesep
    if countrie[6] == HbogoConstants.HANDLER_NORDIC:  # or HANDLER_SPAIN (same value)
        print('Processing operators for: ' + str(countrie[0]) + '...')
        info_string = info_string + "|" + str(countrie[0]) + "|" + "HBO Subscription " + str(countrie[0]) + "|DIRECT|NORDIC/SPAIN|" + os.linesep

print("")
print("Preparing output...")

output = os.linesep + "Last update: " + str(date.today()) + os.linesep + os.linesep + info_string + os.linesep

if os.path.isfile('../hgowiki/Regional-support.md'):
    file_r = codecs.open('../hgowiki/Regional-support.md', encoding='utf-8')
    original = file_r.read()
    file_r.close()
    output = replacetextbetween(original, "<!--- BEGIN GENERATED --->", "<!--- END GENERATED --->", output)
    file = codecs.open("../hgowiki/Regional-support.md", "w", "utf-8")
    file.write(output)
    file.close()
    print("Done!")
else:
    print("operators.md...")
    file = codecs.open("operators.md", "w", "utf-8")
    file.write(output)
    file.close()
    print("Done!")