[Python] Read a FASTA file using Biopython

read_fasta.py

#!/usr/bin/env python

import bz2
import gzip

from Bio import SeqIO


def read_fasta(path):
    if path.endswith('.gz'):
        f = gzip.open(path, 'rt')
    elif path.endswith('.bz2'):
        f = bz2.open(path, 'rt')
    else:
        f = open(path, 'r')
    records = SeqIO.to_dict(SeqIO.parse(f, 'fasta'))
    f.close()
    return records