initial implementation
This commit is contained in:
MaZderMind
parent
2ec14ca175
commit
bf372323f6
3 changed files with 199 additions and 0 deletions
153
c3t_rpc_client.py
Normal file
153
c3t_rpc_client.py
Normal file
|
|
@ -0,0 +1,153 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
# Copyright (C) 2014 derpeter
|
||||||
|
# derpeter@berlin.ccc.de
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
import xmlrpc.client
|
||||||
|
import hashlib
|
||||||
|
import hmac
|
||||||
|
import socket
|
||||||
|
import urllib
|
||||||
|
import xml
|
||||||
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
## client constructor #####
|
||||||
|
# group: worker group
|
||||||
|
# secret: client secret
|
||||||
|
# host: client hostname (will be taken from local host if set to None)
|
||||||
|
# url: tracker url (without the rpc)
|
||||||
|
# method: method to access
|
||||||
|
############################
|
||||||
|
def C3TClient(url, method, group, host, secret, args):
|
||||||
|
url = url + "rpc"
|
||||||
|
# if host == None:
|
||||||
|
# host = socket.getfqdn()
|
||||||
|
|
||||||
|
#####################
|
||||||
|
# generate signature
|
||||||
|
#####################
|
||||||
|
# assemble static part of signature arguments
|
||||||
|
# 1. URL 2. method name 3. worker group token 4. hostname
|
||||||
|
sig_args = url + "&" + method + "&" + group + "&" + host + "&"
|
||||||
|
|
||||||
|
#### add method args
|
||||||
|
if len(args) > 0:
|
||||||
|
i = 0
|
||||||
|
while i < len(args):
|
||||||
|
sig_args = str(sig_args) + str(args[i])
|
||||||
|
if i < (len(args) -1):
|
||||||
|
sig_args = sig_args + "&"
|
||||||
|
i = i + 1
|
||||||
|
|
||||||
|
##### quote URL and generate the hash
|
||||||
|
#hash signature
|
||||||
|
sig_args_enc = urllib.parse.quote_plus(sig_args)
|
||||||
|
|
||||||
|
#### generate the hmac hash with key
|
||||||
|
hash = hmac.new(bytes(secret, 'utf-8'), bytes(sig_args_enc, 'utf-8'), hashlib.sha256)
|
||||||
|
|
||||||
|
#### add signature as last parameter to the arg list
|
||||||
|
args.append(hash.hexdigest())
|
||||||
|
|
||||||
|
#### create xmlrpc client
|
||||||
|
print('creating XML RPC proxy: ' + url + "?group=" + group + "&hostname=" + host)
|
||||||
|
try:
|
||||||
|
proxy = xmlrpc.client.ServerProxy(url + "?group=" + group + "&hostname=" + host);
|
||||||
|
except xmlrpc.client.Fault as err:
|
||||||
|
print("A fault occurred")
|
||||||
|
print("Fault code: %d" % err.faultCode)
|
||||||
|
print("Fault string: %s" % err.faultString)
|
||||||
|
sys.exit()
|
||||||
|
except xmlrpc.client.ProtocolError as err:
|
||||||
|
print("A protocol error occurred")
|
||||||
|
print("URL: %s" % err.url)
|
||||||
|
print("HTTP/HTTPS headers: %s" % err.headers)
|
||||||
|
print("Error code: %d" % err.errcode)
|
||||||
|
print("Error message: %s" % err.errmsg)
|
||||||
|
sys.exit()
|
||||||
|
except socket.gaierror as err:
|
||||||
|
print("A socket error occurred")
|
||||||
|
print(err)
|
||||||
|
sys.exit()
|
||||||
|
except xmlrpc.client.ProtocolError as err:
|
||||||
|
print("A Protocol occurred")
|
||||||
|
print(err)
|
||||||
|
sys.exit()
|
||||||
|
|
||||||
|
#### call the given method with args
|
||||||
|
try:
|
||||||
|
print(method + str(args))
|
||||||
|
result = getattr(proxy,method)(*args)
|
||||||
|
except xml.parsers.expat.ExpatError as err:
|
||||||
|
print("A expat err occured")
|
||||||
|
print(err)
|
||||||
|
sys.exit()
|
||||||
|
except xmlrpc.client.Fault as err:
|
||||||
|
print("A fault occurred")
|
||||||
|
print("Fault code: %d" % err.faultCode)
|
||||||
|
print("Fault string: %s" % err.faultString)
|
||||||
|
sys.exit()
|
||||||
|
except xmlrpc.client.ProtocolError as err:
|
||||||
|
print("A protocol error occurred")
|
||||||
|
print("URL: %s" % err.url)
|
||||||
|
print("HTTP/HTTPS headers: %s" % err.headers)
|
||||||
|
print("Error code: %d" % err.errcode)
|
||||||
|
print("Error message: %s" % err.errmsg)
|
||||||
|
sys.exit()
|
||||||
|
#### return the result
|
||||||
|
return result
|
||||||
|
|
||||||
|
def open_rpc(method, args, url, group, host, secret):
|
||||||
|
result = C3TClient(url, method, group, host, secret, args)
|
||||||
|
return result
|
||||||
|
|
||||||
|
### get Tracker Version
|
||||||
|
def getVersion():
|
||||||
|
tmp_args = ["1"];
|
||||||
|
return str(open_rpc("C3TT.getVersion",tmp_args))
|
||||||
|
|
||||||
|
### check for new ticket on tracker an get assignement
|
||||||
|
def assignNextUnassignedForState(from_state, to_state, url, group, host, secret):
|
||||||
|
tmp_args = [from_state, to_state]
|
||||||
|
xml = open_rpc("C3TT.assignNextUnassignedForState",tmp_args,url, group, host, secret)
|
||||||
|
# if get no xml there seems to be no ticket for this job
|
||||||
|
if xml == False:
|
||||||
|
return False
|
||||||
|
else:
|
||||||
|
print(xml)
|
||||||
|
return xml['id']
|
||||||
|
|
||||||
|
### get ticket properties
|
||||||
|
def getTicketProperties(id ,url, group, host, secret):
|
||||||
|
tmp_args = [id]
|
||||||
|
xml = open_rpc("C3TT.getTicketProperties",tmp_args, url, group, host, secret)
|
||||||
|
if xml == False:
|
||||||
|
print("no xml in answer")
|
||||||
|
return None
|
||||||
|
else:
|
||||||
|
print(xml)
|
||||||
|
return xml
|
||||||
|
|
||||||
|
### set Ticket status on done
|
||||||
|
def setTicketDone(id, url, group, host, secret):
|
||||||
|
tmp_args = [id]
|
||||||
|
xml = open_rpc("C3TT.setTicketDone", tmp_args , url, group, host, secret)
|
||||||
|
print(xml)
|
||||||
|
|
||||||
|
### set ticket status on failed an supply a error text
|
||||||
|
def setTicketFailed(id,error , url, group, host, secret):
|
||||||
|
tmp_args = [id, error]
|
||||||
|
xml = open_rpc("C3TT.setTicketFailed", tmp_args , url, group, host, secret)
|
||||||
39
script-Z-preroll-generator.py
Executable file
39
script-Z-preroll-generator.py
Executable file
|
|
@ -0,0 +1,39 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
import xmlrpc.client
|
||||||
|
import socket
|
||||||
|
import time
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import c3t_rpc_client as rpc
|
||||||
|
|
||||||
|
print("C3TT preroll generator")
|
||||||
|
|
||||||
|
if os.environ.get('CRS_TOKEN') is None or os.environ.get('CRS_SECRET') is None:
|
||||||
|
print('CRS_TOKEN or CRS_SECRET is empty. did you source the tracker-scripts-profile?')
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
ticket_type = 'recording'
|
||||||
|
ticket_state = 'generating'
|
||||||
|
|
||||||
|
host = socket.getfqdn()
|
||||||
|
url = os.environ['CRS_TRACKER']
|
||||||
|
token = os.environ['CRS_TOKEN']
|
||||||
|
secret = os.environ['CRS_SECRET']
|
||||||
|
|
||||||
|
def generatePreroll(ticket):
|
||||||
|
print('generating preroll for', ticket)
|
||||||
|
|
||||||
|
while True:
|
||||||
|
print('Asking RPC for {0}-tickets which are ready for state {1}'.format(ticket_type, ticket_state))
|
||||||
|
|
||||||
|
ticket_id = rpc.assignNextUnassignedForState(ticket_type, ticket_state, url, token, host, secret)
|
||||||
|
if ticket_id != False:
|
||||||
|
ticket = rpc.getTicketProperties(str(ticket_id), url, token, host, secret)
|
||||||
|
generatePreroll(ticket)
|
||||||
|
|
||||||
|
else:
|
||||||
|
print('No ticket found')
|
||||||
|
|
||||||
|
print('Sleeping for 30 seconds')
|
||||||
|
time.sleep(30);
|
||||||
|
|
||||||
7
script-Z-tracker-changes.sql
Normal file
7
script-Z-tracker-changes.sql
Normal file
|
|
@ -0,0 +1,7 @@
|
||||||
|
ALTER type enum_ticket_state ADD VALUE 'generating' after 'recorded';
|
||||||
|
ALTER type enum_ticket_state ADD VALUE 'generated' after 'generating';
|
||||||
|
|
||||||
|
INSERT into tbl_ticket_state VALUES ('recording', 'generating', 0, 10, true), ('recording', 'generated', 0, 5, false);
|
||||||
|
UPDATE tbl_ticket_state SET sort = sort + 2 WHERE ticket_type = 'recording' AND sort > 4;
|
||||||
|
UPDATE tbl_ticket_state SET sort = 5 WHERE ticket_type = 'recording' AND ticket_state = 'generating';
|
||||||
|
UPDATE tbl_ticket_state SET sort = 6 WHERE ticket_type = 'recording' AND ticket_state = 'generated';
|
||||||
Loading…
Add table
Reference in a new issue