U W¨+dgUã@sîdZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZddlmZddlmZdd lmZddlZddlZdd lmZdd lmZdd lmZmZd ddddddgZGdd„deƒZ Gdd„deƒZ!Gdd„de ƒZ"dS)z) High-level abstraction of an EC2 server éN)ÚIObject)ÚBotoConfigPathÚConfig)ÚModel)ÚStringPropertyÚIntegerPropertyÚBooleanPropertyÚCalculatedProperty)Úpropget)ÚZone)ÚKeyPair)Úclosing)ÚEC2ResponseError)ÚsixÚStringIOúm1.smallzm1.largez m1.xlargeú c1.mediumz c1.xlargez m2.2xlargez m2.4xlargec@s8eZdZddd„Zdd„Zdd„Zdd „Zdd d „Zd S)ÚBundlerÚrootcCs*ddlm}||_||_|||d|_dS)Nr©Ú SSHClient)Úuname)Úboto.manage.cmdshellrÚserverrÚ ssh_client)Úselfrrr©rú6/tmp/pip-unpacked-wheel-dlxw5sjy/boto/manage/server.pyÚ__init__-s zBundler.__init__cCsrtdƒ|j ¡tj |¡\}}d||_|j ||j¡tj |¡\}}d||_|j ||j¡tdƒdS)Nz5 copying cert and pk over to /mnt directory on serverz/mnt/%sú ...complete!) ÚprintrZ open_sftpÚosÚpathÚsplitÚremote_key_fileZput_fileÚremote_cert_file)rÚkey_fileÚ cert_filer"ÚnamerrrÚ copy_x5093s   zBundler.copy_x509cCsŽd}|jdkrd}|d7}|d|j|jf7}|d|jjj7}|d|7}|d|7}|d 7}|jjd ksx|jjd kr‚|d 7}n|d 7}|S)NÚrúsudo zec2-bundle-vol z -c %s -k %s z-u %s z-p %s z-s %d z-d /mnt rrz-r i386z -r x86_64)rr%r$rÚ _reservationZowner_idÚ instance_type)rÚprefixÚsizeÚssh_keyÚcommandrrrÚ bundle_image>s    zBundler.bundle_imagecCsZd}|jdkrd}|d7}|d|7}|d|7}|d|jjj7}|d|jjj7}|S) Nr*rr+zec2-upload-bundle z-m /mnt/%s.manifest.xml z-b %s z-a %s z-s %s )rrÚec2Úaws_access_key_idÚaws_secret_access_key)rÚbucketr.r0r1rrrÚ upload_bundleNs   zBundler.upload_bundleNTc Cs`tƒ} |s|  d¡}|s"|  d¡}|s0|  d¡}|s>|  d¡}|sL|  d¡}|sZ|j ¡}| ||¡|sptƒ}| dt ¡| d¡|r–| d¡| |  |||¡¡| d ¡| |  |||¡¡| d ¡| d t ¡| d ¡|  ¡} t d ƒt | ƒ|j | ¡} t d | dƒt d | dƒt dƒt dƒ|jjj|d||fd|_|jS)NzName of S3 bucketzPrefix for AMI filezPath to RSA private key filezPath to RSA public cert filezSize (in MB) of bundled imagezsudo mv %s /mnt/boto.cfg; z0mv ~/.ssh/authorized_keys /mnt/authorized_keys; z history -c; z; zsudo mv /mnt/boto.cfg %s; z.mv /mnt/authorized_keys ~/.ssh/authorized_keysz3running the following command on the remote server:z %srérzregistering image...z%s/%s.manifest.xml)r(Zimage_location)rZ get_stringÚ get_filenameZget_intrÚget_ssh_key_filer)rÚwriterr2r7Úgetvaluer rÚrunr3Zregister_imageZimage_id) rr6r.r&r'r/r0ÚfpZ clear_historyÚiobjectr1ÚtrrrÚbundleYsH             zBundler.bundle)r)NNNNNNNT)Ú__name__Ú __module__Ú __qualname__rr)r2r7rArrrrr+s   ÿrc@sdeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dS)ÚCommandLineGettercCs8g}|j ¡D]$}|j d¡dkr| |j|f¡q|S)NZpyamir)r3Úget_all_imagesÚlocationÚfindÚappend)rZmy_amisÚamirrrÚ get_ami_list‚s zCommandLineGetter.get_ami_listcCsb| dd¡}t|tƒr*tj |¡}||d<|sP|j d¡}tj|tjj d|d<|d  ¡|_dS)NÚregionÚ region_name)Úchoices) ÚgetÚ isinstanceÚ basestringÚbotor3Ú get_regionÚclsÚ find_propertyr ÚregionsÚconnect)rÚparamsrLÚproprrrrSŠs    zCommandLineGetter.get_regioncCs*| dd¡s&|j d¡}t |¡|d<dS)Nr(©rOrTrUr ©rrXrYrrrÚget_name”s  zCommandLineGetter.get_namecCs*| dd¡s&|j d¡}t |¡|d<dS)NÚ descriptionrZr[rrrÚget_description™s  z!CommandLineGetter.get_descriptioncCs,| dd¡s(tddtd}t |¡|d<dS)Nr-ú Instance Type©r(Ú verbose_namerN)rOrÚ InstanceTypesr r[rrrÚget_instance_typežs  ÿz#CommandLineGetter.get_instance_typecCs*| dd¡s&tddd}t |¡|d<dS)NÚquantityzNumber of Instances©r(ra)rOrr r[rrrÚ get_quantity¤s  zCommandLineGetter.get_quantitycCs0| dd¡s,tdd|jjd}t |¡|d<dS)NÚzonezEC2 Availability Zoner`)rOrr3Z get_all_zonesr r[rrrÚget_zone©s  ÿzCommandLineGetter.get_zonecCsxd}|st| dd¡}|s.tddd}t |¡}z.|j |g¡}t|ƒdkrZd}|d|d<Wqtk rpYqXqdS)NFrJZAMIrer8Tr)rOrr r3rFÚlenr)rrXZvalidrJrYÚrsrrrÚ get_ami_id¯s    zCommandLineGetter.get_ami_idcCsh| dd¡}t|tƒr@|j ¡}|D]}|j|kr$|}||d<q$|sdtdd|jjd}t |¡|d<dS)NÚgroupzEC2 Security Groupr`)rOrPrQr3Zget_all_security_groupsr(rr )rrXrlZ group_listÚgrYrrrÚ get_group¾s     ÿzCommandLineGetter.get_groupcCsn| dd¡}t|tƒrD|j ¡}|D]}|j|kr$|j}|j|d<q$|sjtdd|jjd}t |¡j|d<dS)NÚkeypairz EC2 KeyPairr`)rOrPrQr3Zget_all_key_pairsr(rr )rrXroZkey_listÚkrYrrrÚget_keyËs     ÿzCommandLineGetter.get_keycCsr||_| |¡|d ¡|_| |¡| |¡| |¡| |¡| |¡|  |¡|  |¡|  |¡dS)NrL) rTrSrWr3r\r^rcrhrfrkrnrq)rrTrXrrrrOØs        zCommandLineGetter.getN)rBrCrDrKrSr\r^rcrfrhrkrnrqrOrrrrrE€s   rEcseZdZedddZeddZeddZeddZeddZe d d d Z e d e dd Z e de dd Ze de dd Ze de dd Ze dedd Ze de dd Ze de dd Ze de dd Ze de dd Ze de dd Ze dedd ZgZgZedd„ƒZedUdd„ƒZedVdd„ƒZed d!„ƒZ dW‡fd"d#„ Z!d$d%„Z"d&d'„Z#d(d)„Z$d*d+„Z%d,d-„Z&d.d/„Z'd0d1„Z(d2d3„Z)d4d5„Z*d6d7„Z+d8d9„Z,‡fd:d;„Z-‡fdd?„Z/d@dA„Z0dBdC„Z1dDdE„Z2dFdG„Z3dHdI„Z4dJdK„Z5dLdM„Z6dXdOdP„Z7dYdQdR„Z8dSdT„Z9‡Z:S)ZÚServerTÚName)ÚuniqueraÚ Description)razEC2 Region NamezEC2 Instance IDzEC2 Elastic IP AddresszIs This Server ProductionF)raÚdefaultzAMI ID)raZcalculated_typeZ use_methodzAvailability Zone NamezPublic DNS NamezPrivate DNS NamezSecurity GroupszPrimary Security Group NamezKey Namer_zCurrent StatuszServer Launch TimezConsole OutputcCsj| d¡s| d¡| dd|¡| dd|¡| d¡sD| d¡| ddd¡| dd|jjj¡dS)NZ Credentialsr4r5Z DB_ServerZdb_typeZSimpleDBZdb_name)Ú has_sectionÚ add_sectionÚsetÚ_managerÚdomainr()rTÚcfgr4r5rrrÚadd_credentialss    zServer.add_credentialsNc Ks|rt|d}| d¡r@| d¡D]}||kr"| d|¡||<q"tƒ}| ||¡| d¡}| ¡}| ||j|j¡| d¡} | d¡} | d¡} | d¡} |dkr¶|  d d |j ¡t ƒ} |  | ¡t | tƒrÖ| j } t | tƒræ| j } | jd | d d ¡| | g| d ¡| |  ¡d}g}d}| d¡}|j}|dk rŽ| ¡dkrŽ|d}tdƒt d¡| ¡dkrxt d ¡q\| |¡td|ƒ|D]€}|ƒ}||_|dkrº| d¡dnt|ƒ|_ | d¡|_|j |_|j|_|rö|dkrö||_|  ¡| !|¡|d 7}q’|S)aq Create a new instance based on the specified configuration file or the specified configuration and the passed in parameters. If the config_file argument is not None, the configuration is read from there. Otherwise, the cfg argument is used. The config file may include other config files with a #import reference. The included config files must reside in the same directory as the specified file. The logical_volume argument, if supplied, will be used to get the current physical volume ID and use that as an override of the value specified in the config file. This may be useful for debugging purposes when you want to debug with a production config file but a test Volume. The dictionary argument may be used to override any EC2 configuration values in the config file. )r"ZEC2rLrJrorlrgNZEBSZlogical_volume_namer8rdr-)Z min_countÚ max_countÚkey_nameZsecurity_groupsr-Ú placementÚ user_datarÚ elastic_ipzEWaiting for instance to start so we can set its elastic IP address...éÚrunningz.set the elastic IP of the first instance to %sr(r*r])"rrwÚoptionsrOrErWr}r4r5ryr(rr;rPr r r=r<Ú instancesÚ__len__r ÚtimeÚsleepÚupdateZuse_ipr3Ústrr]rMÚidÚ instance_idr‚ÚputrI)rTÚ config_fileZlogical_volumer|rXÚoptionÚgetterrLr3rJZkprlrgZcfg_fpÚ reservationÚlÚir‚r†ÚinstanceÚsrrrÚcreate sp            ú    "   z Server.creater*c Cs¬tj ¡}|D]˜}| ¡}z| |g¡}Wng}YnXt|ƒdkr|ƒ}||_||_||_|j|_||_ |d|_ |j j D]} | j |kr€| |_ q€| ¡|SqdS©Nr8r)rRr3rVrWÚget_all_reservationsrir(r]rMrr,r†rŒÚ _instancerŽ) rTrr(r]rVrLr3rjr–r•rrrÚcreate_from_instance_id^s*       zServer.create_from_instance_idc Cs¼g}tj ¡}|D]¤}| ¡}| ¡}|D]Š}|jD]~}z(ttj|j dƒtj   d|j ¡Wq4t k r°|ƒ}||_|j |_ |j |_|j |_||_| ¡| |¡Yq4Xq4q*q|S)N)rzServer for %s already exists)rRr3rVrWr™r†ÚnextrrrHrŒÚlogÚinfoÚ StopIterationr(rMrr,rŽrI) rTZserversrVrLr3rjr’r•r–rrrÚcreate_from_current_instancesvs(  z$Server.create_from_current_instancesc s>tt|ƒj|f|Žd|_d|_d|_d|_d|_| ¡dS©N) ÚsuperrrrÚ ssh_key_filer3Ú _cmdshellr,ršÚ _setup_ec2)rrŒÚkw©Ú __class__rrrszServer.__init__c Csº|jr|jr|jrdS|jr¶|jr¶tj ¡D]ˆ}|j|jkr,| ¡|_|j r,|js,zL|j  |j g¡}t |ƒdkrœ|dj D] }|j|j krz|d|_||_qzWq,t k r²Yq,Xq,dSr˜)r3ršr,rŒrMrRrVr(rWrr™rir†r)rrLrjr•rrrr¥–s"       zServer._setup_ec2cCs d}|jr|j ¡|jj}|S©Nr*)ršrŠÚstate)rÚstatusrrrÚ_status©s  zServer._statuscCsd}|jr|jj}|Sr©)ršZpublic_dns_name©rÚhostnamerrrÚ _hostname°szServer._hostnamecCsd}|jr|jj}|Sr©)ršZprivate_dns_namer­rrrÚ_private_hostname¶szServer._private_hostnamecCsd}|jr|jj}|Sr©)ršr-)rÚitrrrÚ_instance_type¼szServer._instance_typecCsd}|jr|jj}|Sr©)ršÚ launch_time)rÚltrrrÚ _launch_timeÂszServer._launch_timecCsd}|jr|j ¡}|Sr©)ršZget_console_output)rÚcorrrÚ_console_outputÈs zServer._console_outputcCsg}|jr|jj}|Sr¡)r,Úgroups)rZgnrrrÚ_groupsÎszServer._groupscCs"| ¡}t|ƒdkr|djSdS)Nr8rr*)r¹rirŒ)rr¸rrrÚ_security_groupÔs  zServer._security_groupcCsd}|jr|jj}|Sr¡)ršr€)rrgrrrÚ_zoneÚsz Server._zonecCsd}|jr|jj}|Sr¡)ršr)rÚknrrrÚ _key_nameàszServer._key_namecstt|ƒ ¡| ¡dSr¡)r¢rrrŽr¥©rr§rrrŽæsz Server.putcs |jrtdƒ‚tt|ƒ ¡dS©Nz Can't delete a production server)Ú productionÚ ValueErrorr¢rrÚdeleter¾r§rrrÂêsz Server.deletecCs"|jrtdƒ‚|jr|j ¡dSr¿)rÀrÁršÚstopr¾rrrrÃðsz Server.stopcCs"|jrtdƒ‚|jr|j ¡dSr¿)rÀrÁršÚ terminater¾rrrrÄöszServer.terminatecCs|jr|j ¡dSr¡)ršÚrebootr¾rrrrÅüsz Server.rebootcCs|jdkrt d¡qdS)Nr„rƒ)r«rˆr‰r¾rrrÚwaits z Server.waitcCsb|js\tj d¡}tj |¡rDtj |d|j¡}tj |¡rD||_|js\tƒ}|  d¡|_|jS)Nz~/.sshz%s.pemzPath to OpenSSH Key file) r£r!r"Ú expanduserÚisdirÚjoinrÚisfilerr9)rZssh_dirZssh_filer?rrrr:s    zServer.get_ssh_key_filecCs,|js&ddlm}| ¡| |¡|_|jS)Nr)Úcmdshell)r¤Ú boto.managerËr:Ústart)rrËrrrÚ get_cmdshells   zServer.get_cmdshellcCs d|_dSr¡)r¤r¾rrrÚreset_cmdshellszServer.reset_cmdshellc Cs&t| ¡ƒ}| |¡}W5QRX|Sr¡)r rÎr=)rr1Úcmdr«rrrr=sz Server.runrcCs| ¡t||ƒSr¡)r:r)rrrrrÚ get_bundlerszServer.get_bundlercCs"ddlm}| ¡||||dS)Nrr)rÚssh_pwd)rrr:)rrrÒrrrrÚget_ssh_client#s zServer.get_ssh_clientcCs| d|¡S)Nzapt-get -y install %s)r=)rÚpkgrrrÚinstall(szServer.install)NNN)r*)N)r)rN);rBrCrDrr(r]rMrr‚rrÀr r‹Zami_idrgr®Zprivate_hostnameÚlistr¸Zsecurity_grouprr-r«r³ÚopenZconsole_outputÚpackagesZpluginsÚ classmethodr}r—r›r rr¥r¬r¯r°r²rµr·r¹rºr»r½rŽrÂrÃrÄrÅrÆr:rÎrÏr=rÑrÓrÕÚ __classcell__rrr§rrråsh        P        rr)#Ú__doc__Zboto.ec2rRZboto.mashups.iobjectrZboto.pyami.configrrZboto.sdb.db.modelrZboto.sdb.db.propertyrrrr rÌr Z boto.ec2.zoner Zboto.ec2.keypairr r!rˆÚ contextlibr Zboto.exceptionrZ boto.compatrrrbÚobjectrrErrrrrrÚs*       þUe