ari.py 3.73 KB
Newer Older
1 2 3 4 5 6 7 8 9
import requests
import sqlite3

def ari_call(config, verbs, method, payload):
    '''
    Generate a call to the ARI. Config contains host, protocol, and API base.
    Return the result.json().
    '''
    try:
10
        base_url = config['protocol'] + '://' + config['server'] + '/' + config['api_base']
11 12 13 14
    except KeyError as e:
        raise e
    try:
        full_url = '/'.join([base_url, '/'.join(verbs)])
okhin's avatar
okhin committed
15
        if method.lower() == 'del':
16 17
            call = requests.delete
        else:
okhin's avatar
okhin committed
18
            call = getattr(requests, method.lower())
19
        payload['app'] = config['app']
20
        payload['api_key'] = config['key'] + ':' + config['password']
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
        result = call(full_url, data=payload)
        result.raise_for_status()
        return result.json()
    except:
        raise

class Bridge(object):
    '''
    This is a bridge object, we use it to manage various needed calls
    '''
    def __init__(self, config, name, mode):
        self.name = name
        self.config = config
        self.mode = mode

    def status(self):
        '''
        Get the status of a specific bridge
        '''
        try:
            return ari_call(self.config, ['bridges', self.name], 'GET', {})
        except:
            raise

    def create(self):
        '''
        Create a bridge.
        '''
        try:
            return ari_call(self.config, ['bridges', self.name], 'POST', {'mode': self.mode})
        except:
            raise

    def startMoh(self, playback_class):
        '''
        Let's start a playback on this bridge
        '''
        try:
            return ari_call(self.config, ['bridges', self.name, 'moh'], 'POST', {'mohClass': playback_class})
        except:
            raise

    def addChannel(self, channel):
        '''
        We want to add a channel to the bridge
        '''
        try:
            return ari_call(self.config, ['bridges', self.name, 'addChannel'], 'POST', {'channel': channel})
        except:
            raise
okhin's avatar
okhin committed
71 72 73 74 75 76 77 78 79
    def delete(self):
        '''
        Let's delete the bridge
        '''
        try:
            return ari_call(self.config, ['bridges', self.name], 'DEL', {})
        except:
            raise

80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
class Channel(object):
    '''
    This class is used to regroup all the method to manage a Channel
    '''
    def __init__(self, config, name):
        self.config = config
        self.channel = name

    def status(self):
        '''
        Get the status of a channel
        '''
        try:
            return ari_call(self.config, ['channels', self.name], 'GET', {})
        except:
            raise

    def originate(self, endpoint):
        '''
        Originate a call to endpoint
        '''
        try:
            return ari_call(self.config, ['channels', self.name], 'POST', {'endpoint': endpoint})
        except:
            raise

    def playback(self, playback_id, media, lang='en_US'):
        '''
        Starts a playback on a Channels
        '''
        try:
            return ari_call(self.config, ['channels', self.name, 'play', playback_id], 'POST', {'media': media, 'lang': lang})
        except:
            raise
114 115 116 117 118 119 120 121 122 123

    def delete(self):
        '''
        Delete the channel (hangup)
        '''
        try:
            return ari_call(self.config, ['channels', self.name], 'DEL', {})
        except:
            raise

124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
class Playback(object):
    '''
    This object is used to manage ARI calls for playbacks.
    '''
    def __init__(self, config, name, media, lang='en_US'):
        self.config = config
        self.name = name
        self.media = media
        self.lang = lang

    def stop(self):
        '''
        Let's stop a specific playback
        '''
        try:
            return ari_call(self.config, ['playbacks', self.name], 'DEL', {})
        except:
            raise