zstack_volume_api.py

#!/usr/bin/env python
#coding=utf-8
__author__ = 'yuanbin'
import sys
import os
import config
import time
import requests
import json
from zstack_base_demo import zstack_base_api
reload(sys)
sys.setdefaultencoding('utf-8')

class zstack_volume_api:
    def __init__(self):
        self.header = {"Content-Type": "application/json"}
        self.base = zstack_base_api()
        self.UUID = self.base.UUID

    def create_dataVolume_from_diskOffering(self, name, diskOfferingUuid):
        '''
        request: {
            "org.zstack.storage.backup.sftp.APIAddSftpBackupStorageMsg": {
                "hostname": "10.0.89.18",
                "username": "root",
                "password": "Skt6edg",
                "type": "SftpBackupStorage",
                "resourceUuid": "8923e38182ca454482d820248ad8a53c",
                "name": "bs-gzp5",
                "description": null,
                "url": "/root",
                "session": {"uuid": "5c54fbf4e8784497a87eab47552a0edc"}
            }
        }
        :param name: 资源名字；
        :param hostname: Sftp服务器的主机名，一般情况需要输入服务器IP；
        :param url: BS存储需要的路径，如“/home”,分区等；
        :param username: Sftp服务器用户名；如“root”
        :param password: Sftp服务器登录密码；
        :return:
        '''
        content = {
            "name": name,
            "diskOfferingUuid": diskOfferingUuid,
            "description": "Create a Data Volume %s" % name
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.storage.backup.sftp.APIAddSftpBackupStorageMsg", content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)
            print "successfully Create a Data Volume %s From a Disk Offering, status %s" % (name, status)
        else:
            print "Found Error in API requests.pls check!"

    def create_dataVolume_from_volume_template(self, name, imageUuid, primaryStorageUuid):
        content = {
            "name": name,
            "imageUuid": imageUuid,
            "primaryStorageUuid": primaryStorageUuid,
            "description": "Create a Data Volume %s" % name
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.storage.backup.sftp.APIAddSftpBackupStorageMsg", content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)
            print "successfully Create Data Volume %s From an Image, status %s" % (name, status)
        else:
            print "Found Error in API requests.pls check!"

    def create_dataVolume_from_volume_template(self, name, volumeSnapshotUuid, primaryStorageUuid):
        content = {
            "name": name,
            "volumeSnapshotUuid": volumeSnapshotUuid,
            "primaryStorageUuid": primaryStorageUuid,
            "description": "Create a Data Volume %s" % name
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.storage.backup.sftp.APIAddSftpBackupStorageMsg", content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)
            print "successfully Create Data Volume %s From a Volume Snapshot, status %s" % (name, status)
        else:
            print "Found Error in API requests.pls check!"

    def delete_dataVolume(self, uuid):
        '''
        request: {
            "org.zstack.header.storage.backup.APIDeleteBackupStorageMsg": {
                "uuid": "8923e38182ca454482d820248ad8a53c",
                "session": {
                  "uuid": "5c54fbf4e8784497a87eab47552a0edc"
                }
            }
        }
        :param uuid: Backup Storage uuid  备份存储的uuid
        :return:
        '''
        content = {
            "uuid": uuid
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIDeleteBackupStorageMsg", content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)
            print "successfully Delete Data Volume %s, status %s" % (uuid, status)
        else:
            print "Found Error in API requests.pls check!"
        # pass

    def change_volume_state(self, uuid, stateEvent):
        '''
        request: {
            "org.zstack.header.storage.backup.APIChangeBackupStorageStateMsg": {
                "session": {"uuid": "f63ca5149b154bc4942c374ca35f1209"},
                "uuid": "82cf947ae8b44c479c5c30d183e7ae39",
                "stateEvent": "disable"
            }
        }
        :param uuid: 备份存储的uuid
        :param stateEvent: enable/disable
        :return:
        '''
        content = {
            "uuid": uuid,
            "stateEvent": stateEvent
        }
        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIChangeBackupStorageStateMsg", content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)

            print "successfully Change Volume State %s to %s, status %s" % (uuid, stateEvent, status)
        else:
            print "Found Error in API requests.pls check!"
        # pass

    def attach_dataVolume_to_Vm(self, volumeUuid, vmInstanceUuid):
        """
        request:{
            "org.zstack.header.storage.backup.APIAttachBackupStorageToZoneMsg": {
                "zoneUuid": "c9df649b419243b597b1dec99b90833f",
                "backupStorageUuid": "70a82158faf241e98af898a79d906490",
                "session": {
                  "uuid": "2f918a2b95624f88ae00f524242aeb4a"
                }
            }
        }
        :param backupStorageUuid:
        :param zoneUuid:
        :return:
        """
        content = {
            "volumeUuid": volumeUuid,
            "vmInstanceUuid": vmInstanceUuid
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIAttachBackupStorageToZoneMsg", content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)

            print "successfully Attach DataVolume %s To Vm %s, status %s" % (volumeUuid, vmInstanceUuid, status)
        else:
            print "Found Error in API requests.pls check!"

    def detach_dataVolume_from_Vm(self, uuid):
        """
        request:{
            "org.zstack.header.storage.backup.APIDetachBackupStorageFromZoneMsg": {
                "zoneUuid": "c9df649b419243b597b1dec99b90833f",
                "backupStorageUuid": "70a82158faf241e98af898a79d906490",
                "session": {
                  "uuid": "2f918a2b95624f88ae00f524242aeb4a"
                }
            }
        }
        :param backupStorageUuid:
        :param zoneUuid:
        :return:
        """
        content = {
            "uuid": uuid
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIDetachBackupStorageFromZoneMsg",
                                      content)
        if rsp_code == 200:
            # self.error_if_fail(rsp)
            job_uuid = rsp['uuid']
            status = self.base.query_until_done(job_uuid)

            print "successfully Detach DataVolume State %s From Vm %s, status %s" % (backupStorageUuid, zoneUuid, status)
        else:
            print "Found Error in API requests.pls check!"

    def query_backupstorage_by_uuid(self, uuid):
        '''
        request: {
            "org.zstack.header.storage.backup.APIQueryBackupStorageMsg": {
                "session": {"uuid": "f63ca5149b154bc4942c374ca35f1209"},
                "conditions": [{"name": "uuid", "value": "70a82158faf241e98af898a79d906490", "op": "="},{"name": "state", "value": "enable", "op": "="}]
            }
        }
        :param name: Backup Storage 名称
        :return:
        '''
        content = {
            "conditions": [
                {
                    "name": "uuid",
                    "value": uuid,
                    "op": "="
                }
            ]
        }
        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIQueryBackupStorageMsg", content)
        if rsp_code == 200:
            self.base.error_if_fail(rsp)
            # job_uuid = rsp['uuid']
            # status = self.query_until_done(job_uuid)

            print "successfully query Backup Storage by uuid %s, result: %s" % (uuid, rsp)
            return rsp
        else:
            print "Found Error in API requests. Pls check!"

    def query_backupstorage_by_name(self, name):
        '''
        request: {
            "org.zstack.header.storage.backup.APIQueryBackupStorageMsg": {
                "session": {"uuid": "f63ca5149b154bc4942c374ca35f1209"},
                "conditions": [{"name": "name", "value": "zone-yu", "op": "="},{"name": "state", "value": "enable", "op": "="}]
            }
        }
        :param name: Backup Storage 名称
        :return:
        '''
        content = {
            "conditions": [
                {
                    "name": "name",
                    "value": name,
                    "op": "="
                }
            ]
        }
        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIQueryBackupStorageMsg", content)
        if rsp_code == 200:
            self.base.error_if_fail(rsp)
            # job_uuid = rsp['uuid']
            # status = self.query_until_done(job_uuid)

            print "successfully query Backup Storage by name %s, result: %s" % (name, rsp)
            return rsp
        else:
            print "Found Error in API requests. Pls check!"

    # Query Backup Storage
    def query_all_backupstorage(self):
        '''
        request: {
            "org.zstack.header.storage.backup.APIQueryBackupStorageMsg": {
                "session": {"uuid": "f63ca5149b154bc4942c374ca35f1209"},
                "conditions": []
            }
        }
        :return:
        '''
        content = {
            "conditions": []
        }

        rsp, rsp_code = self.base.api_call(self.UUID, "org.zstack.header.storage.backup.APIQueryBackupStorageMsg", content)
        if rsp_code == 200:
            self.base.error_if_fail(rsp)
            # job_uuid = rsp['uuid']
            # status = self.query_until_done(job_uuid)
            print "successfully Query All Backup Storage %s" % (rsp)
            return rsp
        else:
            print "Found Error in API requests. Pls check!"

    def tags_host(self, tags):
        pass

    def logout(self, session_uuid):
        content = {"sessionUuid": session_uuid}
        rsp, rsp_code = self.base.api_call(None, "org.zstack.header.identity.APILogOutMsg", content)
        if rsp_code == 200:
            self.base.error_if_fail(rsp)
            print "successfully logout"
        else:
            print "Found Error in API requests. Pls check"


if __name__ == '__main__':
    new_zs = zstack_backup_storage_api()
    try:
        # session_uuid = new_zs.login()
        # new_zs.add_sftp_backupstorage('bs_yuan', '/root', '10.0.89.18', 'root', 'Skt6edg')
        # new_zs.delete_backupstorage('0e3292931b88447cb991483890df404c')
        # new_zs.change_backupstorage_state('70a82158faf241e98af898a79d906490', 'enable')
        # new_zs.reconnect_backupstorage('70a82158faf241e98af898a79d906490')
        # new_zs.detach_backup_storage_from_zone('70a82158faf241e98af898a79d906490', 'c9df649b419243b597b1dec99b90833f')
        # new_zs.attach_backup_storage_to_zone('70a82158faf241e98af898a79d906490', 'c9df649b419243b597b1dec99b90833f')
        # new_zs.query_all_backupstorage()
        new_zs.query_backupstorage_by_uuid(uuid='70a82158faf241e98af898a79d906490')
        # new_zs.query_backupstorage_by_name('Backup_Storage-1')
    except Exception, e:
        print "执行错误！", e

    new_zs.logout(new_zs.UUID)