load_references.py

import os


def reference_example():
    # Load references and insert them.
    from snipper.citations import get_bibtex, get_aux #!s=a
    bibfile = "latex/library.bib"
    auxfile = 'latex/index.aux'
    bibtex = get_bibtex(bibfile)
    aux = get_aux(auxfile) #!s

    print(list(bibtex.keys())[:5])
    print(aux)

    from snipper.fix_cite import fix_aux, fix_aux_special, fix_bibtex
    from snipper.fix_s import save_s

    file = "citations.py" #!s=b
    with open(file, 'r') as f:
        lines = f.read().splitlines()
    lines = fix_aux(lines, aux=aux)
    lines = fix_aux_special(lines, aux=aux, command='\\nref', bibref='herlau')
    lines = fix_bibtex(lines, bibtex=bibtex)
    with open('output/citations.py', 'w') as f:
        f.write("\n".join(lines)) #!s=b
    # save_s(lines, output_dir="./output", file_path=file)

    with open(__file__, 'r') as f:
        save_s(f.read().splitlines(), output_dir="./output", file_path=os.path.basename(__file__) )

if __name__ == "__main__":
    reference_example()