void updateArgsBeforeExport()

in src/main/java/com/amazonaws/services/neptune/profiles/neptune_ml/NeptuneMLSourceDataModel.java [34:64]

• 24 lines of code
• 11 McCabe index (conditional complexity)

        void updateArgsBeforeExport(Args args, Collection<TrainingDataWriterConfigV2> trainingJobWriterConfigCollection) {
            if (!args.contains("--exclude-type-definitions") && !args.contains("export-pg-from-queries")) {
                // export-pg-from-queries does not support --exclude-type-definitions, it uses --include-type-definitions instead.
                args.addFlag("--exclude-type-definitions");
            }

            if (args.contains("--include-type-definitions")) {
                args.removeOptions("--include-type-definitions");
            }

            if (args.contains("--edge-label-strategy", EdgeLabelStrategy.edgeLabelsOnly.name())) {
                args.removeOptions("--edge-label-strategy");
            }

            if (!args.contains("--edge-label-strategy", EdgeLabelStrategy.edgeAndVertexLabels.name())) {
                args.addOption("--edge-label-strategy", EdgeLabelStrategy.edgeAndVertexLabels.name());
            }

            if (!args.contains("--merge-files")) {
                args.addFlag("--merge-files");
            }

            if (args.contains("export-pg") &&
                    args.containsAny("--config", "--filter", "-c", "--config-file", "--filter-config-file")) {
                args.replace("export-pg", "export-pg-from-config");
            }

            if (args.contains("export-pg-from-queries") && !args.contains("--structured-output")) {
                args.addFlag("--structured-output");
            }
        }