Files
quantconnect--lean/Common/Scheduling/BaseScheduleRules.cs
T
Martin-Molinero bb5c0eb54e
API Tests / build (push) Has been cancelled
Benchmarks / build (push) Has been cancelled
Build & Test Lean / build (push) Has been cancelled
Regression Tests / build (push) Has been cancelled
Report Generator Tests / build (push) Has been cancelled
Research Regression Tests / build (push) Has been cancelled
Python Virtual Environments / build (push) Has been cancelled
Minor api improvements (#7901)
* Minor api improvements

- Adding Universe.Selected to hold the current set of selected symbols.
  Adding regression tests
- Adjusting TimeRules and DateRules to fallback to the MHDB if the
  security hasn't been added. Updating existing regression tests

* Expand regression test assertion
2024-04-05 17:43:25 -03:00

68 lines
2.4 KiB
C#

/*
* QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
* Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
using NodaTime;
using QuantConnect.Securities;
namespace QuantConnect.Scheduling
{
/// <summary>
/// Base rule scheduler
/// </summary>
public class BaseScheduleRules
{
/// <summary>
/// The algorithm's default time zone
/// </summary>
protected DateTimeZone TimeZone { get; set; }
/// <summary>
/// The security manager
/// </summary>
protected SecurityManager Securities { get; set; }
/// <summary>
/// The market hours database instance to use
/// </summary>
protected MarketHoursDatabase MarketHoursDatabase { get; set; }
/// <summary>
/// Initializes a new instance of the <see cref="TimeRules"/> helper class
/// </summary>
/// <param name="securities">The security manager</param>
/// <param name="timeZone">The algorithm's default time zone</param>
/// <param name="marketHoursDatabase">The market hours database instance to use</param>
public BaseScheduleRules(SecurityManager securities, DateTimeZone timeZone, MarketHoursDatabase marketHoursDatabase)
{
Securities = securities;
TimeZone = timeZone;
MarketHoursDatabase = marketHoursDatabase;
}
/// <summary>
/// Helper method to fetch the security exchange hours
/// </summary>
protected SecurityExchangeHours GetSecurityExchangeHours(Symbol symbol)
{
if (!Securities.TryGetValue(symbol, out var security))
{
return MarketHoursDatabase.GetEntry(symbol.ID.Market, symbol, symbol.SecurityType).ExchangeHours;
}
return security.Exchange.Hours;
}
}
}