Updates the b-factor range of models from 0..1 to 0..100

fix_bfactors.py

from Bio.PDB import *

from argparse import ArgumentParser


def parse_arguments():
    parser = ArgumentParser(description='Fix b-factor range for coloring')
    parser.add_argument('--pdb',
                        help='Model file',
                        required=True)
    parser.add_argument('--output',
                        help='Output',
                        required=True)                   
    args = parser.parse_args()
    return args


def main():
    args = parse_arguments()
    parser = PDBParser()
    structure = parser.get_structure('X', args.pdb)[0]

    for chain in structure:
        for residue in chain:
            for atom in residue:
                atom.set_bfactor(atom.get_bfactor() * 100)

    io = PDBIO()

    io.set_structure(structure)

    io.save(args.output)    


if __name__ == "__main__":
    main()