clean.py

#!/usr/bin/python

import os
import sys
import argparse
import shutil


def main():

    parser = argparse.ArgumentParser()
    parser.add_argument("--rebuild", dest = "rebuild", action = "store_true")
    parser.add_argument("--no-rebuild", dest = "rebuild", action = "store_false")
    parser.set_defaults(rebuild = False)

    args = parser.parse_args()
    rebuild = args.rebuild

    # Recursivly clean all the binary directories.
    list_clean = list()
    path_cur = os.getcwd()
    for path_dir, list_dir, list_file in os.walk(path_cur):
        if path_dir.endswith("build"):
            list_clean.append(path_dir)
        elif path_dir.endswith("lib"):
            list_clean.append(path_dir)
        elif path_dir.endswith("bin"):
            list_clean.append(path_dir)
        elif path_dir.endswith("export"):
            list_clean.append(path_dir)
    for path_clean in list_clean:
        if os.path.isdir(path_clean) == True:
            shutil.rmtree(path_clean)

    # Create the build folder if necessary.
    if rebuild == False:
        return
    for path_dir, list_dir, list_file in os.walk(path_cur):
        if path_dir.endswith("src"):
            path_build = path_dir.replace("src", "build")
            if os.path.isdir(path_build) == False:
                os.makedirs(path_build)
        elif path_dir.endswith("plugin"):
            path_build = os.path.join(path_dir, "build")
            if os.path.isdir(path_build) == False:
                os.makedirs(path_build)
        elif path_dir.endswith("engine"):
            path_build = os.path.join(path_dir, "build")
            if os.path.isdir(path_build) == False:
                os.makedirs(path_build)
        elif path_dir.endswith("MeltingPot"):
            path_build = os.path.join(path_dir, "build")
            if os.path.isdir(path_build) == False:
                os.makedirs(path_build)


if __name__ == "__main__":
    main()