Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
30 lines (25 sloc) 929 Bytes
# main.py
from __future__ import print_function
from pkg.func import getEC2InstanceIPv4
from pkg.func import setCloudflareARecord
import os
def handler(event, context):
instance_name = os.getenv('EC2_INSTANCE_NAME')
cf_zone = os.getenv('CLOUDFLARE_ZONE_ID')
cf_dns = os.getenv('CLOUDFLARE_DNS_ID')
cf_email = os.getenv('CLOUDFLARE_EMAIL')
cf_api_key = os.getenv('CLOUDFLARE_API_KEY')
cf_arecord_name = os.getenv('CLOUDFLARE_A_NAME')
# Get EC2 Instance Public IPv4 Address
public_ipv4_address = getEC2InstanceIPv4(instance_name)
# FOR TESTING:
#public_ipv4_address = "1.2.3.4"
# Update Cloudflare A record with the EC2 Instance public IPv4 address
cf_status_code = setCloudflareARecord(public_ipv4_address, cf_email, cf_api_key, \
cf_arecord_name, cf_zone, cf_dns)
# FOR TESTING:
#def main():
# handler("nothing", "nothing")
#
#if __name__ == "__main__":
# main()