/usr/lib/python2.7/dist-packages/pyfits/scripts/fitscheck.py is in python-pyfits 1:3.2-1build2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 | """
fitscheck is a command line script based on pyfits for verifying
and updating the CHECKSUM and DATASUM keywords of .fits files.
Fitscheck can also detect and often fix other FITS standards
violations. fitscheck facilitates re-writing the non-standard
checksums originally generated by pyfits with standard checksums which
will interoperate with cfitsio.
fitscheck will refuse to write new checksums if the checksum keywords
are missing or their values are bad. Use --force to write new
checksums regardless of whether or not they currently exist or pass.
Use --ignore-missing to tolerate missing checksum keywords without
comment.
Example uses of fitscheck:
1. Verify and update checksums, tolerating non-standard checksums, updating to
standard checksum::
$ fitscheck --checksum either --write *.fits
2. Write new checksums, even if existing checksums are bad or missing::
$ fitscheck --write --force *.fits
3. Verify standard checksums and FITS compliance without changing the files::
$ fitscheck --compliance *.fits
4. Verify original nonstandard checksums only::
$ fitscheck --checksum nonstandard *.fits
5. Only check and fix compliance problems, ignoring checksums::
$ fitscheck --checksum none --compliance --write *.fits
6. Verify standard interoperable checksums::
$ fitscheck *.fits
7. Delete checksum keywords::
$ fitscheck --checksum none --write *.fits
"""
import logging
import optparse
import sys
import textwrap
import warnings
import pyfits
log = logging.getLogger('fitscheck')
warnings.filterwarnings('error', message='Checksum verification failed')
warnings.filterwarnings('error', message='Datasum verification failed')
warnings.filterwarnings('ignore', message='Overwriting existing file')
def handle_options(args):
if not len(args):
args = ['-h']
parser = optparse.OptionParser(usage=textwrap.dedent("""
fitscheck [options] <.fits files...>
.e.g. fitscheck example.fits
Verifies and optionally re-writes the CHECKSUM and DATASUM keywords
for a .fits file.
Optionally detects and fixes FITS standard compliance problems.
""".strip()))
parser.add_option(
'-k', '--checksum', dest='checksum_kind',
type='choice', choices=['standard', 'nonstandard', 'either', 'none'],
help='Choose FITS checksum mode or none. Defaults standard.',
default='standard', metavar='[standard | nonstandard | either | none]')
parser.add_option(
'-w', '--write', dest='write_file',
help='Write out file checksums and/or FITS compliance fixes.',
default=False, action='store_true')
parser.add_option(
'-f', '--force', dest='force',
help='Do file update even if original checksum was bad.',
default=False, action='store_true')
parser.add_option(
'-c', '--compliance', dest='compliance',
help='Do FITS compliance checking; fix if possible.',
default=False, action='store_true')
parser.add_option(
'-i', '--ignore-missing', dest='ignore_missing',
help='Ignore missing checksums.',
default=False, action='store_true')
parser.add_option(
'-v', '--verbose', dest='verbose', help='Generate extra output.',
default=False, action='store_true')
global OPTIONS
OPTIONS, fits_files = parser.parse_args(args)
if OPTIONS.checksum_kind == 'none':
OPTIONS.checksum_kind = False
return fits_files
def setup_logging():
if OPTIONS.verbose:
log.setLevel(logging.INFO)
else:
log.setLevel(logging.WARNING)
handler = logging.StreamHandler()
handler.setFormatter(logging.Formatter('%(message)s'))
log.addHandler(handler)
def verify_checksums(filename):
"""
Prints a message if any HDU in `filename` has a bad checksum or datasum.
"""
errors = 0
try:
hdulist = pyfits.open(filename, checksum=OPTIONS.checksum_kind)
except UserWarning, w:
remainder = '.. ' + ' '.join(str(w).split(' ')[1:]).strip()
# if "Checksum" in str(w) or "Datasum" in str(w):
log.warn('BAD %r %s' % (filename, remainder))
return 1
if not OPTIONS.ignore_missing:
for i, hdu in enumerate(hdulist):
if not hdu._checksum:
log.warn('MISSING %r .. Checksum not found in HDU #%d' %
(filename, i))
return 1
if not hdu._datasum:
log.warn('MISSING %r .. Datasum not found in HDU #%d' %
(filename, i))
return 1
if not errors:
log.info('OK %r' % filename)
return errors
def verify_compliance(filename):
"""Check for FITS standard compliance."""
hdulist = pyfits.open(filename)
try:
hdulist.verify('exception')
except pyfits.VerifyError, e:
log.warn('NONCOMPLIANT %r .. %s' %
(filename), str(e).replace('\n', ' '))
return 1
return 0
def update(filename):
"""
Sets the ``CHECKSUM`` and ``DATASUM`` keywords for each HDU of `filename`.
Also updates fixes standards violations if possible and requested.
"""
hdulist = pyfits.open(filename)
try:
output_verify = 'silentfix' if OPTIONS.compliance else 'ignore'
hdulist.writeto(filename, checksum=OPTIONS.checksum_kind, clobber=True,
output_verify=output_verify)
except pyfits.VerifyError:
pass # unfixable errors already noted during verification phase
finally:
hdulist.close()
def process_file(filename):
"""
Handle a single .fits file, returning the count of checksum and compliance
errors.
"""
try:
checksum_errors = verify_checksums(filename)
if OPTIONS.compliance:
compliance_errors = verify_compliance(filename)
else:
compliance_errors = 0
if OPTIONS.write_file and checksum_errors == 0 or OPTIONS.force:
update(filename)
return checksum_errors + compliance_errors
except Exception, e:
log.error('EXCEPTION %r .. %s' % (filename, e))
return 1
def main():
"""
Processes command line parameters into options and files, then checks
or update FITS DATASUM and CHECKSUM keywords for the specified files.
"""
errors = 0
fits_files = handle_options(sys.argv[1:])
setup_logging()
for filename in fits_files:
errors += process_file(filename)
if errors:
log.warn('%d errors' % errors)
return int(bool(errors))
|