You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
"""
|
|
Code maintenance script modified from PyMC
|
|
"""
|
|
|
|
#!/usr/bin/env python
|
|
import sys
|
|
import os
|
|
|
|
# This is a function, not a test case, because it has to be run from inside
|
|
# the source tree to work well.
|
|
|
|
mod_strs = ['IPython', 'pylab', 'matplotlib', 'scipy','Pdb']
|
|
|
|
dep_files = {}
|
|
for mod_str in mod_strs:
|
|
dep_files[mod_str] = []
|
|
|
|
def remove_whitespace(fname):
|
|
# Remove trailing whitespace
|
|
fd = open(fname,mode='U') # open in universal newline mode
|
|
lines = []
|
|
for line in fd.readlines():
|
|
lines.append( line.rstrip() )
|
|
fd.close()
|
|
|
|
fd = open(fname,mode='w')
|
|
fd.seek(0)
|
|
for line in lines:
|
|
fd.write(line+'\n')
|
|
fd.close()
|
|
# print 'Removed whitespace from %s'%fname
|
|
|
|
def find_whitespace(fname):
|
|
fd = open(fname, mode='U')
|
|
for line in fd.readlines():
|
|
#print repr(line)
|
|
if ' \n' in line:
|
|
print fname
|
|
break
|
|
|
|
|
|
# print
|
|
print_only = True
|
|
|
|
# ====================
|
|
# = Strip whitespace =
|
|
# ====================
|
|
for dirname, dirs, files in os.walk('.'):
|
|
if dirname[1:].find('.')==-1:
|
|
# print dirname
|
|
for fname in files:
|
|
if fname[-2:] in ['c', 'f'] or fname[-3:]=='.py' or fname[-4:] in ['.pyx', '.txt', '.tex', '.sty', '.cls'] or fname.find('.')==-1:
|
|
# print fname
|
|
if print_only:
|
|
find_whitespace(dirname + '/' + fname)
|
|
else:
|
|
remove_whitespace(dirname + '/' + fname)
|
|
|
|
|
|
"""
|
|
|
|
# ==========================
|
|
# = Check for dependencies =
|
|
# ==========================
|
|
for dirname, dirs, files in os.walk('pymc'):
|
|
for fname in files:
|
|
if fname[-3:]=='.py' or fname[-4:]=='.pyx':
|
|
if dirname.find('sandbox')==-1 and fname != 'test_dependencies.py'\
|
|
and dirname.find('examples')==-1:
|
|
for mod_str in mod_strs:
|
|
if file(dirname+'/'+fname).read().find(mod_str)>=0:
|
|
dep_files[mod_str].append(dirname+'/'+fname)
|
|
|
|
|
|
print 'Instances of optional dependencies found are:'
|
|
for mod_str in mod_strs:
|
|
print '\t'+mod_str+':'
|
|
for fname in dep_files[mod_str]:
|
|
print '\t\t'+fname
|
|
if len(dep_files['Pdb'])>0:
|
|
raise ValueError, 'Looks like Pdb was not commented out in '+', '.join(dep_files[mod_str])
|
|
|
|
|
|
"""
|