41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
'''
|
||
|
|
Create GCE resources for use in integration tests.
|
||
|
|
|
||
|
|
Takes a prefix as a command-line argument and creates a persistent disk
|
||
|
|
named ${prefix}-base and a snapshot of it named ${prefix}-snapshot.
|
||
|
|
prefix will be forced to lowercase, to ensure the names are legal GCE
|
||
|
|
resource names.
|
||
|
|
'''
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import optparse
|
||
|
|
|
||
|
|
import gce_credentials
|
||
|
|
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
parser = optparse.OptionParser(
|
||
|
|
usage="%s [options] <prefix>" % (sys.argv[0],), description=__doc__)
|
||
|
|
gce_credentials.add_credentials_options(parser)
|
||
|
|
parser.add_option("--prefix",
|
||
|
|
action="store", dest="prefix",
|
||
|
|
help="String used to prefix GCE resource names (default: %default)")
|
||
|
|
|
||
|
|
(opts, args) = parser.parse_args()
|
||
|
|
gce_credentials.check_required(opts, parser)
|
||
|
|
if not args:
|
||
|
|
parser.error("Missing required argument: name prefix")
|
||
|
|
return (opts, args)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
|
||
|
|
(opts, args) = parse_args()
|
||
|
|
gce = gce_credentials.get_gce_driver(opts)
|
||
|
|
prefix = args[0].lower()
|
||
|
|
try:
|
||
|
|
base_volume = gce.create_volume(
|
||
|
|
size=10, name=prefix+'-base', location='us-central1-a')
|
||
|
|
gce.create_volume_snapshot(base_volume, name=prefix+'-snapshot')
|
||
|
|
except KeyboardInterrupt, e:
|
||
|
|
print "\nExiting on user command."
|