#!/usr/bin/python3

import json
import sys
from sh import rbd

def purge(instance):
	with open('/var/lib/ganeti/config.data') as f:
		data = json.loads(f.readline())

	instance_disks = []
	for i in data['instances']:
		i = data['instances'][i]
		if i['name'] != instance:
			continue
		instance_disks = i['disks']
		break
	
	for i in instance_disks:
		_id = data['disks'][i]['logical_id']
		_id = [i for i in _id if i != 'rbd'][0]
		print('Purging snaps from %s ..' % (_id,))
		rbd('snap', 'purge', _id)

if __name__ == '__main__':
	purge(sys.argv[1])