ó {ŽŠXc@sˆdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZd„Zdefd„ƒYZdefd„ƒYZd efd „ƒYZd efd „ƒYZd efd„ƒYZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#d„Z$d„Z%d„Z&dd„Z(d„Z)ddd „Z*d!d"„Z+d#„Z,d$„Z-d%ej.fd&„ƒYZ/d'e0fd(„ƒYZ1d)e0fd*„ƒYZ2d+ej.fd,„ƒYZ3d-ej4fd.„ƒYZ5dS(/s_ Miscellaneous utility functions and classes. .. testsetup:: * from wizard.util import * iÿÿÿÿN(tusercCsmytt|ƒƒSWnRttfk rh|dksZ|dksZ|dksZ|dkr^tSt|ƒSXdS(s Parse the contents of an environment variable as a boolean. This recognizes more values as ``False`` than :func:`bool` would. >>> boolish("0") False >>> boolish("no") False >>> boolish("1") True tNotnotfalsetFalseN(tbooltintt ValueErrort TypeErrorR(tval((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytboolishs 0tChangeDirectorycBs)eZdZd„Zd„Zd„ZRS(sš Context for temporarily changing the working directory. >>> with ChangeDirectory("/tmp"): ... print os.getcwd() /tmp cCs||_d|_dS(N(tdirtNonetolddir(tselfR ((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt__init__7s cCs tjƒ|_t|jƒdS(N(tostgetcwdRtchdirR (R((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt __enter__:scGst|jƒdS(N(RR(Rtargs((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt__exit__=s(t__name__t __module__t__doc__RRR(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyR /s  tCountercBsMeZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS(sñ Object for counting different values when you don't know what they are a priori. Supports index access and iteration. >>> counter = Counter() >>> counter.count("foo") >>> print counter["foo"] 1 cCs i|_dS(N(tdict(R((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyRJscCs*|jj|dƒ|j|cd7>> dictmap(lambda x: x + 2, {'a': 1, 'b': 2}) {'a': 3, 'b': 4} c3s'|]\}}|ˆ|ƒfVqdS(N((t.0tktv(tf(s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pys ¿s(Rtitems(R\td((R\s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytdictmap¸scs t‡fd†|jƒDƒƒS(s‘ A map function for dictionaries that passes key and value. >>> dictkmap(lambda x, y: x + y, {1: 4, 3: 4}) {1: 5, 3: 7} c3s*|] \}}|ˆ||ƒfVqdS(N((RYRZR[(R\(s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pys Ès(RR](R\R^((R\s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytdictkmapÁscCs…|jdƒ}t}d}xc|dD]W}|jƒ}|sDq&n|ddkr`t}q&n|r&t}|jdƒdSq&W|S(se Reads the traceback from a Python program and grabs the fully qualified exception name. s s (unknown)iit t:(tsplitRtrstriptTruet partition(toutputtlinestcuetresulttline((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytget_exception_nameÊs cCstj|ƒjS(s4Finds the uid of the person who owns this directory.(RRNtst_uid(R ((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt get_dir_uidÝscCsotjjtjjtjjtjjtƒƒƒdƒ}tjdd|ddgdtjƒj ƒdj ƒS(s4Returns the commit ID of the current Wizard install.s.gittgits --git-dir=s rev-parsetHEADR.i( RRHtjointdirnametabspatht__file__R)R*R+t communicateRd(t wizard_git((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt get_revisionás9cCsDtjƒ}|dkr!t‚ntj|ƒ}d|j|jfS(s³ Returns ``Real Name `` suitable for use in Git ``Something-by:`` string. Throws :exc:`NoOperatorInfo` if no operator information is available. s%s <%s>N(RtoperatorR tNoOperatorInfotpwnamtrealnametemail(toptinfo((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytget_operator_gités    cCsbtjƒ}|dkrdStj|ƒ}|js8dStjd|jƒtjd|jƒdS(s§ Sets :envvar:`GIT_COMMITTER_NAME` and :envvar:`GIT_COMMITTER_EMAIL` environment variables if applicable. Does nothing if no information is available NtGIT_COMMITTER_NAMEtGIT_COMMITTER_EMAIL(RRxR RzR{RtputenvR|(R}R~((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytset_operator_envõs   cCs[tjƒ}|dkrdS|js)dStjdd|jƒtjdd|jƒdS(s¦ Sets :envvar:`GIT_AUTHOR_NAME` and :envvar:`GIT_AUTHOR_EMAIL` environment variables if applicable. Does nothing if :func:`wizard.user.passwd` fails. NtGIT_AUTHOR_NAMEs%stGIT_AUTHOR_EMAIL(RtpasswdR R{RR‚R|(R~((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytset_author_envs   cCstƒtƒdS(s;Sets all appropriate environment variables for Git commits.N(RƒR‡(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt set_git_envscCs*djdtƒddjtjƒgƒS(s9Returns strings for placing in Git log info about Wizard.s sWizard-revision: %ssWizard-args: %sRa(RqRwR-targv(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytget_git_footerscCswtjj|ƒsdSd|}d}x:tjƒD],}d||f}tjj|ƒs3Pq3q3Wtj||ƒ|S(s&Moves a file/dir to a backup location.s%s.baks%s.%dN(RRHtlexistsR t itertoolsRtrename(tfiletprefixtnameRU((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt safe_unlinks cCs)ytj|ƒWntk r$nXdS(s6Unlink a file, but don't complain if it doesn't exist.N(RRLRC(RŽ((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt soft_unlink*s cCsCytj|ƒWn+tk r>}|jtjkr8q?‚nXdS(sv Create a directory path (a la ``mkdir -p`` or ``os.makedirs``), but don't complain if it already exists. N(RtmakedirsRCRDRE(RHtexc((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyR“1s c CsEytj|ƒ}|jdƒd|jdƒ}|r”idd6}tjd||ƒtjdtj|ƒƒ|j d|tj|ƒ|ƒn#tjd||ƒ|j d|ƒ|j ƒ}tjd |j ƒtjd |j ƒ|j ƒ}|jƒ|SWn:tjk r@} | jtjkr:t|ƒ‚qA‚nXdS( Nt/s!application/x-www-form-urlencodeds Content-typesPOST request to http://%s%ssPOST contents: tPOSTsGET request to http://%s%stGETsResponse code: %dsResponse headers: %s(thttplibtHTTPConnectionRdtlstripRJR~tdebugturllibt urlencodetrequestt getresponsetstatustmsgRFR0tsockettgaierrorRDt EAI_NONAMEtDNSError( thostRHtsubpathtpostthtfullpaththeadersR<tdataRV((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytfetch>s(  "   cCs9t|dƒ}|jƒt|jtƒ}|jƒ|S(s4Returns ``True`` if ``filename`` has mixed newlines.tU(R>RFt isinstancetnewlinesttupleR0(tfilenameR\tret((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytmixed_newlinesWs   s.gitc Cs|dkrtjƒ}nd}xætj|ƒD]Õ\}}}xÃ|D]»}tjj||ƒjtjj||ƒƒsGtjj||ƒ}y2tjj|ƒrªwGn|tjj|ƒ7}Wqt k rþ}|j t j krøt j d|ƒqÿ‚qXqGqGWq1W|S(s¹ Recursively determines the disk usage of a directory, excluding .git directories. Value is in bytes. If ``dir`` is omitted, the current working directory is assumed. is#%s disappeared before we could statN(R RRtwalkRHRqt startswithtislinktgetsizeRCRDRPRJRK( R t excluded_dirt sum_sizestrootR;tfilesRRŽRV((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt disk_usage_s   -icCsdjd„t|ƒDƒƒS(s7Generates a random alphanumeric key of ``length`` size.tcss(|]}tjtjtjƒVqdS(N(trandomtchoicetstringtletterstdigits(RYRU((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pys ws(Rqtxrange(tlength((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt random_keyuscCst|ƒjdƒdS(s.Truncates the Scripts specific version number.s-scriptsi(tstrRf(tversion((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyttruncateyscCsvd}tjj|ƒs(tj|ƒnttjj|dƒdƒjdƒttjj|dƒdƒjdƒdS(su Generates a .wizard directory and initializes it with some common files. This operation is idempotent. s.wizards .htaccessR:sDeny from all s .gitignores* N(RRHtisdirtmkdirR>RqRA(t wizard_dir((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pytinit_wizard_dir}s %RycBseZdZRS(s?No information could be found about the operator from Kerberos.(RRR(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyRy‰sRTcBseZejZRS((RRRDRS(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyRTsRXcBseZejZRS((RRRDRP(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyRXsRRcBseZd„Zd„ZRS(cCs ||_dS(N(R (RR ((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyR”scCsdS(Nsa ERROR: Could not acquire lock on directory. Maybe there is another migration process running? ((R((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyt__str__–s(RRRRÎ(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyRR“s R¥cBs)eZejZdZd„Zd„ZRS(cCs ||_dS(N(R¦(RR¦((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyR¡scCs d|jS(Ns( ERROR: Could not resolve hostname %s. (R¦(R((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyRΣsN( RRR¢R¤RDR R¦RRÎ(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyR¥s  (6Rtos.pathRR)tpwdR-R¢RDRŒR3R˜RœRMRJR¿RÁtwizardRR tobjectR RR&R2R7RR_R`RlRnRwRRƒR‡RˆRŠR‘R’R“R R­R´R½RÆRÉRÍtErrorRyRQRTRXRRR£R¥(((s9/afs/athena.mit.edu/contrib/scripts/wizard/wizard/util.pyts`                  3