mirror of
https://github.com/vmware/vsphere-automation-sdk-python.git
synced 2024-11-27 03:30:00 -05:00
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
|
"""
|
||
|
|
* *******************************************************
|
||
|
|
* Copyright (c) VMware, Inc. 2016. All Rights Reserved.
|
||
|
|
* *******************************************************
|
||
|
|
*
|
||
|
|
* DISCLAIMER. THIS PROGRAM IS PROVIDED TO YOU "AS IS" WITHOUT
|
||
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, WHETHER ORAL OR WRITTEN,
|
||
|
|
* EXPRESS OR IMPLIED. THE AUTHOR SPECIFICALLY DISCLAIMS ANY IMPLIED
|
||
|
|
* WARRANTIES OR CONDITIONS OF MERCHANTABILITY, SATISFACTORY QUALITY,
|
||
|
|
* NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from vsphere.samples.vim import datastore_file
|
||
|
|
from vsphere.samples.vim.file import (detect_file, delete_file,
|
||
|
|
parse_datastore_path)
|
||
|
|
from vsphere.samples.vim.inventory import get_datastore_mo
|
||
|
|
|
||
|
|
|
||
|
|
def setup_floppy_image(context):
|
||
|
|
"""Copy floppy image used to run vcenter samples"""
|
||
|
|
floppy_src_url = context.testbed.config['FLOPPY_SRC_URL']
|
||
|
|
datacenter_name = context.testbed.config['FLOPPY_DATACENTER_NAME']
|
||
|
|
datastore_path = context.testbed.config['FLOPPY_DATASTORE_PATH']
|
||
|
|
(datastore_name, path) = parse_datastore_path(datastore_path)
|
||
|
|
|
||
|
|
datastore_mo = get_datastore_mo(context.stub_config,
|
||
|
|
context.service_instance._stub,
|
||
|
|
datacenter_name,
|
||
|
|
datastore_name)
|
||
|
|
if not datastore_mo:
|
||
|
|
raise Exception("Could not find datastore '{}'".format(datastore_name))
|
||
|
|
|
||
|
|
# See if the Floppy image exists. Copy it into the system if it does not
|
||
|
|
# exist
|
||
|
|
dsfile = datastore_file.File(datastore_mo)
|
||
|
|
if not dsfile.exists(datastore_path):
|
||
|
|
print("Putting Floppy file from '{}' at '{}'".
|
||
|
|
format(floppy_src_url, datastore_path))
|
||
|
|
dsfile.put(path=path, src_url=floppy_src_url)
|
||
|
|
|
||
|
|
|
||
|
|
def cleanup_floppy_image(context):
|
||
|
|
"""Delete floppy image after running samples"""
|
||
|
|
datacenter_name = context.testbed.config['FLOPPY_DATACENTER_NAME']
|
||
|
|
datastore_path = context.testbed.config['FLOPPY_DATASTORE_PATH']
|
||
|
|
delete_file(context.stub_config,
|
||
|
|
context.service_instance,
|
||
|
|
'Floppy Image',
|
||
|
|
datacenter_name,
|
||
|
|
datastore_path)
|
||
|
|
|
||
|
|
|
||
|
|
def detect_floppy_image(context):
|
||
|
|
"""Find the floppy image used to run vcenter samples"""
|
||
|
|
datacenter_name = context.testbed.config['FLOPPY_DATACENTER_NAME']
|
||
|
|
datastore_path = context.testbed.config['FLOPPY_DATASTORE_PATH']
|
||
|
|
return detect_file(context, 'Floppy Image', datacenter_name, datastore_path)
|
||
|
|
|
||
|
|
|
||
|
|
def setup(context):
|
||
|
|
setup_floppy_image(context)
|
||
|
|
|
||
|
|
|
||
|
|
def cleanup(context):
|
||
|
|
cleanup_floppy_image(context)
|
||
|
|
|
||
|
|
|
||
|
|
def validate(context):
|
||
|
|
return detect_floppy_image(context)
|