U e{j@sdZddlZddlZddlZddlZddlmZddlmZe dZ d Z ddZ d d Zd d Zd dZddZddZdddZdS)a ``fitscheck`` is a command line script based on astropy.io.fits 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 astropy.io.fits 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. Add checksums:: $ fitscheck --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. Only check and fix compliance problems, ignoring checksums:: $ fitscheck --checksum none --compliance --write *.fits 5. Verify standard interoperable checksums:: $ fitscheck *.fits 6. Delete checksum keywords:: $ fitscheck --checksum remove --write *.fits N) __version__)fitsZ fitscheckag 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. This script is part of the Astropy package. See https://docs.astropy.org/en/latest/io/fits/usage/scripts.html#module-astropy.io.fits.scripts.fitscheck for further documentation. cCst|sdg}tjttjd}|jdddtd|jddd d d |jd d ddddgddd|jddddddd|jddddddd|jdd d!d"ddd|jd#d$d%d&ddd|jd'd(d)d*ddd||atj dkrdt_ n*tj dkrd+t_ ntj dkrd+t_ d+t_ tj S),Nz-h) descriptionformatter_classz --versionversionz %(prog)s )actionr fits_filesfile+z.fits files to process.)metavarnargshelpz-kz --checksum checksum_kindstandardremovenonez6Choose FITS checksum mode or none. Defaults standard.)destchoicesr defaultz-wz--write write_filez6Write out file checksums and/or FITS compliance fixes.F store_true)rr rrz-fz--forceforcez1Do file update even if original checksum was bad.z-cz --compliance compliancez-Do FITS compliance checking; fix if possible.z-iz--ignore-missingignore_missingzIgnore missing checksums.z-vz --verboseverbosezGenerate extra output.T) lenargparseArgumentParser DESCRIPTIONRawDescriptionHelpFormatter add_argumentr parse_argsOPTIONSrrrr)argsparserr%`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/fits/scripts/fitscheck.pyhandle_optionsBs         r'cCsPtjtjrttjn ttjt }| t dt |dS)Nz %(message)s) loghandlersclearr"rsetLevelloggingINFOWARNING StreamHandler setFormatter Formatter addHandler)handlerr%r%r& setup_loggings  r4c Cstjdd}tdtj|tjd}t|D]r\}}tjs2|j srt d ||W5QRW5QRdS|j s2t d ||W5QRW5QRdSq2W5QRXW5QRX|D].}t|jdrt d |t|jdSqt d |d S) zR Prints a message if any HDU in `filename` has a bad checksum or datasum. T)recordalways)checksumz-MISSING {!r} .. Checksum not found in HDU #{}z,MISSING {!r} .. Datasum not found in HDU #{})zChecksum verification failedzDatasum verification failedz BAD %r %szOK r)warningscatch_warnings simplefilterropenr"r enumeraterZ _checksumr(warningformatZ_datasumstrmessage startswithinfo)filenameZwlisthdulistiZhduwr%r%r&verify_checksumss: 0 rHc Csxt|d}z|dWnNtjk rh}z.td|t|ddWYW5QRdSd}~XYnXW5QRXdS)z#Check for FITS standard compliance. exceptionzNONCOMPLIANT %r .. %s  r8Nr)rr<verifyZ VerifyErrorr(r>r@replace)rDrEexcr%r%r&verify_compliances ,rOc CsZtjr dnd}t:ttj|dtjdd}|j|dW5QRXW5QRXdS)z Sets the ``CHECKSUM`` and ``DATASUM`` keywords for each HDU of `filename`. Also updates fixes standards violations if possible and requested. Z silentfixignoreTupdate)Zdo_not_scale_image_datar7mode) output_verifyN) r"rr9r: resetwarningsrr<rflush)rDrSrEr%r%r&rQs rQc CszBt|}tjrt|}nd}tjr,|dks2tjr:t|||WStk r}z t d|d|WYdSd}~XYnXdS)za Handle a single .fits file, returning the count of checksum and compliance errors. rz EXCEPTION z .. r8N) rHr"rrOrrrQ Exceptionr(error)rDZchecksum_errorsZcompliance_errorser%r%r& process_files  rYcCsVd}t|ptjdd}t|D]}|t|7}q$|rJt|dtt|S)z Processes command line parameters into options and files, then checks or update FITS DATASUM and CHECKSUM keywords for the specified files. rr8Nz errors) r'sysargvr4rYr(r>intbool)r#errorsrrDr%r%r&mainsr_)N)__doc__rr,rZr9ZastropyrZ astropy.ior getLoggerr(striprr'r4rHrOrQrYr_r%r%r%r&s)    T %