from __future__ import with_statementfrom grizzled.os import working_directory
with working_directory(path_to_directory):# code in here occurs within the directory
# code here is in the original directory
import os
class cd:"""Context manager for changing the current working directory"""def __init__(self, newPath):self.newPath = os.path.expanduser(newPath)
def __enter__(self):self.savedPath = os.getcwd()os.chdir(self.newPath)
def __exit__(self, etype, value, traceback):os.chdir(self.savedPath)
import subprocess # just to call an arbitrary command e.g. 'ls'
# enter the directory like this:with cd("~/Library"):# we are in ~/Librarysubprocess.call("ls")
# outside the context manager we are back wherever we started.
def quote_against_shell_expansion(s):import pipesreturn pipes.quote(s)
def put_text_back_into_terminal_input_buffer(text):# use of this means that it only works in an interactive session# (and if the user types while it runs they could insert characters between the characters in 'text'!)import fcntl, termiosfor c in text:fcntl.ioctl(1, termios.TIOCSTI, c)
def change_parent_process_directory(dest):# the horrorput_text_back_into_terminal_input_buffer("cd "+quote_against_shell_expansion(dest)+"\n")
from contextlib import contextmanagerimport os
@contextmanagerdef cd(newdir):prevdir = os.getcwd()os.chdir(os.path.expanduser(newdir))try:yieldfinally:os.chdir(prevdir)
然后,即使在抛出异常后,目录也会恢复:
os.chdir('/home')
with cd('/tmp'):# ...raise Exception("There's no place like /home.")# Directory is now back to '/home'.
if _%1 == _. goto cdDoneif _%1 == _? goto helpif /i _%1 NEQ _-H goto doCd:helpecho d.bat and dSup.py 2016.03.05. Extended chdir.echo -C = clear traversal list.echo -B or nothing = backward (to previous dir).echo -F or - = forward (to next dir).echo -R = remove current from list and return to previous.echo -S = select from list.echo -H, -h, ? = help.echo . = make window title current directory.echo Anything else = target directory.goto done
:doCd%~dp0dSup.py %1for /F %%d in ( %~dp0dSupList ) do (cd %%dif errorlevel 1 ( %~dp0dSup.py -R )goto cdDone):cdDonetitle %CD%:done
python脚本,dSup.py是:
import sys, os, msvcrt
def indexNoCase ( slist, s ) :for idx in range( len( slist )) :if slist[idx].upper() == s.upper() :return idxraise ValueError
# .........main process ...................if len( sys.argv ) < 2 :cmd = 1 # No argument defaults to -B, the most common operationelif sys.argv[1][0] == '-':if len(sys.argv[1]) == 1 :cmd = 2 # '-' alone defaults to -F, second most common operation.else :cmd = 'CBFRS'.find( sys.argv[1][1:2].upper())else :cmd = -1dir = os.path.abspath( sys.argv[1] ) + '\n'
# cmd is -1 = path, 0 = C, 1 = B, 2 = F, 3 = R, 4 = S
fo = open( os.path.dirname( sys.argv[0] ) + '\\dSupList', mode = 'a+t' )fo.seek( 0 )dlist = fo.readlines( -1 )if len( dlist ) == 0 :dlist.append( os.getcwd() + '\n' ) # Prime new directory list with current.
if cmd == 1 : # B: move backward, i.e. to previoustarget = dlist.pop(0)dlist.append( target )elif cmd == 2 : # F: move forward, i.e. to nexttarget = dlist.pop( len( dlist ) - 1 )dlist.insert( 0, target )elif cmd == 3 : # R: remove current from list. This forces cd to previous, a# desireable side-effectdlist.pop( 0 )elif cmd == 4 : # S: select from list# The current directory (dlist[0]) is included essentially as ESC.for idx in range( len( dlist )) :print( '(' + str( idx ) + ')', dlist[ idx ][:-1])while True :inp = msvcrt.getche()if inp.isdigit() :inp = int( inp )if inp < len( dlist ) :print( '' ) # Print the newline we didn't get from getche.breakprint( ' is out of range' )# Select 0 means the current directory and the list is not changed. Otherwise# the selected directory is moved to the top of the list. This can be done by# either rotating the whole list until the selection is at the head or pop it# and insert it to 0. It isn't obvious which would be better for the user but# since pop-insert is simpler, it is used.if inp > 0 :dlist.insert( 0, dlist.pop( inp ))
elif cmd == -1 : # -1: dir is the requested new directory.# If it is already in the list then remove it before inserting it at the head.# This takes care of both the common case of it having been recently visited# and the less common case of user mistakenly requesting current, in which# case it is already at the head. Deleting and putting it back is a trivial# inefficiency.try:dlist.pop( indexNoCase( dlist, dir ))except ValueError :passdlist = dlist[:9] # Control list length by removing older dirs (should be# no more than one).dlist.insert( 0, dir )
fo.truncate( 0 )if cmd != 0 : # C: clear the listfo.writelines( dlist )
fo.close()exit(0)