import model.entity
import basic.constants as B

def get_entity_object(job, name, args):
    if name in B.SUBJECT_STEPS:
        entity = getStep(job)
    elif name in B.SUBJECT_STORIES:
        entity = getStory(job)
    elif name in B.SUBJECT_VARIANTS:
        entity = getVariant(job)
    elif name in B.SUBJECT_DATATABLES:
        entity = getDatatable(job)
    elif name in B.SUBJECT_USECASES:
        entity = getUsecase(job)
    elif name in B.SUBJECT_APPS:
        entity = getApplication(job)
    elif name in B.SUBJECT_COMPS:
        entity = getComponent(job)
    elif name in B.SUBJECT_TESTCASES:
        entity = getTestcase(job)
    elif name in B.SUBJECT_TESTSUITES:
        entity = getTestsuite(job)
    elif name in B.SUBJECT_TESTPLAN:
        entity = getTestplan(job)
    else:
        return None
    entity.setAttributes(job, args, name, entity.getFieldList(), entity.getNodeList(), entity.getSubtableList())
    return entity

def getEnvironment(job=None):
    import model.environment
    return model.environment.Environment(job)

def getApplication(job=None, name=""):
    import model.application
    return model.application.Application()

def getComponent(job=None, name=""):
    import model.component
    return model.component.Component(job)

def getTestplan(job=None, project="", application="", name=""):
    import model.testplan
    return model.testplan.Testplan(job, project)

def getTestsuite(job=None, project="", application="", name=""):
    import model.testsuite
    return model.testsuite.Testsuite(job, project)

def getTestcase(job=None, project="", application="", name=""):
    import model.testcase
    return model.testcase.Testcase(job, project, name=name)

def getStep(job=None, project="", name=""):
    import model.step
    return model.step.Step(job, project, name)

def getStory(job=None, project="", name=""):
    import model.story
    return model.story.Story(job, project, name)

def getUsecase(job=None, project="", name=""):
    import model.usecase
    return model.usecase.Usecase(job, project, name)

def getVariant(job=None, project="", name=""):
    import model.variant
    return model.variant.Variant(job, project, name)

def getDatatable(job=None, project="", name=""):
    import model.datatable
    return model.datatable.Datatable(job, project, name)