Files
quantconnect--lean/ToolBox/IEX/IEXDataDownloader.cs
2019-06-26 11:46:54 -07:00

81 lines
3.2 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.Brokerages.InteractiveBrokers;
using QuantConnect.Data;
using QuantConnect.Data.Market;
using QuantConnect.Logging;
using QuantConnect.Securities;
using System;
using System.Collections.Generic;
using System.Linq;
namespace QuantConnect.ToolBox.IEX
{
public class IEXDataDownloader : IDataDownloader, IDisposable
{
private IEXDataQueueHandler _handler;
private readonly string _apiKey;
public IEXDataDownloader(string apiKey)
{
_handler = new IEXDataQueueHandler(false, apiKey);
_apiKey = apiKey;
}
public void Dispose()
{
_handler.Dispose();
}
/// <summary>
/// Get historical data enumerable for a single symbol, type and resolution given this start and end time (in UTC).
/// </summary>
/// <param name="symbol">Symbol for the data we're looking for.</param>
/// <param name="resolution">Resolution of the data request</param>
/// <param name="startUtc">Start time of the data in UTC</param>
/// <param name="endUtc">End time of the data in UTC</param>
/// <returns>Enumerable of base data for this symbol</returns>
public IEnumerable<BaseData> Get(Symbol symbol, Resolution resolution, DateTime startUtc, DateTime endUtc)
{
if (!(resolution == Resolution.Daily || resolution == Resolution.Minute))
throw new NotSupportedException("Resolution not available: " + resolution);
if (endUtc < startUtc)
throw new ArgumentException("The end date must be greater or equal than the start date.");
var historyRequests = new[] {
new HistoryRequest(startUtc,
endUtc,
typeof(QuoteBar),
symbol,
resolution,
SecurityExchangeHours.AlwaysOpen(TimeZones.EasternStandard),
DateTimeZone.Utc,
resolution,
false,
false,
DataNormalizationMode.Adjusted,
TickType.Quote)
};
foreach (var slice in _handler.GetHistory(historyRequests, TimeZones.EasternStandard))
{
yield return slice[symbol];
}
}
}
}