#!/usr/bin/env python

from dm.cat_web_service.api.datasetRestApi import DatasetRestApi
from dm.common.exceptions.invalidRequest import InvalidRequest
from catWebServiceSessionCli import CatWebServiceSessionCli

class UpdateExperimentDatasetCli(CatWebServiceSessionCli):
    def __init__(self):
        CatWebServiceSessionCli.__init__(self, validArgCount=self.ANY_NUMBER_OF_POSITIONAL_ARGS)
        self.addOption('', '--dataset', dest='datasetName', help='Dataset name.')
        self.addOption('', '--experiment', dest='experimentName', help='Experiment name.')

    def checkArgs(self):
        if self.options.experimentName is None:
            raise InvalidRequest('Experiment name must be provided.')
        if self.options.datasetName is None:
            raise InvalidRequest('Dataset name must be provided.')

    def getExperimentName(self):
        return self.options.experimentName

    def getDatasetName(self):
        return self.options.datasetName

    def runCommand(self):
        self.parseArgs(usage="""
    dm-update-experiment-dataset --dataset=DATASETNAME 
        --experiment=EXPERIMENTNAME 
        [key1:value1, key2:value2, ...]

Description:
    Updates experiment dataset in the metadata catalog. All provided key/value 
    pairs are interpreted as dataset metadata, and will be merged with existing
    metadata, overwriting values for existing keys, and adding values for
    new keys.
        """)
        self.checkArgs()
        api = DatasetRestApi(self.getLoginUsername(), self.getLoginPassword(), self.getServiceHost(), self.getServicePort(), self.getServiceProtocol())
        datasetInfo = self.splitArgsIntoDict()
        datasetInfo['experimentName'] = self.getExperimentName()
        datasetInfo['datasetName'] = self.getDatasetName()
        datasetMetadata = api.updateExperimentDataset(datasetInfo)
        print datasetMetadata.getDisplayString(self.getDisplayKeys(), self.getDisplayFormat())

#######################################################################
# Run command.
if __name__ == '__main__':
    cli = UpdateExperimentDatasetCli()
    cli.run()