#!/usr/bin/env python

import requests
import json
import requests.packages.urllib3
from requests.packages.urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

xms_user = 'user'
xms_pass = 'pass'
xms_ip   = '1.1.1.1'
xms_addr = 'https://%s/api/json/v2/types/' % (xms_ip)
xms_cluster = 'xio-cluster-name'

def refreshSnapSet(cluster,snapfrom,snapto,snapnew,snaptype,suffix):
    try:
        payload = {'cluster-id' : cluster,
                   'to-snapshot-set-id' : snapto,
                   'snapshot-set-name' : snapnew,
                   'backup-snapshot-type' : snaptype,
                   'backup-snap-suffix' : '-' + suffix}
        if suffix == 'SSRW':
            payload['from-snapshot-set-id'] = snapfrom
        else:
            payload['from-consistency-group-id'] = snapfrom
        r = requests.post(xms_addr+'snapshots',json=payload,auth=(xms_user,xms_pass),verify=False)
        print r.status_code
    except Exception as e:
        print "Error: ",e
        return 1

def renameSnapSet(cluster,snapfrom,snapto):
    try:
        payload = {'new-name' : snapto}
        r = requests.put('%ssnapshot-sets?name=%s&cluster-name=%s'%(xms_addr,snapfrom,xms_cluster),
                        json=payload,
                        auth=(xms_user,xms_pass),
                        verify=False)
        print r.status_code
    except Exception as e:
        print "Error: ",e
        return 1

def deleteSnapSet(cluster,snapname):
    try:
        r = requests.delete('%ssnapshot-sets?name=%s&cluster-name=%s'%(xms_addr,snapname,xms_cluster),
                            auth=(xms_user,xms_pass),
                            verify=False)
        print r.status_code
    except Exception as e:
        print "Error: ",e
        return 1

cgname = 'CACHE_DRVG'
ssro   = cgname + '-SSRO'
ssroold = cgname + '-SSRO-OLD'
ssronew = cgname + '-SSRO-NEW'
ssrw   = cgname + '-SSRW'
ssrwold = cgname + '-SSRW-OLD'
ssrwnew = cgname + '-SSRW-NEW'

refreshSnapSet(xms_cluster,cgname,ssro,ssronew,'readonly','SSRO')
refreshSnapSet(xms_cluster,ssro,ssrw,ssrwnew,'regular','SSRW')
renameSnapSet(xms_cluster,ssro,ssroold)
renameSnapSet(xms_cluster,ssronew,ssro)
renameSnapSet(xms_cluster,ssrw,ssrwold)
renameSnapSet(xms_cluster,ssrwnew,ssrw)
deleteSnapSet(xms_cluster,ssroold)
deleteSnapSet(xms_cluster,ssrwold)