fs_utils.py

"""
These are filesytem based utilities to help with housekeeping.
"""

import os

def LoadPrograms(**kwargs):
    """
    Iterate through the python programs in a folder.
    """
    path = kwargs['path']
    for file in os.listdir(path):
        if file.endswith(".py"):
            with open(os.path.join(path, file)) as f:
                code = f.read()
                yield file, code

def LoadProgramVisitor(path, func):
    for pname, code in LoadPrograms(path=path):
        func(pname, code)

def buildLoader(client, Program, ndb, user_id, folder_id, modDBFunctions):

    def loader(pname, code):
        code = code.replace('from vpython import *', 'Web VPython 3.2\n')
        modDBFunctions.SetProgramSource(ndb, Program, user_id, folder_id, pname, code)

    wrapped_loader = modDBFunctions.with_context(client)(loader)

    return wrapped_loader

if __name__ == '__main__':
    LoadProgramVisitor(path='/Users/steve/Development/instructormi/assets/lecture-demo-programs/PUBLIC', func=lambda pname, code: print(pname))