MythUtil-Channel-videosource

#!/usr/bin/python3

import argparse
import sys
import json
import re
import natsort
import requests
import requests.auth

class MythTVServices():
    def __init__(self, host=None, port=None, username=None, password=None):
        if host is None:
            host = 'localhost'
        self.host = host
        if port is None:
            port = 6544
        self.port = port
        self.session = requests.Session()
        if username and password:
            self.session.auth = requests.auth.HTTPDigestAuth(username, password)
            self.request(service='Myth', api='version')

    def request(self, service=None, api=None, data={}, method=None, stream=False):
        version = '0.28'
        headers = {'User-Agent':'{} Python Services API Client'.format(version),
                   'Accept':'application/json',
                   'Accept-Encoding':'gzip,deflate'}
        if api is None:
            raise ValueError('api must be specified')
        url = 'http://{}:{}/{}/{}'.format(self.host, self.port, service, api)
        if method is None:
            if bool(data):
                method = 'post'
            else:
                method = 'get'
        if method == 'get':
            response = self.session.get(url, headers=headers, params=data, stream=stream)
        elif method == 'post':
            response = self.session.post(url, headers=headers, data=data, stream=stream)
        else:
            raise ValueError('method is not post or get: {}'.format(method))
        response.raise_for_status()
        if stream:
            response.raw.decode_content = True
            return response.raw
        else:
            return response.json()

    def Capture(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Capture', api=api, data=data, method=method, stream=stream)
    def Channel(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Channel', api=api, data=data, method=method, stream=stream)
    def Content(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Content', api=api, data=data, method=method, stream=stream)
    def Dvr(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Dvr', api=api, data=data, method=method, stream=stream)
    def Frontend(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Frontend', api=api, data=data, method=method, stream=stream)
    def Guide(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Guide', api=api, data=data, method=method, stream=stream)
    def Myth(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Myth', api=api, data=data, method=method, stream=stream)
    def Video(self, api=None, data={}, method=None, stream=False):
        return self.request(service='Video', api=api, data=data, method=method, stream=stream)

def TrueOrFalse(arg):
    ua = str(arg).lower()
    if ua == 'true'[:len(ua)]:
        return True
    elif ua == 'yes'[:len(ua)]:
        return True
    elif ua == '1':
        return True
    elif ua == 'false'[:len(ua)]:
        return False
    elif ua == 'no'[:len(ua)]:
        return False
    elif ua == '0':
        return False
    raise argparse.ArgumentTypeError('{} is not a valid true/false value'.format(arg))

def unsignedInt(v):
    try:
        v = int(v)
    except ValueError:
        raise argparse.ArgumentTypeError("{} is not a valid positive integer".format(v))
    if v < 0:
        raise argparse.ArgumentTypeError("{} is not a valid positive integer".format(v))
    return v

def versionTuple(v):
    return tuple(map(int, (v.split("."))))

def transformAPIElementNames(APIname, existingDict):
    transforms = \
        {
            'VideoSource':
                {
                    'Id':                     'SourceId',
                },
            'Channel':
                {
                    'SourceId':               'SourceID',
                    'MplexId':                'MplexID',
                    'ChanId':                 'ChannelID',
                    'ChanNum':                'ChannelNumber',
                    'ServiceId':              'ServiceID',
                    'ATSCMajorChan':          'ATSCMajorChannel',
                    'ATSCMinorChan':          'ATSCMinorChannel',
                    'Visible':                'visible',
                    'FrequencyId':            'FrequencyID',
                    'DefaultAuth':            'DefaultAuthority'
                }
        }
    transform = transforms.get(APIname, {})
    newDict = {}
    for key, value in existingDict.items():
        newKey = transform.get(key, key)
        if newKey is not None:
            newDict[newKey] = value
    return newDict

if __name__ == '__main__':

    parser = argparse.ArgumentParser()
    parser.add_argument('--backend', '--host', action='store', type=str, default='localhost',
                        help='the host (backend) to access.  The default is localhost.')
    parser.add_argument('--port', action='store', type=int, default=6544,
                        help='the port to connect to on on the host.  The default is 6544')
    parser.add_argument('--username', action='store', type=str, default=None,
                        help='the username to use for host authentication')
    parser.add_argument('--password', action='store', type=str, default=None,
                        help='the password to use for host authentication')
    sourcegroup = parser.add_mutually_exclusive_group(required=False)
    sourcegroup.add_argument('--videosource-name', action='store', type=str, dest='sourceName',
                             help='the video source name')
    sourcegroup.add_argument('--videosource-id', action='store', type=unsignedInt, dest='sourceId',
                             help='the video source id')
    actiongroup = parser.add_mutually_exclusive_group(required=False)
    actiongroup.add_argument('--add', action='store_true', default=False,
                             help='add a video source')
    actiongroup.add_argument('--delete', action='store_true', default=False,
                             help='delete a video source')
    actiongroup.add_argument('--update', action='store_true', default=False,
                             help='update a video source')
    parser.add_argument('--grabber', action='store', type=str,
                        help='grabber name')
    parser.add_argument('--config', action='store', type=str,
                        help='grabber configuration file')
    parser.add_argument('--use-eit', default=None, dest='useEIT', type=TrueOrFalse,
                        help='use eit for guide')
    parser.add_argument('--network-id', action='store', type=int, dest='networkId',
                        help='provider network id')
    parser.add_argument('--bouquet-id', action='store', type=unsignedInt, dest='bouquetId',
                        help='provider bouquet id')
    parser.add_argument('--region-id', action='store', type=unsignedInt, dest='regionId',
                        help='provider region id')
    parser.add_argument('--scan-frequency', action='store', type=unsignedInt, dest='scanFreq',
                        help='provider start scan frequency')
    parser.add_argument('--lcnoffset', action='store', type=unsignedInt, dest='lcnoffset',
                        help='offset for lcn')
    parser.add_argument('--frequency-table', default=None, type=str.lower, dest='freqTable',
                        choices=["default", "us-bcast", "us-cable", "us-cable-hrc",
                                 "us-cable-irc", "japan-bcast", "japan-cable", "europe-west",
                                 "europe-east", "italy", "newzealand", "australia", "ireland",
                                 "france", "china-bcast", "southafrica", "argentina",
                                 "australia-optus", "singapore", "malaysia",
                                 "israel-hot-matav"],
                        help='frequency table to use')
    parser.add_argument('--grabber-userid', action='store', type=str, dest='grabberUserid',
                        help='userid for schedulesdirect1 (internal) grabber')
    parser.add_argument('--grabber-password', action='store', type=str, dest='grabberPassword',
                        help='password for schedulesdirect1 (internal) grabber')
    parser.add_argument('--lineupid', action='store', type=str,
                        help='lineupid for schedulesdirect1 (internal) grabber')
    parser.add_argument('--new-videosource-name', action='store', type=str, dest='sourceNameNew',
                        help='new video source name for update')
    args = parser.parse_args()

    s = MythTVServices(args.backend, args.port, args.username, args.password)

    try:
        hostname = s.Myth('GetHostName')['String']
    except Exception:
        print('Unable to obtain hostname from host {}:{}'.format(args.backend, args.port))
        sys.exit(1)

    ChannelServiceVersion = versionTuple(s.Channel('version')['String'])

    # If no action, just a list of video sources
    if (not args.add) and (not args.update) and (not args.delete):
        if (args.grabber is not None) or (args.config is not None) or (args.useEIT is not None) or (args.networkId is not None) or (args.freqTable is not None) or (args.grabberUserid is not None) or (args.grabberPassword is not None) or (args.bouquetId is not None) or (args.regionId is not None) or (args.scanFreq is not None) or (args.sourceNameNew is not None) or (args.lcnoffset is not None):
            print('List requests cannot specify the grabber, config, use-eit, network-id, frequency-table, grabber-userid, grabber-password, bouquet-id, region-id, scan-frequency, lcnoffset or new-videosource-name')
            sys.exit(1)
        mythsl = s.Channel('GetVideoSourceList')['VideoSourceList']['VideoSources']
        print('Configured videosources:')
        for source in mythsl:
            if ((args.sourceName is not None) or (args.sourceId is not None)) and (int(source['Id']) != args.sourceId and source['SourceName'] != args.sourceName):
                continue
            print('  {}'.format(source['SourceName']))
            print('    ID:        {}'.format(source['Id']))
            print('    grabber:   {}'.format(source['Grabber']))
            print('    config:    {}'.format(source['ConfigPath']))
            print('    UseEIT:    {}'.format(source['UseEIT']))
            print('    FreqTable: {}'.format(source['FreqTable']))
            print('    NetworkID: {}'.format(source['NITId']))
            print('    userid:    {}'.format(source['UserId']))
            print('    password:  {}'.format(source['Password']))
            print('    lineupid:  {}'.format(source['LineupId']))
            if ChannelServiceVersion >= versionTuple("1.8"):
                print('    BouquetID: {}'.format(source['BouquetId']))
                print('    ReqionID:  {}'.format(source['RegionId']))
                print('    ScanFreq:  {}'.format(source['ScanFrequency']))
            if ChannelServiceVersion >= versionTuple("1.10"):
                print('    LCNOffset: {}'.format(source['LCNOffset']))
        sys.exit(0)

    # Try to locate source name/id if specified
    mythsl = s.Channel('GetVideoSourceList')['VideoSourceList']['VideoSources']
    sourceId = None
    sourceName = None
    for source in mythsl:
        if int(source['Id']) == args.sourceId or source['SourceName'] == args.sourceName:
            sourceId = int(source['Id'])
            sourceName = source['SourceName']
            break

    # Perform actions
    if args.add:
        if args.sourceName is None:
            print('Video source name name must be specified for an add request')
            sys.exit(1)
        if sourceId is not None:
            print('Video source already exists with that name')
            sys.exit(1)
        data = {}
        data['SourceName'] = args.sourceName
        data['Grabber'] = '/bin/true'
        data['ConfigPath'] = None
        data['LineupId'] = None
        data['UserId'] = None
        data['Password'] = None
        data['FreqTable'] = 'default'
        data['UseEIT'] = False
        data['NITId'] = -1
        data['BouquetId'] = 0
        data['RegionId'] = 0
        data['ScanFrequency'] = 0
        data['LCNOffset'] = 0
        if args.grabber is not None:
            if (ChannelServiceVersion >= versionTuple("1.8")) and (args.grabber in ['schedulesdirect1', 'datadirect']):
                print("Warning: backend version no longer supports the internal Schedules Direct grabber")
            data['Grabber'] = args.grabber
        if args.config is not None:
            data['ConfigPath'] = args.config
        if args.lineupid is not None:
            data['LineupId'] = args.lineupid
        if args.grabberUserid is not None:
            data['UserId'] = args.grabberUserid
        if args.grabberPassword is not None:
            data['Password'] = args.grabberPassword
        if args.useEIT is not None:
            data['UseEIT'] = args.useEIT
        if args.freqTable is not None:
            data['FreqTable'] = args.freqTable
        if args.networkId is not None:
            data['NITId'] = args.networkId
        if args.bouquetId is not None:
            if ChannelServiceVersion < versionTuple("1.8"):
                print("Warning: backend version does not support bouquet-id")
            data['BouquetId'] = args.bouquetId
        if args.regionId is not None:
            if ChannelServiceVersion < versionTuple("1.8"):
                print("Warning: backend version does not support region-id")
            data['RegionId'] = args.regionId
        if args.scanFreq is not None:
            if ChannelServiceVersion < versionTuple("1.8"):
                print("Warning: backend version does not support scan-frequency")
            data['ScanFrequency'] = args.scanFreq
        if args.lcnoffset is not None:
            if ChannelServiceVersion < versionTuple("1.10"):
                print("Warning: backend version does not support lcnoffset")
            data['LCNOffset'] = args.lcnoffset
        if int(s.Channel('AddVideoSource', transformAPIElementNames('VideoSource', data))['int']) > 0:
            print('Video source added')
            try:
                s.Dvr('RescheduleRecordings')
            except Exception:
                pass
            sys.exit(0)
        else:
            print('Video source addition failed')
            sys.exit(1)

    elif args.delete:
        if (args.sourceId is None) and (args.sourceName is None):
            print('No video source specified to delete')
            sys.exit(1)
        if sourceId is None:
            print('Video source was not found to delete')
            sys.exit(1)
        data = {}
        data['Id'] = source['Id']
        if bool(s.Channel('RemoveVideoSource', transformAPIElementNames('VideoSource', data))['bool']):
            print('Video source deleted')
            try:
                s.Dvr('RescheduleRecordings')
            except Exception:
                pass
            sys.exit(0)
        else:
            print('Video source deletion failed')
            sys.exit(1)

    elif args.update:
        if (args.sourceId is None) and (args.sourceName is None):
            print('No video source specified to update')
            sys.exit(1)
        if sourceId is None:
            print('Video source was not found to update')
            sys.exit(1)
        data = source.copy()
        if args.sourceNameNew is not None:
            data['SourceName'] = args.sourceNameNew
        if args.grabber is not None:
            if (ChannelServiceVersion >= versionTuple("1.8")) and (args.grabber in ['schedulesdirect1', 'datadirect']):
                print("Warning: backend version no longer supports the internal Schedules Direct grabber")
            data['Grabber'] = args.grabber
        if args.config is not None:
            data['ConfigPath'] = args.config
        if args.lineupid is not None:
            data['LineupId'] = args.lineupid
        if args.grabberUserid is not None:
            data['UserId'] = args.grabberUserid
        if args.grabberPassword is not None:
            data['Password'] = args.grabberPassword
        if args.useEIT is not None:
            data['UseEIT'] = args.useEIT
        if args.freqTable is not None:
            data['FreqTable'] = args.freqTable
        if args.networkId is not None:
            data['NITId'] = args.networkId
        if args.bouquetId is not None:
            if ChannelServiceVersion < versionTuple("1.8"):
                print("Warning: backend version does not support bouquet-id")
            data['BouquetId'] = args.bouquetId
        if args.regionId is not None:
            if ChannelServiceVersion < versionTuple("1.8"):
                print("Warning: backend version does not support region-id")
            data['RegionId'] = args.regionId
        if args.scanFreq is not None:
            if ChannelServiceVersion < versionTuple("1.8"):
                print("Warning: backend version does not support scan-frequency")
            data['ScanFrequency'] = args.scanFreq
        if args.lcnoffset is not None:
            if ChannelServiceVersion < versionTuple("1.10"):
                print("Warning: backend version does not support lcnoffset")
            data['LCNOffset'] = args.lcnoffset
        if bool(s.Channel('UpdateVideoSource', transformAPIElementNames('VideoSource', data))['bool']):
            print('Video source updated')
            try:
                s.Dvr('RescheduleRecordings')
            except Exception:
                pass
            sys.exit(0)
        else:
            print('Video source update failed')
            sys.exit(1)