nbpyfile_cleaner

nb_cleaner.py

__author__ = 'szeitlin'

#script to remove # In[1]: etc. from IPython notebook exported files

#import os
import re
#import sys



def nb_cleaner(nbpyfile, nonnbfile="newfile.py"):
    '''

    Helper function for converting python files exported from IPython notebooks into actual programs.

    Takes an IPython notebook exported 'python' file, and a new file name.
    These files suffer from retaining useless codecells throughout, designated as "# In[1]:".

    Iterates through the file exported from the notebook, compares to regex and skips lines we don't want,
    copying everything else to a new file.

    :return: cleaner copy of the file.

    '''

    with open(nbpyfile, 'r') as old_file:
        p = re.compile("^# In")

        for line in old_file:
            if p.match(line):
                continue
            else:
                with open(nonnbfile, 'a') as new_file:
                    new_file.write(str(line))
    return 


#nbpyfile = "bike1year_viz2.py"

nb_cleaner(nbpyfile, nonnbfile)

Helper function for converting python files exported from IPython notebooks into actual programs.

Takes an IPython notebook exported 'python' file, and a new file name.
Removes useless codecells throughout, designated as "# In[1]:"
Returns a cleaner copy of the python file.