def remove_file(fname, verbose=True):
success = False
if os.path.isfile(fname):
if verbose: print('-> Removing ' + fname + '...', end='')
if normcase(realpath(fname)) != normcase(realpath(__file__)):
try:
os.remove(fname)
success = True
except:
try:
## Removing files via command line sometimes helps if os.remove doesn't work:
del_script = "python -c \"import os, time; time.sleep(1); os.remove('{}');\"".format(
fname)
subprocess.Popen(del_script)
success = True
except:
warnings.warn('Unable to remove ' + fname + '...!')
success = False
else:
warnings.warn('Attempt to delete __init__ file was prevented')
success = False
if verbose:
if success:
print(' Done!')
else:
print(' FAILED!')
else:
success = True
return success
def fix_pypdf(input_fname, output_fname=None, verbose=True):
if verbose: print('Fixing ' + input_fname)
if output_fname == None:
output_fname = input_fname
pw = PyPdfFileWriter()
temp_output = io.BytesIO()
#TO-DO: byte level operations should go to classes.py
with open(input_fname, 'rb') as fr:
try:
fr.seek(-1024, 2)
last1k = fr.read()
eof_addr = last1k.rfind(b'%%EOF')
pypdf_str = last1k[eof_addr:].split()[2]
if pypdf_str == b'PyPDF':
warnings.warn(input_fname +
' is already compliant PyPDF-file, skipping!')
return
except (IndexError):
pass
pr = PdfFileReader(fr)
pw.cloneReaderDocumentRoot(pr)
pw.write(temp_output)
do_write = True
if output_fname == input_fname:
if not remove_file(input_fname, verbose=verbose):
output_fname = _available_fname(output_fname)
warnings.warn('File saved as ' + output_fname + 'instead')
if verbose: print('Writing ' + output_fname + '...', end='')
temp_output.seek(0)
with open(output_fname, 'wb') as fw:
fw.write(temp_output.read())
if verbose: print(' Done!')
def finalize_pypdf(pw, output_fname, pack_list, cleanup, pickle_figure,
verbose, prompt_overwrite, **kwargs):
global _py_file, _pypdf_fname
writesuccess = False
## Name the output file
if output_fname == None:
output_fname = os.path.splitext(_pypdf_fname)[
0] + '.pdf' #DO-DO: Make sure to prevent self-deletion!!
elif os.path.splitext(output_fname)[-1] == '':
output_fname += '.pdf'
elif os.path.splitext(output_fname)[1] not in [
'.pdf', '.py'
]: #TO-DO: Shouldn't this be just '.pdf'?
output_fname = os.path.splitext(output_fname)[0] + '.pdf'
warnings.warn('Invalid extension, saving as {:s}'.format(output_fname))
_py_packed_fname = output_fname[:-3] + 'py'
## Attach Python file and auxiliary files:
if pickle_figure:
if verbose: print('-> Pickling figure...')
if len(pack_list):
warnings.warn('pack_list will be ignored when pickling figure!')
fig_fname = output_fname[:-3] + 'pkl'
fig = plt.gcf()
fig.canvas = plt.figure().canvas
fdata = pickle.dumps(fig)
pw.addAttachment(fig_fname, fdata)
flines = [
b"import pypdfplot.backend.unpack",
b"import matplotlib.pyplot as plt", b"from pickle import load",
b"", b"with open('" + fig_fname.encode() + b"','rb') as f:",
b" fig = load(f)", b"", b"plt.figure(fig.number)", b"",
b"## Plot customizations go here...", b"",
b"plt.savefig('" + output_fname.encode() + b"',",
b" pack_list = ['" + fig_fname.encode() + b"'])", b"",
b'"""'
]
fdata = b'\n'.join(flines)
pw.addAttachment(_py_packed_fname, fdata)
else:
for fname in pack_list:
if verbose: print('-> Attaching ' + fname)
with open(fname, 'rb') as fa:
fdata = fa.read()
pw.addAttachment(fname, fdata)
if verbose: print('-> Attaching ' + _py_packed_fname)
fdata = _py_file + b_('\n"""')
pw.addAttachment(_py_packed_fname, fdata)
pw.setPyFile(_py_packed_fname)
pw.setPyPDFVersion(__PYPDFVERSION__)
## If the output file already exists, try to remove it:
if os.path.isfile(output_fname):
do_overwrite = False
if prompt_overwrite:
warnings.warn('Local copy of ' + output_fname +
' found\nOverwrite file? (y/n)')
yes_no = input('')
if yes_no.strip().lower()[0] == 'y':
do_overwrite = True
else:
output_fname = available_filename(output_fname)
warnings.warn(
'Publishing as {:s} instead'.format(output_fname))
else:
do_overwrite = True
if do_overwrite:
try:
os.remove(output_fname)
except:
warnings.warn('Unable to overwrite local file ' + output_fname)
output_fname = available_filename(output_fname)
warnings.warn(
'Publishing as {:s} instead'.format(output_fname))
## Write the output file:
if verbose: print('\nSaving ' + output_fname + '...\n')
with open(output_fname, 'wb+') as fw:
pw.write(fw)
write_success = True
## Remove the generating python file or create a new purely Python one:
if cleanup or not _pure_py:
if remove_file(_pypdf_fname, verbose=verbose):
warnings.warn(
_pypdf_fname +
' removed:\nSaving script in editor will make it reappear...!\n'
)
## Write the Python file if needed, and remove it if not:
if cleanup:
# Remove the local copy of the packed Python file if it happens to be present
# TO-DO: This might be overzealous, consider removing...
remove_file(_py_packed_fname, verbose=verbose)
else:
if not _pure_py:
with open(_py_packed_fname, 'wb') as fw:
fw.write(_py_file)
## Cleanup files if needed:
if cleanup:
if write_success:
if verbose and len(pack_list):
print('\nCleaning up attached files:')
for fname in pack_list:
remove_file(fname, verbose=verbose)
else:
warnings.warn(
"Files weren't packed into PyPDF file yet, aborting cleanup")