586 lines
18 KiB
Python
586 lines
18 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
###############################################################################
|
|
# This file holds the implementations for all the API clients.
|
|
#
|
|
# If you want to develop a new client, here are some suggestions:
|
|
# Get the fetch methods working first, then the push, then the liquidsoap notifier.
|
|
# You will probably want to create a script on your server side to automatically
|
|
# schedule a playlist one minute from the current time.
|
|
###############################################################################
|
|
|
|
import sys
|
|
import time
|
|
import urllib
|
|
import logging
|
|
from util import json
|
|
import os
|
|
from urlparse import urlparse
|
|
|
|
|
|
def api_client_factory(config):
|
|
if config["api_client"] == "airtime":
|
|
return AirTimeApiClient(config)
|
|
elif config["api_client"] == "obp":
|
|
return ObpApiClient(config)
|
|
else:
|
|
print 'API Client "'+config["api_client"]+'" not supported. Please check your config file.'
|
|
print
|
|
sys.exit()
|
|
|
|
class ApiClientInterface:
|
|
|
|
# Implementation: optional
|
|
#
|
|
# Called from: beginning of all scripts
|
|
#
|
|
# Should exit the program if this version of pypo is not compatible with
|
|
# 3rd party software.
|
|
def is_server_compatible(self, verbose = True):
|
|
pass
|
|
|
|
# Implementation: Required
|
|
#
|
|
# Called from: fetch loop
|
|
#
|
|
# This is the main method you need to implement when creating a new API client.
|
|
# start and end are for testing purposes.
|
|
# start and end are strings in the format YYYY-DD-MM-hh-mm-ss
|
|
def get_schedule(self, start=None, end=None):
|
|
return 0, []
|
|
|
|
# Implementation: Required
|
|
#
|
|
# Called from: fetch loop
|
|
#
|
|
# This downloads the media from the server.
|
|
def get_media(self, src, dst):
|
|
pass
|
|
|
|
# Implementation: optional
|
|
#
|
|
# Called from: push loop
|
|
#
|
|
# Tell server that the scheduled *playlist* has started.
|
|
def notify_scheduled_item_start_playing(self, pkey, schedule):
|
|
pass
|
|
|
|
# Implementation: optional
|
|
# You dont actually have to implement this function for the liquidsoap playout to work.
|
|
#
|
|
# Called from: pypo_notify.py
|
|
#
|
|
# This is a callback from liquidsoap, we use this to notify about the
|
|
# currently playing *song*. We get passed a JSON string which we handed to
|
|
# liquidsoap in get_liquidsoap_data().
|
|
def notify_media_item_start_playing(self, data, media_id):
|
|
pass
|
|
|
|
# Implementation: optional
|
|
# You dont actually have to implement this function for the liquidsoap playout to work.
|
|
def generate_range_dp(self):
|
|
pass
|
|
|
|
# Implementation: optional
|
|
#
|
|
# Called from: push loop
|
|
#
|
|
# Return a dict of extra info you want to pass to liquidsoap
|
|
# You will be able to use this data in update_start_playing
|
|
def get_liquidsoap_data(self, pkey, schedule):
|
|
pass
|
|
|
|
# Put here whatever tests you want to run to make sure your API is working
|
|
def test(self):
|
|
pass
|
|
|
|
|
|
#def get_media_type(self, playlist):
|
|
# nil
|
|
|
|
################################################################################
|
|
# Airtime API Client
|
|
################################################################################
|
|
|
|
class AirTimeApiClient(ApiClientInterface):
|
|
|
|
def __init__(self, config):
|
|
self.config = config
|
|
|
|
def __get_airtime_version(self, verbose = True):
|
|
logger = logging.getLogger()
|
|
url = self.config["base_url"] + self.config["api_base"] + self.config["version_url"]
|
|
url = url.replace("%%api_key%%", self.config["api_key"])
|
|
|
|
try:
|
|
logger.debug("Trying to contact %s", url)
|
|
response = urllib.urlopen(url)
|
|
data = response.read()
|
|
logger.debug("Data: %s", data)
|
|
response_json = json.read(data)
|
|
version = response_json['version']
|
|
logger.debug("Airtime Version %s detected", version)
|
|
except Exception, e:
|
|
try:
|
|
if e[1] == 401:
|
|
if (verbose):
|
|
print '#####################################'
|
|
print '# YOUR API KEY SEEMS TO BE INVALID:'
|
|
print '# ' + self.config["api_key"]
|
|
print '#####################################'
|
|
return False
|
|
except Exception, e:
|
|
pass
|
|
|
|
try:
|
|
if e[1] == 404:
|
|
if (verbose):
|
|
print '#####################################'
|
|
print '# Unable to contact the Airtime-API'
|
|
print '# ' + url
|
|
print '#####################################'
|
|
return False
|
|
except Exception, e:
|
|
pass
|
|
|
|
version = 0
|
|
logger.error("Unable to detect Airtime Version - %s, Response: %s", e, response)
|
|
|
|
return version
|
|
|
|
|
|
def test(self):
|
|
logger = logging.getLogger()
|
|
status, items = self.get_schedule('2010-01-01-00-00-00', '2011-01-01-00-00-00')
|
|
schedule = items["playlists"]
|
|
logger.debug("Number of playlists found: %s", str(len(schedule)))
|
|
count = 1
|
|
for pkey in sorted(schedule.iterkeys()):
|
|
logger.debug("Playlist #%s",str(count))
|
|
count+=1
|
|
#logger.info("found playlist at %s", pkey)
|
|
playlist = schedule[pkey]
|
|
for item in playlist["medias"]:
|
|
filename = urlparse(item["uri"])
|
|
filename = filename.query[5:]
|
|
self.get_media(item["uri"], filename)
|
|
|
|
|
|
def is_server_compatible(self, verbose = True):
|
|
version = self.__get_airtime_version(verbose)
|
|
if (version == 0 or version == False):
|
|
if (verbose):
|
|
print 'Unable to get Airtime version number.'
|
|
print
|
|
return False
|
|
elif (version[0:4] != "1.6."):
|
|
if (verbose):
|
|
print 'Airtime version: ' + str(version)
|
|
print 'pypo not compatible with this version of Airtime.'
|
|
print
|
|
return False
|
|
else:
|
|
if (verbose):
|
|
print 'Airtime version: ' + str(version)
|
|
print 'pypo is compatible with this version of Airtime.'
|
|
print
|
|
return True
|
|
|
|
|
|
def get_schedule(self, start=None, end=None):
|
|
logger = logging.getLogger()
|
|
|
|
"""
|
|
calculate start/end time range (format: YYYY-DD-MM-hh-mm-ss,YYYY-DD-MM-hh-mm-ss)
|
|
(seconds are ignored, just here for consistency)
|
|
"""
|
|
tnow = time.localtime(time.time())
|
|
if (not start):
|
|
tstart = time.localtime(time.time() - 3600 * int(self.config["cache_for"]))
|
|
start = "%04d-%02d-%02d-%02d-%02d" % (tstart[0], tstart[1], tstart[2], tstart[3], tstart[4])
|
|
|
|
if (not end):
|
|
tend = time.localtime(time.time() + 3600 * int(self.config["prepare_ahead"]))
|
|
end = "%04d-%02d-%02d-%02d-%02d" % (tend[0], tend[1], tend[2], tend[3], tend[4])
|
|
|
|
range = {}
|
|
range['start'] = start
|
|
range['end'] = end
|
|
|
|
# Construct the URL
|
|
export_url = self.config["base_url"] + self.config["api_base"] + self.config["export_url"]
|
|
#logger.debug("Exporting schedule using URL: "+export_url)
|
|
|
|
# Insert the start and end times into the URL
|
|
export_url = export_url.replace('%%api_key%%', self.config["api_key"])
|
|
export_url = export_url.replace('%%from%%', range['start'])
|
|
export_url = export_url.replace('%%to%%', range['end'])
|
|
logger.info("Fetching schedule from %s", export_url)
|
|
|
|
response = ""
|
|
status = 0
|
|
try:
|
|
response_json = urllib.urlopen(export_url).read()
|
|
#logger.debug("%s", response_json)
|
|
response = json.read(response_json)
|
|
#logger.info("export status %s", response['check'])
|
|
status = response['check']
|
|
except Exception, e:
|
|
print e
|
|
|
|
#schedule = response["playlists"]
|
|
#scheduleKeys = sorted(schedule.iterkeys())
|
|
#
|
|
## Remove all playlists that have passed current time
|
|
#try:
|
|
# tnow = time.localtime(time.time())
|
|
# str_tnow_s = "%04d-%02d-%02d-%02d-%02d-%02d" % (tnow[0], tnow[1], tnow[2], tnow[3], tnow[4], tnow[5])
|
|
# toRemove = []
|
|
# for pkey in scheduleKeys:
|
|
# if (str_tnow_s > schedule[pkey]['end']):
|
|
# toRemove.append(pkey)
|
|
# else:
|
|
# break
|
|
# #logger.debug("Remove keys: %s", toRemove)
|
|
# for index in toRemove:
|
|
# del schedule[index]
|
|
# #logger.debug("Schedule dict: %s", schedule)
|
|
#except Exception, e:
|
|
# logger.debug("'Ignore Past Playlists' feature not supported by API: %s", e)
|
|
#response["playlists"] = schedule
|
|
|
|
return status, response
|
|
|
|
|
|
def get_media(self, src, dst):
|
|
logger = logging.getLogger()
|
|
|
|
try:
|
|
src = src + "/api_key/" + self.config["api_key"]
|
|
logger.debug("try to download from %s to %s", src, dst)
|
|
# check if file exists already before downloading again
|
|
filename, headers = urllib.urlretrieve(src, dst)
|
|
|
|
logger.info("downloaded %s to %s", src, dst)
|
|
except Exception, e:
|
|
logger.error("%s", e)
|
|
|
|
|
|
"""
|
|
Tell server that the scheduled *playlist* has started.
|
|
"""
|
|
def notify_scheduled_item_start_playing(self, pkey, schedule):
|
|
logger = logging.getLogger()
|
|
playlist = schedule[pkey]
|
|
schedule_id = playlist["schedule_id"]
|
|
url = self.config["base_url"] + self.config["api_base"] + self.config["update_item_url"]
|
|
url = url.replace("%%schedule_id%%", str(schedule_id))
|
|
url = url.replace("%%api_key%%", self.config["api_key"])
|
|
logger.debug(url)
|
|
|
|
try:
|
|
response = urllib.urlopen(url)
|
|
response = json.read(response.read())
|
|
logger.info("API-Status %s", response['status'])
|
|
logger.info("API-Message %s", response['message'])
|
|
|
|
except Exception, e:
|
|
logger.critical("Unable to connect - %s", e)
|
|
|
|
return response
|
|
|
|
|
|
"""
|
|
This is a callback from liquidsoap, we use this to notify about the
|
|
currently playing *song*. We get passed a JSON string which we handed to
|
|
liquidsoap in get_liquidsoap_data().
|
|
"""
|
|
def notify_media_item_start_playing(self, data, media_id):
|
|
logger = logging.getLogger()
|
|
response = ''
|
|
if (data[0] != '{'):
|
|
return response
|
|
try:
|
|
data = json.read(data)
|
|
logger.debug(str(data))
|
|
schedule_id = data["schedule_id"]
|
|
url = self.config["base_url"] + self.config["api_base"] + self.config["update_start_playing_url"]
|
|
url = url.replace("%%media_id%%", str(media_id))
|
|
url = url.replace("%%schedule_id%%", str(schedule_id))
|
|
url = url.replace("%%api_key%%", self.config["api_key"])
|
|
logger.debug(url)
|
|
response = urllib.urlopen(url)
|
|
response = json.read(response.read())
|
|
logger.info("API-Status %s", response['status'])
|
|
logger.info("API-Message %s", response['message'])
|
|
|
|
except Exception, e:
|
|
logger.critical("Exception: %s", e)
|
|
|
|
return response
|
|
|
|
|
|
def generate_range_dp(self):
|
|
pass
|
|
#logger = logging.getLogger()
|
|
#
|
|
#url = self.api_url + 'schedule/generate_range_dp.php'
|
|
#
|
|
#try:
|
|
# response = urllib.urlopen(url, self.api_auth)
|
|
# response = json.read(response.read())
|
|
# logger.debug("Trying to contact %s", url)
|
|
# logger.info("API-Status %s", response['status'])
|
|
# logger.info("API-Message %s", response['message'])
|
|
#
|
|
#except Exception, e:
|
|
# print e
|
|
# api_status = False
|
|
# logger.critical("Unable to handle the request - %s", e)
|
|
#
|
|
#return response
|
|
|
|
def get_liquidsoap_data(self, pkey, schedule):
|
|
logger = logging.getLogger()
|
|
playlist = schedule[pkey]
|
|
data = dict()
|
|
try:
|
|
data["schedule_id"] = playlist['id']
|
|
except Exception, e:
|
|
data["schedule_id"] = 0
|
|
data = json.write(data)
|
|
return data
|
|
|
|
|
|
|
|
################################################################################
|
|
# OpenBroadcast API Client
|
|
################################################################################
|
|
# Also check out the php counterpart that handles the api requests:
|
|
# https://lab.digris.ch/svn/elgg/trunk/unstable/mod/medialibrary/application/controllers/api/pypo.php
|
|
|
|
OBP_MIN_VERSION = 2010100101 # required obp version
|
|
|
|
class ObpApiClient():
|
|
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self.api_auth = urllib.urlencode({'api_key': self.config["api_key"]})
|
|
|
|
def is_server_compatible(self, verbose = True):
|
|
obp_version = self.get_obp_version()
|
|
|
|
if obp_version == 0:
|
|
if (verbose):
|
|
print '#################################################'
|
|
print 'Unable to get OBP version. Is OBP up and running?'
|
|
print '#################################################'
|
|
print
|
|
return False
|
|
elif obp_version < OBP_MIN_VERSION:
|
|
if (verbose):
|
|
print 'OBP version: ' + str(obp_version)
|
|
print 'OBP min-version: ' + str(OBP_MIN_VERSION)
|
|
print 'pypo not compatible with this version of OBP'
|
|
print
|
|
return False
|
|
else:
|
|
if (verbose):
|
|
print 'OBP API: ' + str(API_BASE)
|
|
print 'OBP version: ' + str(obp_version)
|
|
print 'OBP min-version: ' + str(OBP_MIN_VERSION)
|
|
print 'pypo is compatible with this version of OBP'
|
|
print
|
|
return True
|
|
|
|
|
|
def get_obp_version(self):
|
|
logger = logging.getLogger()
|
|
|
|
# lookup OBP version
|
|
url = self.config["base_url"] + self.config["api_base"]+ self.config["version_url"]
|
|
|
|
try:
|
|
logger.debug("Trying to contact %s", url)
|
|
response = urllib.urlopen(url, self.api_auth)
|
|
response_json = json.read(response.read())
|
|
obp_version = int(response_json['version'])
|
|
logger.debug("OBP Version %s detected", obp_version)
|
|
|
|
except Exception, e:
|
|
try:
|
|
if e[1] == 401:
|
|
print '#####################################'
|
|
print '# YOUR API KEY SEEMS TO BE INVALID'
|
|
print '# ' + self.config["api_auth"]
|
|
print '#####################################'
|
|
sys.exit()
|
|
|
|
except Exception, e:
|
|
pass
|
|
|
|
try:
|
|
if e[1] == 404:
|
|
print '#####################################'
|
|
print '# Unable to contact the OBP-API'
|
|
print '# ' + url
|
|
print '#####################################'
|
|
sys.exit()
|
|
|
|
except Exception, e:
|
|
pass
|
|
|
|
obp_version = 0
|
|
logger.error("Unable to detect OBP Version - %s", e)
|
|
|
|
return obp_version
|
|
|
|
|
|
def get_schedule(self, start=None, end=None):
|
|
logger = logging.getLogger()
|
|
|
|
"""
|
|
calculate start/end time range (format: YYYY-DD-MM-hh-mm-ss,YYYY-DD-MM-hh-mm-ss)
|
|
(seconds are ignored, just here for consistency)
|
|
"""
|
|
tnow = time.localtime(time.time())
|
|
if (not start):
|
|
tstart = time.localtime(time.time() - 3600 * int(self.config["cache_for"]))
|
|
start = "%04d-%02d-%02d-%02d-%02d" % (tstart[0], tstart[1], tstart[2], tstart[3], tstart[4])
|
|
|
|
if (not end):
|
|
tend = time.localtime(time.time() + 3600 * int(self.config["prepare_ahead"]))
|
|
end = "%04d-%02d-%02d-%02d-%02d" % (tend[0], tend[1], tend[2], tend[3], tend[4])
|
|
|
|
range = {}
|
|
range['start'] = start
|
|
range['end'] = end
|
|
|
|
# Construct the URL
|
|
export_url = self.config["base_url"] + self.config["api_base"] + self.config["export_url"]
|
|
|
|
# Insert the start and end times into the URL
|
|
export_url = export_url.replace('%%api_key%%', self.config["api_key"])
|
|
export_url = export_url.replace('%%from%%', range['start'])
|
|
export_url = export_url.replace('%%to%%', range['end'])
|
|
logger.info("export from %s", export_url)
|
|
|
|
response = ""
|
|
status = 0
|
|
try:
|
|
response_json = urllib.urlopen(export_url).read()
|
|
logger.debug("%s", response_json)
|
|
response = json.read(response_json)
|
|
logger.info("export status %s", response['check'])
|
|
status = response['check']
|
|
except Exception, e:
|
|
print e
|
|
|
|
return status, response
|
|
|
|
|
|
def get_media(self, src, dest):
|
|
try:
|
|
print '** urllib auth with: ',
|
|
print self.api_auth
|
|
urllib.urlretrieve(src, dst, False, self.api_auth)
|
|
logger.info("downloaded %s to %s", src, dst)
|
|
except Exception, e:
|
|
logger.error("%s", e)
|
|
|
|
|
|
"""
|
|
Tell server that the scheduled *playlist* has started.
|
|
"""
|
|
def notify_scheduled_item_start_playing(self, pkey, schedule):
|
|
#def update_scheduled_item(self, item_id, value):
|
|
logger = logging.getLogger()
|
|
url = self.config["base_url"] + self.config["api_base"] + self.config["update_item_url"]
|
|
url = url.replace("%%item_id%%", str(schedule[pkey]["id"]))
|
|
url = url.replace("%%played%%", "1")
|
|
|
|
try:
|
|
response = urllib.urlopen(url, self.api_auth)
|
|
response = json.read(response.read())
|
|
logger.info("API-Status %s", response['status'])
|
|
logger.info("API-Message %s", response['message'])
|
|
|
|
except Exception, e:
|
|
print e
|
|
api_status = False
|
|
logger.critical("Unable to connect to the OBP API - %s", e)
|
|
|
|
return response
|
|
|
|
"""
|
|
This is a callback from liquidsoap, we use this to notify about the
|
|
currently playing *song*. We get passed a JSON string which we handed to
|
|
liquidsoap in get_liquidsoap_data().
|
|
"""
|
|
def notify_media_item_start_playing(self, data, media_id):
|
|
# def update_start_playing(self, playlist_type, export_source, media_id, playlist_id, transmission_id):
|
|
logger = logging.getLogger()
|
|
playlist_type = data["playlist_type"]
|
|
export_source = data["export_source"]
|
|
playlist_id = data["playlist_id"]
|
|
transmission_id = data["transmission_id"]
|
|
|
|
url = self.config["base_url"] + self.config["api_base"] + self.config["update_start_playing_url"]
|
|
url = url.replace("%%playlist_type%%", str(playlist_type))
|
|
url = url.replace("%%export_source%%", str(export_source))
|
|
url = url.replace("%%media_id%%", str(media_id))
|
|
url = url.replace("%%playlist_id%%", str(playlist_id))
|
|
url = url.replace("%%transmission_id%%", str(transmission_id))
|
|
print url
|
|
|
|
try:
|
|
response = urllib.urlopen(url, self.api_auth)
|
|
response = json.read(response.read())
|
|
logger.info("API-Status %s", response['status'])
|
|
logger.info("API-Message %s", response['message'])
|
|
logger.info("TXT %s", response['str_dls'])
|
|
|
|
except Exception, e:
|
|
print e
|
|
api_status = False
|
|
logger.critical("Unable to connect to the OBP API - %s", e)
|
|
|
|
return response
|
|
|
|
|
|
def generate_range_dp(self):
|
|
logger = logging.getLogger()
|
|
|
|
url = self.config["base_url"] + self.config["api_base"] + self.config["generate_range_url"]
|
|
|
|
try:
|
|
response = urllib.urlopen(url, self.api_auth)
|
|
response = json.read(response.read())
|
|
logger.debug("Trying to contact %s", url)
|
|
logger.info("API-Status %s", response['status'])
|
|
logger.info("API-Message %s", response['message'])
|
|
|
|
except Exception, e:
|
|
print e
|
|
api_status = False
|
|
logger.critical("Unable to handle the OBP API request - %s", e)
|
|
|
|
return response
|
|
|
|
def get_liquidsoap_data(self, pkey, schedule):
|
|
playlist = schedule[pkey]
|
|
data = dict()
|
|
data["ptype"] = playlist['subtype']
|
|
try:
|
|
data["user_id"] = playlist['user_id']
|
|
data["playlist_id"] = playlist['id']
|
|
data["transmission_id"] = playlist['schedule_id']
|
|
except Exception, e:
|
|
data["playlist_id"] = 0
|
|
data["user_id"] = 0
|
|
data["transmission_id"] = 0
|
|
data = json.write(data)
|
|
return data
|
|
|