private static void HandlePagination(GenericScreenerViewModel newViewModel, PaginatedList <Stock> pageList)
{
newViewModel.PageIndex = pageList.PageIndex;
newViewModel.TotalPages = pageList.TotalPages;
newViewModel.HasNextPage = pageList.HasNextPage;
newViewModel.HasPreviousPage = pageList.HasPreviousPage;
}
private bool HandleColumnsGeneric(GenericScreenerViewModel viewModel, GenericScreenerViewModel newViewModel,
List <string> columnsTodisplay)
{
if (!newViewModel.Criterias.Any() && string.IsNullOrEmpty(newViewModel.Columns))
{
columnsTodisplay.Add(ColumnConstants.MarketCap);
columnsTodisplay.Add(ColumnConstants.Per);
}
if (!string.IsNullOrEmpty(newViewModel.Columns))
{
var currentColumns = newViewModel.Columns.Split(',').ToList();
if (currentColumns.Any(x => !ColumnConstants.Columns.ContainsKey(x)))
{
return(true);
}
columnsTodisplay.AddRange(currentColumns);
}
if (string.IsNullOrEmpty(newViewModel.ColumnToAdd) && string.IsNullOrEmpty(newViewModel.ColumnToRemove))
{
if (!string.IsNullOrEmpty(newViewModel.CriteriaToAdd))
{
if (!columnsTodisplay.Contains(newViewModel.CriteriaToAdd) &&
ColumnConstants.Columns.ContainsKey(newViewModel.CriteriaToAdd))
{
columnsTodisplay.Add(newViewModel.CriteriaToAdd);
}
}
}
if (!string.IsNullOrEmpty(newViewModel.ColumnToAdd))
{
if (!columnsTodisplay.Contains(newViewModel.ColumnToAdd) &&
ColumnConstants.Columns.ContainsKey(newViewModel.ColumnToAdd))
{
columnsTodisplay.Add(newViewModel.ColumnToAdd);
}
}
if (!string.IsNullOrEmpty(newViewModel.ColumnToRemove))
{
if (columnsTodisplay.Contains(newViewModel.ColumnToRemove))
{
columnsTodisplay.Remove(newViewModel.ColumnToRemove);
}
}
foreach (var column in ColumnConstants.Columns.Where(x => !columnsTodisplay.Contains(x.Key)))
{
viewModel.AvailableAdditionalColumns.Add(column.Key, column.Value);
}
newViewModel.Columns = string.Join(',', columnsTodisplay);
newViewModel.ColumnToAdd = null;
newViewModel.ColumnToRemove = null;
newViewModel.CriteriaToAdd = null;
newViewModel.CriteriaToRemove = null;
return(false);
}
public async Task <IActionResult> ScreenGeneric(GenericScreenerViewModel viewModel)
{
ViewData["Title"] = "Create your Screener";
var stocks = await _context.Stocks
.Include(s => s.MarketData)
.Include(s => s.PricingResult)
.ThenInclude(p => p.PiotroskiResults)
.Include(s => s.PricingResult)
.ThenInclude(p => p.AnnualResults)
.Where(x => x.PricingResult != null && x.MarketData != null && x.MarketData.MarketCapitalization > 0).AsNoTracking().ToListAsync();
var newViewModel = ManageCriterias(viewModel);
foreach (var criteriaViewModel in newViewModel.Criterias)
{
var criteria = _criteriaFactory.GetCriteria(criteriaViewModel.Id);
if (criteria == null)
{
continue;
}
stocks = stocks
.Where(x => criteria.StockMatch(x, criteriaViewModel)).ToList();
}
stocks = stocks.OrderBy(x => x.Ticker).ToList();
var pageList = PaginatedList <Stock> .Create(stocks, newViewModel.PageIndex > 0?newViewModel.PageIndex : 1, 25);
List <string> columnsTodisplay = new List <string>();
if (HandleColumnsGeneric(viewModel, newViewModel, columnsTodisplay))
{
return(NotFound());
}
newViewModel.ColumnTitles.AddRange(_cellsGenerator.GetColumnTitles(columnsTodisplay));
newViewModel.Rows.AddRange(_cellsGenerator.GetRows(pageList, columnsTodisplay));
HandlePagination(newViewModel, pageList);
return(View(newViewModel));
}
private GenericScreenerViewModel ManageCriterias(GenericScreenerViewModel viewModel)
{
GenericScreenerViewModel newViewModel;
if (viewModel != null)
{
newViewModel = viewModel;
if (newViewModel.Criterias == null)
{
newViewModel.Criterias = new List <ScreenerCriteriaViewModel>();
}
if (!string.IsNullOrEmpty(viewModel.CriteriaToAdd) && _criteriaFactory.CriteriaExists(viewModel.CriteriaToAdd))
{
var criteria = _criteriaFactory.GetCriteria(viewModel.CriteriaToAdd);
var criteriaViewModel = new ScreenerCriteriaViewModel
{
Name = criteria.DisplayName,
Id = criteria.Id,
Operation = criteria.DefaultOperator,
ValueType = criteria.ValueType
};
newViewModel.Criterias.Add(criteriaViewModel);
}
}
else
{
newViewModel = new GenericScreenerViewModel {
Criterias = new List <ScreenerCriteriaViewModel>()
};
}
if (!string.IsNullOrEmpty(newViewModel.CriteriaToRemove) &&
newViewModel.Criterias.Any(x => x.Id == newViewModel.CriteriaToRemove))
{
var criteria = newViewModel.Criterias.First(x => x.Id == newViewModel.CriteriaToRemove);
newViewModel.Criterias.Remove(criteria);
}
return(newViewModel);
}
