From 7f4e03ed3cd8dc2ec51c84733bed4248f195e5ea Mon Sep 17 00:00:00 2001 From: Denis Tereshkin Date: Mon, 26 Jul 2021 18:09:24 +0700 Subject: [PATCH] yahoocsvfeed: set ticker from file name --- src/nailab/data/yahoocsvfolderdatasource.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/nailab/data/yahoocsvfolderdatasource.py b/src/nailab/data/yahoocsvfolderdatasource.py index 60b3e40..5fade3f 100644 --- a/src/nailab/data/yahoocsvfolderdatasource.py +++ b/src/nailab/data/yahoocsvfolderdatasource.py @@ -34,7 +34,9 @@ class YahooCsvFolderDataSource(DataSource): for path, f_id in self.feeds: if f_id == feed_id: with open(path, 'r') as f: - return YahooCSVFeed(f) + feed = YahooCSVFeed(f) + feed.set_ticker(feed_id) + return feed raise NailabError('Unable to obtain feed: {}'.format(feed_id))