Search in sources :

Example 1 with BaseResponse

use of com.spectralogic.ds3autogen.python.model.response.BaseResponse in project ds3_autogen by SpectraLogic.

the class PythonCodeGenerator method generateCommands.

/**
     * Generates the ds3.py python code which includes: request handlers, response handlers,
     * response payload descriptors, and the client
     */
@Override
public void generateCommands(final ImmutableList<Ds3Request> ds3Requests, final ImmutableMap<String, Ds3Type> typeMap, final Ds3DocSpec docSpec) throws IOException, TemplateException {
    if (isEmpty(ds3Requests)) {
        LOG.info("There are no requests to generate");
        return;
    }
    final ImmutableList<BaseRequest> baseRequests = toRequestModelList(ds3Requests, docSpec);
    final ImmutableList<BaseResponse> baseResponses = toResponseModelList(ds3Requests);
    final ImmutableList<TypeDescriptor> baseTypes = toTypeDescriptorList(typeMap);
    final ImmutableList<BaseClient> clientCommands = toClientCommands(ds3Requests, docSpec);
    final CommandSet commandSet = new CommandSet(baseRequests, baseResponses, baseTypes, clientCommands);
    final Template tmpl = getCommandTemplate(config);
    final Path path = toBaseProjectPath("ds3.py");
    LOG.info("Getting OutputStream for file: " + path.toString());
    try (final OutputStream outStream = fileUtils.getOutputFile(path);
        final Writer writer = new OutputStreamWriter(outStream)) {
        tmpl.process(commandSet, writer);
    }
}
Also used : Path(java.nio.file.Path) OutputStream(java.io.OutputStream) BaseClient(com.spectralogic.ds3autogen.python.model.client.BaseClient) BaseResponse(com.spectralogic.ds3autogen.python.model.response.BaseResponse) TypeDescriptor(com.spectralogic.ds3autogen.python.model.type.TypeDescriptor) BaseRequest(com.spectralogic.ds3autogen.python.model.request.BaseRequest) CommandSet(com.spectralogic.ds3autogen.python.model.commands.CommandSet) OutputStreamWriter(java.io.OutputStreamWriter) OutputStreamWriter(java.io.OutputStreamWriter) Writer(java.io.Writer)

Example 2 with BaseResponse

use of com.spectralogic.ds3autogen.python.model.response.BaseResponse in project ds3_autogen by SpectraLogic.

the class PythonCodeGenerator_Test method toResponseModel_Test.

@Test
public void toResponseModel_Test() {
    final Ds3Request request = getBucketRequest();
    final BaseResponse result = toResponseModel(request);
    assertThat(result.getName(), is("GetBucketResponseHandler"));
    assertThat(result.getCodes().size(), is(1));
    final String expectedParseResponse = "if self.response.status == 200:\n" + "      self.result = parseModel(xmldom.fromstring(response.read()), ListBucketResult())";
    assertThat(result.getParseResponseCode(), is(expectedParseResponse));
}
Also used : BaseResponse(com.spectralogic.ds3autogen.python.model.response.BaseResponse) Ds3Request(com.spectralogic.ds3autogen.api.models.apispec.Ds3Request) Test(org.junit.Test)

Aggregations

BaseResponse (com.spectralogic.ds3autogen.python.model.response.BaseResponse)2 Ds3Request (com.spectralogic.ds3autogen.api.models.apispec.Ds3Request)1 BaseClient (com.spectralogic.ds3autogen.python.model.client.BaseClient)1 CommandSet (com.spectralogic.ds3autogen.python.model.commands.CommandSet)1 BaseRequest (com.spectralogic.ds3autogen.python.model.request.BaseRequest)1 TypeDescriptor (com.spectralogic.ds3autogen.python.model.type.TypeDescriptor)1 OutputStream (java.io.OutputStream)1 OutputStreamWriter (java.io.OutputStreamWriter)1 Writer (java.io.Writer)1 Path (java.nio.file.Path)1 Test (org.junit.Test)1