3
\me @ s d dl Z d dlZd dlmZ d dlmZ d dlmZmZm Z m
Z
m Z d dl m
Z
d dlmZ ejeZe d$ddZd d
Zd d Zd
d Zdd Zdd Zdd Ze dd Zdd Zdd Z d%d d Z d d Z d d! Z d"d# Z dS )& N) lru_cache)
exceptions)dminetsubp
url_helperutil)NoDHCPLeaseError)EphemeralDHCPv4c C s t d}xt D ]}yDt| |d|id( t|||||} tj| }
t|
|
S Q R X W q ttj t t
j fk
r } zt j
d| | }W Y d d } ~ X qX qW |d S )NzFailed to DHCPurl)ifaceZconnectivity_url_datazDHCP Exception: %s) RuntimeErrorget_interface_listr
read_metadatajsonloadsrefactor_metadatar r ZProcessExecutionErrorr ZRequestExceptionLOGerror) Zdistror timeoutretries sec_betweenagentZtmp_dirZ exceptionr Zv1metadataexc r /usr/lib/python3.6/vultr.py get_metadata s&
r c C sL | d | d< | d | d< | d d }d| d kr<| d d }|j | d< d S )Nzinstance-v2-idz instance-idZhostnamezlocal-hostnameregionZ
regioncodeZ countrycode)lower)r r r r r r 4 s r c C s. g } x$t j D ]}d|kr q| j| qW | S )NZdummy)r Zfind_candidate_nicsappend)Zifacesr r r r r > s r c C s t jdt jddS )Nzsystem-manufacturerzsystem-serial-number) manufacturerZsubid)r Z
read_dmi_datar r r r get_sysinfoJ s r" c C s t d dkrdS dS )Nr! VultrTF)r" r r r r is_baremetalR s r$ c C s. t } | d dkrdS dtj j kr*dS dS )Nr! r# TZvultrF)r" r Z get_cmdlinesplit)Zsysinfor r r is_vultrY s r& c C sH d| } d|d}t j| ||||d}|j s>td| |j|jj S )Nz
%s/v1.json cloudinit)zMetadata-Tokenz
User-Agent)r r headersr z!Failed to connect to %s: Code: %s)r Zreadurlokr
codecontentsdecode)r r r r r r( Zresponser r r r h s
r c C s t j S )N)r Zget_interfaces_by_macr r r r get_interface_map s r- c C s t }| |krd S |j| S )N)r- get)macZ macs_to_nicr r r get_interface_name s r0 c C s ddddgdgd}t | dkr@t| d dd }|d
j| x@tdt | D ].}| | }|jd rhqPt|}|d
j| qPW |S ) N Z
nameserverz 108.61.10.10z2001:19f0:300:1704::6)typeaddress)versionconfigr T)primaryr5 Z unconfigured)lengenerate_interfacer ranger. )Z
interfacesnetworkZpublici interfaceZprivater r r generate_network_config s
r= Fc C s t | d }|s td| d |d| d d}|rTd|d< ddd d
dd g|d < |s|d d| d
d | d
d dg|d < t| | t| | |S )Nr/ z.Interface: %s could not be found on the systemphysical)namer2 mac_addressr1 z accept-raZdhcpauto)r2 controlZ
ipv6_slaacsubnetsstaticipv4r3 netmask)r2 rB r3 rF )r0 r
generate_interface_routes'generate_interface_additional_addresses)r< r6 interface_namenetcfgr r r r8 s&
r8 c C sH d| kr| d |d< d| kr(| d |d< d| krD| d |d d d< d S )NZmtuz accept-raroutesrC r r )r< rJ r r r rG s rG c C s t | d d }d| krn|dkrnxL| d d D ]<}dd|d |d d}d |kr\|d |d < |d
j| q.W t | d d }d | kr|dkrxR| d d D ]B}d dd
|d |d f d}d |kr|d |d < |d
j| qW d S )NrE
additionalr rD rA r3 rF )r2 rB r3 rF rK rC Zipv6Zstatic6z%s/%sr: prefix)r2 rB r3 )r7 r )r< rJ Zadditional_countrL addr r r rH s( rH c C sL xF| d D ]:}|d dkr q
t |d }|s r@ z.Interface: %s could not be found on the systemr? )r0 r
)rJ r< rI r r r add_interface_names s rO )N)F)!r Zlogging functoolsr Zrequestsr r' r r r r r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr
Z getLogger__name__r r r r r" r$ r& r r- r0 r= r8 rG rH rO r r r r s,
% #