use of com.github.robozonky.strategy.natural.conditions.MarketplaceFilter in project robozonky by RoboZonky.
the class ParsedStrategyTest method matchesAlsoSellFilter.
@Test
void matchesAlsoSellFilter() {
final MarketplaceFilter accepting = MarketplaceFilter.of(MarketplaceFilterCondition.alwaysAccepting());
final Collection<MarketplaceFilter> filters = Collections.singleton(accepting);
final DefaultValues v = new DefaultValues(DefaultPortfolio.PROGRESSIVE);
final FilterSupplier s = new FilterSupplier(v, Collections.emptySet(), Collections.emptySet(), filters);
final ParsedStrategy ps = new ParsedStrategy(v, Collections.emptyList(), Collections.emptyMap(), s);
final Loan l = mockLoan(200_000);
final LoanDescriptor ld = new LoanDescriptor(l);
assertThat(ps.getApplicableLoans(Collections.singleton(ld))).isEmpty();
final Participation p = mock(Participation.class);
final ParticipationDescriptor pd = new ParticipationDescriptor(p, l);
assertThat(ps.getApplicableParticipations(Collections.singleton(pd))).isEmpty();
}
use of com.github.robozonky.strategy.natural.conditions.MarketplaceFilter in project robozonky by RoboZonky.
the class SimpleStrategyService method parseOrThrow.
private static InvestmentStrategy parseOrThrow(final String strategy) {
final ImmutableConfiguration config = ImmutableConfiguration.from(strategy);
// set default values
final DefaultValues d = new DefaultValues(DefaultPortfolio.EMPTY);
d.setMinimumBalance(SimpleStrategyService.getMinimumBalance(config));
d.setTargetPortfolioSize(SimpleStrategyService.getTargetPortfolioSize(config));
d.setInvestmentShare(new DefaultInvestmentShare(SimpleStrategyService.getInvestmentShare(config)));
final LoanRatingEnumeratedCondition c = new LoanRatingEnumeratedCondition();
Stream.of(Rating.values()).filter(r -> StrategyFileProperty.REQUIRE_CONFIRMATION.getValue(r, config::getBoolean)).forEach(c::add);
d.setConfirmationCondition(c);
// assemble strategy
final Map<Rating, InvestmentSize> investmentSizes = Stream.of(Rating.values()).collect(Collectors.toMap(Function.identity(), r -> {
final int min = StrategyFileProperty.MINIMUM_LOAN_AMOUNT.getValue(r, config::getInt);
final int max = StrategyFileProperty.MAXIMUM_LOAN_AMOUNT.getValue(r, config::getInt);
return new InvestmentSize(min, max);
}));
final Collection<PortfolioShare> portfolio = Stream.of(Rating.values()).map(r -> {
final int min = SimpleStrategyService.getShare(config, StrategyFileProperty.TARGET_SHARE, r);
final int max = SimpleStrategyService.getShare(config, StrategyFileProperty.MAXIMUM_SHARE, r);
return new PortfolioShare(r, min, max);
}).collect(Collectors.toList());
// filter loans with less than minimum term
final Stream<Optional<MarketplaceFilter>> s1 = Stream.of(Rating.values()).map(r -> {
final int min = StrategyFileProperty.MINIMUM_TERM.getValue(r, config::getInt);
if (min < 2) {
// there will be no loan with term smaller than 1
return Optional.empty();
}
final MarketplaceFilterCondition f = new LoanTermCondition(0, min - 1);
return Optional.of(getRatingFilter(r, f));
});
// filter loans with more than maximum term
final Stream<Optional<MarketplaceFilter>> s2 = Stream.of(Rating.values()).map(r -> {
final int max = StrategyFileProperty.MAXIMUM_TERM.getValue(r, config::getInt);
if (max < 1) {
return Optional.empty();
}
final MarketplaceFilterCondition f = new LoanTermCondition(max + 1);
return Optional.of(getRatingFilter(r, f));
});
// filter loans with ask for less than minimum
final Stream<Optional<MarketplaceFilter>> s3 = Stream.of(Rating.values()).map(r -> {
final int min = StrategyFileProperty.MINIMUM_ASK.getValue(r, config::getInt);
if (min < 2) {
// amount smaller than 1 is 0, no need to have that filter
return Optional.empty();
}
final MarketplaceFilterCondition f = new LoanAmountCondition(0, min - 1);
return Optional.of(getRatingFilter(r, f));
});
// filter loans with ask more than maximum
final Stream<Optional<MarketplaceFilter>> s4 = Stream.of(Rating.values()).map(r -> {
final int max = StrategyFileProperty.MAXIMUM_ASK.getValue(r, config::getInt);
if (max < 1) {
return Optional.empty();
}
final MarketplaceFilterCondition f = new LoanAmountCondition(max + 1);
return Optional.of(getRatingFilter(r, f));
});
// put all filters together
final Collection<MarketplaceFilter> filters = Stream.of(s1, s2, s3, s4).flatMap(Function.identity()).flatMap(s -> s.map(Stream::of).orElse(Stream.empty())).collect(Collectors.toList());
// and create the strategy
final ParsedStrategy p = new ParsedStrategy(d, portfolio, investmentSizes, new FilterSupplier(d, filters));
LOGGER.debug("Converted strategy: {}.", p);
final InvestmentStrategy result = new NaturalLanguageInvestmentStrategy(p);
return new SimpleStrategyService.ExclusivelyPrimaryMarketplaceInvestmentStrategy(result);
}
use of com.github.robozonky.strategy.natural.conditions.MarketplaceFilter in project robozonky by RoboZonky.
the class SimpleStrategyService method getRatingFilter.
private static MarketplaceFilter getRatingFilter(final Rating r, final MarketplaceFilterCondition condition) {
final LoanRatingEnumeratedCondition c = new LoanRatingEnumeratedCondition();
c.add(Collections.singleton(r));
final MarketplaceFilter f = new MarketplaceFilter();
f.when(Arrays.asList(c, condition));
return f;
}
Aggregations