delete_class.py

#!/usr/bin/env python
""" Utility for deleting a class file """

import sys
import os
import argparse
import datetime
import re
from cmakelists_utils import *


    
#======================================================================
def delete_one(oldfilename):
    cmd = "git rm " + oldfilename
    print "Running:", cmd
    os.system(cmd)

#======================================================================
def delete_all(subproject, classname, args):
    
    # Directory at base of subproject
    basedir = os.path.join(os.path.curdir, "Framework/" + subproject)
    
    headerfile = os.path.join(basedir, "inc/Mantid" + subproject + "/" + args.source_subfolder + classname + ".h")
    sourcefile = os.path.join(basedir, "src/" + args.source_subfolder + classname + ".cpp")
    testfile = os.path.join(basedir, "test/" + classname + "Test.h")
      
    print
    if args.header:
        delete_one(headerfile)
    if args.cpp:
        delete_one(sourcefile)
    if args.test:
        delete_one(testfile)
    
    # Insert into the cmake list
    remove_from_cmake(subproject, classname, args, args.source_subfolder)
    
    print "   Files were removed to Framework/%s/CMakeLists.txt !" % subproject
    print 



#======================================================================
if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Utility to delete a Mantid class from a project. ' 
                                     'Please note, you may still have more fixes to do to get compilation!')
    
    parser.add_argument('subproject', metavar='SUBPROJECT', type=str,
                        help='The subproject under Framework/; e.g. Kernel')
    parser.add_argument('classname', metavar='CLASSNAME', type=str,
                        help='Name of the class to create')
    parser.add_argument('--subfolder', dest='source_subfolder', 
                        default="",
                        help='The source is in a subfolder below the main part of the project, e.g. Geometry/Instrument.')
    
    parser.add_argument('--no-header', dest='header', action='store_const',
                        const=False, default=True,
                        help="Don't delete the header file")
    parser.add_argument('--no-test', dest='test', action='store_const',
                        const=False, default=True,
                        help="Don't delete the test file")
    parser.add_argument('--no-cpp', dest='cpp', action='store_const',
                        const=False, default=True,
                        help="Don't delete the cpp file")
    
    parser.add_argument('--project', dest='project', 
                    default="Framework",
                    help='The project in which this goes. Default: Framework. Can be MantidQt, Vates')
     
    args = parser.parse_args()
    subproject = args.subproject
    classname = args.classname
    
    # Make sure the subfolders end with a /
    if args.source_subfolder != "":
        if args.source_subfolder[-1:] != "/":
            args.source_subfolder += "/"
    
    delete_all(subproject, classname,  args)