diff --git a/update-efs-mount.py b/update-efs-mount.py index 696a4a7a05623a7307cd5cada043dcc2f9c1414e..4a91f92675b39e78ccfb85321c4c53c90b3a8880 100755 --- a/update-efs-mount.py +++ b/update-efs-mount.py @@ -1,8 +1,7 @@ #!/usr/bin/python import boto3 -import urllib2 -import json +import requests import sys import re import tempfile @@ -42,8 +41,8 @@ def main(): # Returns the region the instance is currently running in (eg: 'us-east') def get_region(): - instance_info = urllib2.urlopen('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/').read() - instance_info = json.loads(instance_info) + session_token = requests.put('http://169.254.169.254/latest/api/token', headers={'X-aws-ec2-metadata-token-ttl-seconds': 60}).text + instance_info = requests.get('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/', headers={'X-aws-ec2-metadata-token': session_token}).json() return instance_info['region'] # Returns the instance_id of this VM.