150 lines
5.9 KiB
Python
150 lines
5.9 KiB
Python
# Copyright (c) 2006-2010 Mitch Garnaat http://garnaat.org/
|
|
# Copyright (c) 2010, Eucalyptus Systems, Inc.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a
|
|
# copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish, dis-
|
|
# tribute, sublicense, and/or sell copies of the Software, and to permit
|
|
# persons to whom the Software is furnished to do so, subject to the fol-
|
|
# lowing conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
|
|
# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
|
|
# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
# IN THE SOFTWARE.
|
|
|
|
from boto.cloudfront.identity import OriginAccessIdentity
|
|
|
|
def get_oai_value(origin_access_identity):
|
|
if isinstance(origin_access_identity, OriginAccessIdentity):
|
|
return origin_access_identity.uri()
|
|
else:
|
|
return origin_access_identity
|
|
|
|
class S3Origin(object):
|
|
"""
|
|
Origin information to associate with the distribution.
|
|
If your distribution will use an Amazon S3 origin,
|
|
then you use the S3Origin element.
|
|
"""
|
|
|
|
def __init__(self, dns_name=None, origin_access_identity=None):
|
|
"""
|
|
:param dns_name: The DNS name of your Amazon S3 bucket to
|
|
associate with the distribution.
|
|
For example: mybucket.s3.amazonaws.com.
|
|
:type dns_name: str
|
|
|
|
:param origin_access_identity: The CloudFront origin access
|
|
identity to associate with the
|
|
distribution. If you want the
|
|
distribution to serve private content,
|
|
include this element; if you want the
|
|
distribution to serve public content,
|
|
remove this element.
|
|
:type origin_access_identity: str
|
|
|
|
"""
|
|
self.dns_name = dns_name
|
|
self.origin_access_identity = origin_access_identity
|
|
|
|
def __repr__(self):
|
|
return '<S3Origin: %s>' % self.dns_name
|
|
|
|
def startElement(self, name, attrs, connection):
|
|
return None
|
|
|
|
def endElement(self, name, value, connection):
|
|
if name == 'DNSName':
|
|
self.dns_name = value
|
|
elif name == 'OriginAccessIdentity':
|
|
self.origin_access_identity = value
|
|
else:
|
|
setattr(self, name, value)
|
|
|
|
def to_xml(self):
|
|
s = ' <S3Origin>\n'
|
|
s += ' <DNSName>%s</DNSName>\n' % self.dns_name
|
|
if self.origin_access_identity:
|
|
val = get_oai_value(self.origin_access_identity)
|
|
s += ' <OriginAccessIdentity>%s</OriginAccessIdentity>\n' % val
|
|
s += ' </S3Origin>\n'
|
|
return s
|
|
|
|
class CustomOrigin(object):
|
|
"""
|
|
Origin information to associate with the distribution.
|
|
If your distribution will use a non-Amazon S3 origin,
|
|
then you use the CustomOrigin element.
|
|
"""
|
|
|
|
def __init__(self, dns_name=None, http_port=80, https_port=443,
|
|
origin_protocol_policy=None):
|
|
"""
|
|
:param dns_name: The DNS name of your Amazon S3 bucket to
|
|
associate with the distribution.
|
|
For example: mybucket.s3.amazonaws.com.
|
|
:type dns_name: str
|
|
|
|
:param http_port: The HTTP port the custom origin listens on.
|
|
:type http_port: int
|
|
|
|
:param https_port: The HTTPS port the custom origin listens on.
|
|
:type http_port: int
|
|
|
|
:param origin_protocol_policy: The origin protocol policy to
|
|
apply to your origin. If you
|
|
specify http-only, CloudFront
|
|
will use HTTP only to access the origin.
|
|
If you specify match-viewer, CloudFront
|
|
will fetch from your origin using HTTP
|
|
or HTTPS, based on the protocol of the
|
|
viewer request.
|
|
:type origin_protocol_policy: str
|
|
|
|
"""
|
|
self.dns_name = dns_name
|
|
self.http_port = http_port
|
|
self.https_port = https_port
|
|
self.origin_protocol_policy = origin_protocol_policy
|
|
|
|
def __repr__(self):
|
|
return '<CustomOrigin: %s>' % self.dns_name
|
|
|
|
def startElement(self, name, attrs, connection):
|
|
return None
|
|
|
|
def endElement(self, name, value, connection):
|
|
if name == 'DNSName':
|
|
self.dns_name = value
|
|
elif name == 'HTTPPort':
|
|
try:
|
|
self.http_port = int(value)
|
|
except ValueError:
|
|
self.http_port = value
|
|
elif name == 'HTTPSPort':
|
|
try:
|
|
self.https_port = int(value)
|
|
except ValueError:
|
|
self.https_port = value
|
|
elif name == 'OriginProtocolPolicy':
|
|
self.origin_protocol_policy = value
|
|
else:
|
|
setattr(self, name, value)
|
|
|
|
def to_xml(self):
|
|
s = ' <CustomOrigin>\n'
|
|
s += ' <DNSName>%s</DNSName>\n' % self.dns_name
|
|
s += ' <HTTPPort>%d</HTTPPort>\n' % self.http_port
|
|
s += ' <HTTPSPort>%d</HTTPSPort>\n' % self.https_port
|
|
s += ' <OriginProtocolPolicy>%s</OriginProtocolPolicy>\n' % self.origin_protocol_policy
|
|
s += ' </CustomOrigin>\n'
|
|
return s
|
|
|