1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#!/usr/bin/env python
"""This is for cleaning up binary files improperly added to CVS.
This script scans the given path to find binary files;
checks with CVS to see if the sticky options are set to -kb;
finally if sticky options are not -kb then uses 'cvs admin' to
set the -kb option.
This script ignores CVS directories, symbolic links,
and files not known under CVS control (cvs status is 'Unknown').
Run this on a CHECKED OUT module sandbox, not on the repository itself.
After if fixes the sticky options on any files you should manually do
a 'cvs commit' to accept the changes. Then be sure to have all users do
a 'cvs up -A' to update the Sticky Option status.
Noah Spurrier
20030426
"""
import os, sys, time
import pexpect
VERBOSE = 1
def is_binary (filename):
"""Assume that any file with a character where
the 8th bit is set is binary.
"""
fin = open(filename, 'rb')
wholething = fin.read()
fin.close()
for c in wholething:
if ord(c) & 0x80:
return 1
return 0
def is_kb_sticky (filename):
"""This checks if 'cvs status' reports '-kb' for Sticky options.
If the Sticky Option status is '-ks' then this returns 1.
If the status is 'Unknown' then it returns 1.
Otherwise 0 is returned.
"""
try:
s = pexpect.spawn ('cvs status %s' % filename)
i = s.expect (['Sticky Options:\s*(.*)\r\n', 'Status: Unknown'])
if i==1 and VERBOSE:
print 'File not part of CVS repository:', filename
return 1 # Pretend it's OK.
if s.match.group(1) == '-kb':
return 1
s = None
except:
print 'Something went wrong trying to run external cvs command.'
print ' cvs status %s' % filename
print 'The cvs command returned:'
print s.before
return 0
def cvs_admin_kb (filename):
"""This uses 'cvs admin' to set the '-kb' sticky option.
"""
s = pexpect.run ('cvs admin -kb %s' % filename)
# There is a timing issue. If I run 'cvs admin' too quickly
# cvs sometimes has trouble obtaining the directory lock.
time.sleep(1)
def walk_and_clean_cvs_binaries (arg, dirname, names):
"""This contains the logic for processing files.
This is the os.path.walk callback.
This skips dirnames that end in CVS.
"""
if len(dirname)>3 and dirname[-3:]=='CVS':
return
for n in names:
fullpath = os.path.join (dirname, n)
if os.path.isdir(fullpath) or os.path.islink(fullpath):
continue
if is_binary(fullpath):
if not is_kb_sticky (fullpath):
if VERBOSE: print fullpath
cvs_admin_kb (fullpath)
def main ():
if len(sys.argv) == 1:
root = '.'
else:
root = sys.argv[1]
os.path.walk (root, walk_and_clean_cvs_binaries, None)
if __name__ == '__main__':
main ()
|
