| Current File : //lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyc |
�
oB�]c @ s� d d l m Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m
Z
e j e � Z
d Z d d e e f d d e e f d
d
e e f d d e e f d
d
e e f g Z d e j f d � � YZ d � Z d � Z d � Z d d e d d d � Z d � Z d e f d � � YZ d e f d � � YZ e e j e j f f g Z d � Z e d k r�d � Z! e! � n d S( i����( t print_functionN( t log( t sources( t
url_helper( t utils
2012-03-01s meta-data/instance-ids meta-data/local-hostnames meta-data/public-keyss meta-data/vendor-datas vendor-datas user-datat DataSourceMAASc B sn e Z d Z d Z d
Z d
Z d � Z e d � � Z d � Z
d � Z d � Z d � Z
d � Z d � Z RS( s�
DataSourceMAAS reads instance information from MAAS.
Given a config metadata_url, and oauth tokens, it expects to find
files under the root named:
instance-id
user-data
hostname
vendor-data
t MAASc C sS t j j | | | | � d | _ t j j | j d � | _ t | j
� | _ d S( Nt maas( R t
DataSourcet __init__t Nonet base_urlt ost patht joint seed_dirt get_id_from_ds_cfgt ds_cfgt id_hash( t selft sys_cfgt distrot paths( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR 0 s c C s% | j s t | j � | _ n | j S( N( t
_oauth_helpert get_oauth_helperR ( R ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt oauth_helper6 s c C s# t j j | � } d | | j f S( Ns %s [%s]( R R t __str__R ( R t root( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR < s c
C s | j } y$ | j | j t | j � � t SWn< t k
r@ n, t k
rk } t j d | j | � � n X| j d d � } | s� t SyL | j | � s� t S| j | t
| d | j j d | j d d �� t SWn% t k
r� t j t d | � t SXd S( Ns %s was malformed: %st metadata_urlt read_file_or_urlR t retriesi s$ Failed fetching metadata from url %s( R t _set_dataR t read_maas_seed_dirt Truet MAASSeedDirNonet MAASSeedDirMalformedt LOGt warningt getR
t Falset wait_for_metadata_servicet read_maas_seed_urlR t readurlR t ExceptionR t logexc( R t mcfgt exct url( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt _get_data@ s.
c C s� | | _ | \ } } } | | _ | | _ | | _ | r� y t j | � | _ Wq� t k
r} } t j d | � d | _ q� Xn d S( Ns" Invalid content in vendor-data: %s( R t userdata_rawt metadatat vendordata_pureR t convert_vendordatat vendordata_rawt
ValueErrorR$ R% R
( R R/ t datat udt mdt vdt e( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR b s c C s d | j S( s/ Return the subplatform metadata source details.s
seed-dir (%s)( R ( R ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt _get_subplatformp s c C se | j } d } y t | j d | � � } Wn$ t k
rQ t j t d | � n X| d k rb t Sd } y+ | | k r� t | j d | � � } n Wn! t k
r� t j d | � n Xt j � } | j
d � r� | d } n d
| t f } | g } | j j
d | d | d | � \ } } | r8t j d | � n# t j d
| t t j � | � � t | � S( Nix t max_waits Failed to get max wait. using %si i2 t timeouts Failed to get timeout, using %st /i����s %s/%s/meta-data/instance-idt urlss Using metadata source: '%s's( Giving up on md from %s after %i seconds( R t intR& R+ R R, R$ R' R% t timet endswitht
MD_VERSIONR t wait_for_urlt debugt criticalt bool( R R/ R- R= R> t starttimet check_urlR@ t _response( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR( t s4
c C sD | j d k r t St j | d | j f i � } | j t | � k S( s locally check if the current system is the same instance.
MAAS doesn't provide a real instance-id, and if it did, it is
still only available over the network. We need to check based
only on local resources. So compute a hash based on Oauth tokens.t
datasourceN( R R
R' R t get_cfg_by_patht dsnameR ( R R t ncfg( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt check_instance_id� s N( t __name__t
__module__t __doc__RN R
R R R t propertyR R R0 R R< R( RP ( ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR ! s " "c C sA d } t g | D] } | | j | � f ^ q � } t j | � S( s� Return an oauth helper instance for values in cfg.
@raises ValueError from OauthUrlHelper if some required fields have
true-ish values but others do not.t consumer_keyt consumer_secrett token_keyt token_secret( RU RV RW RX ( t dictR& R t OauthUrlHelper( t cfgt keyst rt kwargs( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR � s .c C sT d } d j g | D] } | j | d � ^ q � } d t j | j d � � j � S( s; Given a config, generate a unique identifier for this node.RU RW RX t t s v1:s utf-8( s consumer_keys token_keys token_secret( R R& t hashlibt sha256t encodet hexdigest( R t fieldst ft idstr( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR � s .c C sm | j d � r | d } n t j j | � sJ t t j | � � d k rY t d � � n t d | d d �S( Ns file://i i s %s: not a directorys file://%st version( t
startswithR R
t isdirt lent listdirR" R) R
( t seed_d( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR � s
.c C sx | d k r t j } n | j d � r4 | d } n i } x.t D]&\ } } } }
| d k rr d | | f } n d | | | f } y� t j | � } | | d | d | d | �}
|
j � r� | r� |
j | | <qt j |
j � | | <n t
j d | |
j � WqA t j
k
rf} | j d k rK|
rKt d
| | f � � qg| j d k rg| � qgqA XqA Wt | | � S( s�
Read the maas datasource at seed_url.
read_file_or_url is a method that should provide an interface
like util.read_file_or_url
Expected format of seed_url is are the following files:
* <seed_url>/<version>/meta-data/instance-id
* <seed_url>/<version>/meta-data/local-hostname
* <seed_url>/<version>/user-data
If version is None, then <version>/ will not be used.
R? i����s %s/%ss %s/%s/%sR R> t ssl_detailss4 Fetching from %s resulted in an invalid http code %si� s Missing required %s: %sN( R
R R RC t DS_FIELDSR t fetch_ssl_detailst okt contentst
decode_binaryR$ R% t codet UrlErrorR# t check_seed_contents( t seed_urlR R> Rh R R R9 R
t _dictnamet binaryt optionalR/ Rn t respR; ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR) � s4
c
C st i } g } x� t D]� \ } } } } | | k rM | s | j | � q q n d | k r� | j d � \ } } }
| | k r� i | | <n | | | | |
<q | | | | <q Wt | � d k r� t d | � � n | r� t d | | f � � n d } | j d � rUt � } t j
| j d � d | d t �} | | k rUt d � � qUn | j d � | j d
� | f S( s� Validate if dictionary content valid as a return for a datasource.
Either return a (userdata, metadata, vendordata) tuple or
Raise MAASSeedDirMalformed or MAASSeedDirNone
R? i s %s: no data files founds %s: missing files %ss vendor-datat defaultt alloweds% vendor-data was not loadable as yaml.s user-datas meta-dataN( Ro t appendt partitionRk R" R# R
R& t objectR t load_yaml(
t contentt seedt rett missingt spatht dpatht _binaryRz t topt _t pt vd_datat err( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyRv � s2
R" c B s e Z RS( ( RQ RR ( ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR" s R# c B s e Z RS( ( RQ RR ( ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR# s c C s t j | t � S( N( R t list_from_dependst datasources( t depends( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt get_datasource_list# s t __main__c
s� d d l } d d l } d d l } | j d d � } | j d d d d d d
d �| j d d d d d
d
d �| j d d d d d d
d �| j d d d d d d
d �| j d d d d d d
d �| j d d d d d d
t �| j d d d d � } xQ dD dE dF f D]@ \ } } | j | d | �} | j d# d d$ d% d& d
d �qW| j � } i | j
d' 6| j d( 6| j d) 6| j
d* 6} | j d k rxf dG D][ }
d- |
d. } t j j | � r�t j | t j � r�| j j d/ | � | | _ q�q�Wn | j r�t j | j � } d0 | k rF| d0 d1 } n xA | j � D]3 }
|
| k rS| |
d k rS| |
| |
<qSqSW| j d k r�d2 | k r�| d2 | _ q�n | j d k r�| j j d3 � | j d4 � n t | � � � f d5 � � � f d6 � � � � � f d7 � � | j d! k r1| j j d8 | j � � j } | j d9 d: k sz| j j d; � r�t | j � \ } } } n- t | j d | j! d< | d= d> �\ } } } t" d? � t" | d k r�d@ n | j# � � t" dA � | j | � t" dB � | j | d k r'd@ n | � nc | j d k rP� | j � nD | j d k r�| j j$ d: � s�dC | j | _ n � | j � n d S(H s�
Call with single argument of directory or http or https url.
If url is given additional arguments are allowed, which will be
interpreted as consumer_key, token_key, token_secret, consumer_secret
i����Nt descriptions Interact with MAAS DSs --configt metavart filet helps specify DS config fileR| s --ckeyt keys the consumer key to auth withs --tkeys the token key to auth withs --csect secrets the consumer secret (likely '')R` s --tsecs the token secret to auth withs --apiverRh s the apiver to use ( can be used)t titlet subcommandst destt subcmdt crawls crawl the datasourceR& s do a single GET of provided urls
check-seeds read and verify seed at urlR/ s the datasource urlt nargst ?RU RW RX RV t 91_kernel_cmdline_urlt 90_dpkg_maass /etc/cloud/cloud.cfg.d/s .cfgs Used config in %s.
RL R R s) Must provide a url or a config with url.
i c s � j | d d �j S( NR i ( R* Rr ( R/ ( R ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt geturle s c s$ t d | � | � j � f � d S( Ns == %s ==
%s
( t printt decode( R/ ( R� ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt printurli s c s� | j d � r� x� � | � j � j � D]c } | j d � rT � d | | f � q( | d k rw � d | d f � q( � d | | f � q( Wn
� | � d S( NR? s %s%ss meta-datas
meta-data/( RC R� t
splitlines( R/ t line( R� R� R� ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyR� l s s Checking seed at %s
i R? s file://R R i s === user-data ===s N/As === meta-data ===s === vendor-data ===s %s/( s crawls crawl the datasource( s gets do a single GET of provided url( s
check-seeds read and verify seed at url( R� R� (% t argparset pprintt syst ArgumentParsert add_argumentR
RD t add_subparserst
add_parsert
parse_argst ckeyt tkeyt tsect csect configR R
t existst accesst R_OKt stderrt writeR t read_confR\ R/ t exitR R� R* Ri R R) t apiverR� R� RC ( R� R� R� t parsert subcmdst nameR� R� t argst credst fnamet fpathR[ R� R* t userdataR2 R: ( ( R� R� R R� sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt main( s�
'
%
"
"(" t
__future__R Ra R RB t cloudinitR t loggingR R R t getLoggerRQ R$ RD R' R! Ro R R R R R R
R) Rv R+ R" R# t DEP_FILESYSTEMt DEP_NETWORKR� R� R� ( ( ( sD /usr/lib/python2.7/site-packages/cloudinit/sources/DataSourceMAAS.pyt <module> s8 �
. & i