from sdcBePy.tosca.models.normativeElementCandidate import NormativeElementCandidate


def get_normative_element_candidate_list(base_file_location):
    return [
        get_data(base_file_location),
        get_capability(base_file_location),
        get_relationship(base_file_location),
        get_interface_lifecycle(base_file_location),
        get_categories(base_file_location)
    ]


def get_normative_element_with_metadata_list(base_file_location):
    return [
        get_group(base_file_location),
        get_policy(base_file_location)
    ]


def get_data(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "data-types/",
                                     "/sdc2/rest/v1/catalog/uploadType/datatypes",
                                     "dataTypes",
                                     "dataTypesZip")


def get_capability(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "capability-types/",
                                     "/sdc2/rest/v1/catalog/uploadType/capability",
                                     "capabilityTypes",
                                     "capabilityTypeZip")


def get_relationship(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "relationship-types/",
                                     "/sdc2/rest/v1/catalog/uploadType/relationship",
                                     "relationshipTypes",
                                     "relationshipTypeZip")


def get_interface_lifecycle(base_file_location="../../../import/tosca/"):
    return NormativeElementCandidate(base_file_location + "interface-lifecycle-types/",
                                     "/sdc2/rest/v1/catalog/uploadType/interfaceLifecycle",
                                     "interfaceLifecycleTypes",
                                     "interfaceLifecycleTypeZip")


def get_categories(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "categories/",
                                     "/sdc2/rest/v1/catalog/uploadType/categories",
                                     "categoryTypes",
                                     "categoriesZip")


def get_group(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "group-types/",
                                     "/sdc2/rest/v1/catalog/uploadType/grouptypes",
                                     "groupTypes",
                                     "groupTypesZip",
                                     with_metadata=True)


def get_policy(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "policy-types/",
                                     "/sdc2/rest/v1/catalog/uploadType/policytypes",
                                     "policyTypes",
                                     "policyTypesZip",
                                     with_metadata=True)


def get_annotation(base_file_location="/"):
    return NormativeElementCandidate(base_file_location + "annotation-types",
                                     "/sdc2/rest/v1/catalog/uploadType/annotationtypes",
                                     "annotationTypesZip",
                                     "annotationTypes")