Replaces the parent brand of a clip

replace_parent_brand.py

#!/usr/bin/env python

from requests import get, put
import os
import argparse
import xml.etree.ElementTree as ET

parser = argparse.ArgumentParser(description='Entities and their pids.')
parser.add_argument('child_pid', type=str, help='The pid of the child entity')
parser.add_argument('parent_pid', type=str, help='The pid of the parent entity')
args = parser.parse_args()

cert = os.environ.get('CERT')
pips_base = os.environ.get('PIPS_BASE')

ET.register_namespace('pips', 'http://ns.webservices.bbc.co.uk/2006/02/pips')
ns = {'pips': 'http://ns.webservices.bbc.co.uk/2006/02/pips'}

def make_link(element_name, parent_pid):
    element = ET.Element('pips:' + element_name)
    link = ET.SubElement(element, 'pips:link')
    link.set('rel', 'pips-meta:brand')
    link.set('pid', parent_pid)
    link.set('href', pips_base + 'brand/pid.' + parent_pid + '/')
    return element

def main():
    r = get(pips_base + 'clip/pid.' + args.child_pid + '/', cert=cert)
    root = ET.XML(r.content)

    for element in ['clip_of', 'member_of_brand']:
        root[0].remove(root[0].find('pips:' + element, ns))
        root[0].insert(2, make_link(element, args.parent_pid))

    # note that this makes a bytestring not a unicode value
    data = ET.tostring(root)

    put_response = put(pips_base + 'clip/pid.' + args.child_pid + '/', cert=cert, data=data)

    print(put_response)

if __name__ == '__main__':
    main()