From 7bbccddc62a267375e0611c86c0561ce38d46df0 Mon Sep 17 00:00:00 2001
From: Lester Wade <lester@firefly.home.net>
Date: Mon, 22 Apr 2013 17:27:57 -0700
Subject: [PATCH] Add region support to ec2 volume module.

---
 ec2_vol | 34 +++++++++++++++++++++++++---------
 1 file changed, 25 insertions(+), 9 deletions(-)

diff --git a/ec2_vol b/ec2_vol
index f2d41657364..bf6435fd078 100644
--- a/ec2_vol
+++ b/ec2_vol
@@ -40,6 +40,12 @@ options:
     required: false
     default: null
     aliases: []
+  region:
+    description:
+      - region in which to create the volume
+    required: false
+    default: null
+    aliases: []
   zone:
     description:
       - zone in which to create the volume, if unset uses the zone the instance is in (if set) 
@@ -71,7 +77,7 @@ import sys
 import time
 
 try:
-    import boto 
+    import boto.ec2
 except ImportError:
     print "failed=True msg='boto required for this module'"
     sys.exit(1)
@@ -82,6 +88,7 @@ def main():
             instance = dict(),
             volume_size = dict(required=True),
             device_name = dict(),
+            region = dict(),
             zone = dict(),
             ec2_url = dict(aliases=['EC2_URL']),
             ec2_secret_key = dict(aliases=['EC2_SECRET_KEY']),
@@ -92,6 +99,7 @@ def main():
     instance = module.params.get('instance')
     volume_size = module.params.get('volume_size')
     device_name = module.params.get('device_name')
+    region = module.params.get('region')
     zone = module.params.get('zone')
     ec2_url = module.params.get('ec2_url')
     ec2_secret_key = module.params.get('ec2_secret_key')
@@ -104,14 +112,22 @@ def main():
         ec2_secret_key = os.environ['EC2_SECRET_KEY']
     if not ec2_access_key and 'EC2_ACCESS_KEY' in os.environ:
         ec2_access_key = os.environ['EC2_ACCESS_KEY']
-
-    try:
-        if ec2_url: # if we have an URL set, connect to the specified endpoint 
-            ec2 = boto.connect_ec2_endpoint(ec2_url, ec2_access_key, ec2_secret_key)
-        else: # otherwise it's Amazon.
-            ec2 = boto.connect_ec2(ec2_access_key, ec2_secret_key)
-    except boto.exception.NoAuthHandlerFound, e:
-        module.fail_json(msg = str(e))
+    
+    # If we have a region specified, connect to its endpoint.
+    if region: 
+        try:
+          ec2 = boto.ec2.connect_to_region(region, aws_access_key_id=ec2_access_key, aws_secret_access_key=ec2_secret_key)
+        except boto.exception.NoAuthHandlerFound, e:
+          module.fail_json(msg = str(e))
+    # Otherwise, no region so we fallback to the old connection method
+    else: 
+        try:
+            if ec2_url: # if we have an URL set, connect to the specified endpoint 
+                ec2 = boto.connect_ec2_endpoint(ec2_url, ec2_access_key, ec2_secret_key)
+            else: # otherwise it's Amazon.
+                ec2 = boto.connect_ec2(ec2_access_key, ec2_secret_key)
+        except boto.exception.NoAuthHandlerFound, e:
+            module.fail_json(msg = str(e))
 
     # Here we need to get the zone info for the instance. This covers situation where 
     # instance is specified but zone isn't.