53
src_file = os.path.join( self.src_dir, rel_file )
54
dst_file = os.path.join( self.dst_dir, rel_file )
56
src_type = Walker.get_file_type( src_file )
57
dst_type = Walker.get_file_type( dst_file )
59
recurse = self.process(
60
rel_file, src_file, src_type, dst_file, dst_type )
53
src = Walker.File( os.path.join( self.src_dir, rel_file ) )
54
dst = Walker.File( os.path.join( self.dst_dir, rel_file ) )
57
recurse = self.process( rel_file, src, dst )
62
59
# Set up skipping, as required. Note that we don't check to see if
63
60
# we're dealing with a directory here. We can't, because we've no
67
64
# it shouldn't matter. We adding an os.pathset to the end of the
68
65
# filename, so it wuill only match files that are descendents of a
69
66
# directory with the name of this file.
70
if not recurse: skip = rel_file + os.pathsep
67
if not recurse: skip = rel_file + os.sep
72
def __init__( self, full_file ):
74
if not os.path.exists( self.file ):
76
elif os.path.isfile( self.file ):
78
elif os.path.isdir( self.file ):
82
if os.path.islink( self.file ):
83
self.link_type = self.type
86
self.link_type = False
88
def get_type_name( self ):
89
if self.type == 'l': return 'symlink'
90
elif self.type == 'f': return 'file'
91
elif self.type == 'd': return 'directory'
92
elif self.type == '_': return 'missing'
93
else: return 'unknown'
97
if( self.link_type ): type += '/' + self.link_type
98
return 'File( %s (%s) )' % ( self.file, type )
74
def get_file_type( full_file ):
75
"""Returns the type of a given file, at the time of calling. Types are 'd' for
76
directory, 'f' for file, 'l' for symlink, '_' for missing and '?' for
102
def generate_walk_list( rel_file = '', full_dir = None ):
103
"""Returns a list of files and directories in full_dir, specified as relative
104
files (relative to full_dir), breadth first.
80
if not os.path.lexists( full_file ):
82
elif os.path.islink( full_file ):
84
elif os.path.isfile( full_file ):
107
# default place to walk
108
if full_dir is None: full_dir = the.repo.full_dir
111
if rel_file in { '.bzr', '.bzrignore', '.stdhome', '.stdhomerc' }:
114
full_file = os.path.join( full_dir, rel_file )
116
# files and links are returned
117
if os.path.isfile( full_file ) or os.path.islink( full_file ):
120
# directories are returned and recursed in to
86
121
elif os.path.isdir( full_file ):
122
ret = [ rel_file ] if rel_file != '' else []
123
for file in os.listdir( full_file ):
124
ret.extend( Walker.generate_walk_list(
125
os.path.join( rel_file, file ), full_dir ) )
128
# other kinds are invalid
93
def generate_walk_list( full_dir, rel_dir = '' ):
94
"""Returns a list of files and directories in full_dir, specified as relative
95
files (relative to full_dir), breadth first.
100
for file in os.listdir( os.path.join( full_dir, rel_dir ) ):
102
rel_file = os.path.join( rel_dir, file )
103
if rel_file == '.bzr': continue
105
full_file = os.path.join( full_dir, rel_file )
107
if os.path.isfile( full_file ) or os.path.islink( full_file ):
108
ret.append( rel_file )
109
elif os.path.isdir( full_file ):
110
ret.append( rel_file )
111
ret.extend( generate_file_list( full_dir, rel_file ) )
114
'unknown/exotic file: %s' % full_file )
120
def name_of_type( type ):
121
if type == 'd': return 'a directory'
122
elif type == 'f': return 'a file'
123
elif type == 'l': return 'a symlink'
124
elif type == '_': return 'missing'
125
else: return 'something exotic'
131
'unknown/exotic file: %s' % full_file )
added
removed
lib/stdhome/walker/walker.py
