up 1.0.8 fix run on Kaggle

WISEPLAT · Jan 2, 2024 · 16c8eed · 16c8eed
1 parent cdb2be1
commit 16c8eed
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/backtrader_moexalgo/moexalgo_feed.py b/backtrader_moexalgo/moexalgo_feed.py
@@ -304,8 +304,11 @@ def get_candles(self, from_date, symbol, interval, skip_first_date=False, skip_l
             iterator = ticker.candles(date=last_date, till_date=till_date, period=interval,
                                       limit=self.limit)  # История. Максимум, 50000 баров
             rows_list = []  # Будем собирать строки в список
-            for it in iterator:  # Итерируем генератор
-                rows_list.append(it.__dict__)  # Класс превращаем в словарь, добавляем строку в список
+            try:
+                for it in iterator:  # Итерируем генератор
+                    rows_list.append(it.__dict__)  # Класс превращаем в словарь, добавляем строку в список
+            except:  # if error - we are in notebook
+                rows_list = iterator
             stats = pd.DataFrame(rows_list)  # Из списка создаем pandas DataFrame
             stats.rename(columns={'begin': 'datetime'}, inplace=True)  # Переименовываем колонку даты и времени
 
@@ -416,8 +419,11 @@ def get_super_candles(self, from_date, symbol, interval="5m", metric='tradestats
                 break
 
             rows_list = []  # Будем собирать строки в список
-            for it in iterator:  # Итерируем генератор
-                rows_list.append(it.__dict__)  # Класс превращаем в словарь, добавляем строку в список
+            try:
+                for it in iterator:  # Итерируем генератор
+                    rows_list.append(it.__dict__)  # Класс превращаем в словарь, добавляем строку в список
+            except:  # if error - we are in notebook
+                rows_list = iterator
             stats = pd.DataFrame(rows_list)  # Из списка создаем pandas DataFrame
             stats.drop('secid', axis='columns', inplace=True)  # Удаляем колонку тикера. Название тикера есть в имени файла
             stats.rename(columns={'ts': 'datetime'}, inplace=True)  # Переименовываем колонку даты и времени

diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
     long_description = f.read()
 
 setup(name='backtrader_moexalgo',
-      version='1.0.7',
+      version='1.0.8',
       author='wiseplat',
       author_email='oshpagin@gmail.com',
       license='MIT License',