backtrader-next 2.3.0__tar.gz → 2.3.2__tar.gz

{backtrader_next-2.3.0 → backtrader_next-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: backtrader_next
-Version: 2.3.0
+Version: 2.3.2
 Summary: Live Trading and backtesting platform in Python
 Project-URL: Homepage, https://github.com/smalinin/backtrader_next
 Project-URL: Source, https://github.com/smalinin/backtrader_next

{backtrader_next-2.3.0 → backtrader_next-2.3.2}/backtrader_next/broker.py

@@ -19,10 +19,12 @@
 #
 ###############################################################################
 from __future__ import (absolute_import, division, print_function, unicode_literals)
+from collections import defaultdict
 
 from backtrader_next.comminfo import CommInfoBase
 from backtrader_next.metabase import MetaParams
-from backtrader_next.utils.py3 import with_metaclass
+
+from .utils.py3 import with_metaclass, iteritems
 
 from . import fillers as fillers
 from . import fillers as filler
@@ -52,6 +54,7 @@ class BrokerBase(with_metaclass(MetaBroker, object)):
 
     def __init__(self):
         self.comminfo = dict()
+        self.info = defaultdict(dict)
         self.init()
 
     def init(self):
@@ -65,6 +68,13 @@ class BrokerBase(with_metaclass(MetaBroker, object)):
     def stop(self):
         pass
 
+    def add_info(self, strat_name:str, **kwargs):
+        '''Add the keys, values of kwargs to the internal info dictionary to
+        hold custom information in the broker
+        '''
+        for key, val in iteritems(kwargs):
+            self.info[strat_name][key] = val
+
     def add_order_history(self, orders, notify=False):
         '''Add order history. See cerebro for details'''
         raise NotImplementedError

{backtrader_next-2.3.0 → backtrader_next-2.3.2}/backtrader_next/order.py

@@ -707,16 +707,15 @@ class Order(OrderBase):
         del params['_owner_classname']
         params['restored'] = True
         ordtype = odict.get('ordtype', None)
-        ord = None
+        new_ord = None
         if ordtype is None:
-            ord = Order(**params)
+            new_ord = Order(**params)
         if ordtype == cls.Buy:
-            ord = BuyOrder(**params)
+            new_ord = BuyOrder(**params)
         elif ordtype == cls.Sell:
-            ord = SellOrder(**params)
+            new_ord = SellOrder(**params)
         else:
-            ord = Order(**params)
-
+            new_ord = Order(**params)
         for key, val in odict.items():
             if key in ('created', 'executed'):
                 odata = OrderData(
@@ -736,21 +735,21 @@ class Order(OrderBase):
                 odata.psize = val.get('psize', 0)
                 odata.pprice = val.get('pprice', 0.0)
                 if key == 'created':
-                    ord.created = odata
+                    new_ord.created = odata
                 else:
-                    ord.executed = odata
+                    new_ord.executed = odata
             elif key in ('ordtype', 'params'):
                 pass # already handled
             elif key == 'info':
                 for ik, iv in val.items():
-                    ord.info[ik] = iv
+                    new_ord.info[ik] = iv
             elif key == 'comminfo':
-                ord.comminfo = None
+                new_ord.comminfo = None
             elif key == 'parent':
-                ord.parent = None
-            elif hasattr(ord, key):
-                setattr(ord, key, val)
-        return ord
+                new_ord.parent = None
+            elif hasattr(new_ord, key):
+                setattr(new_ord, key, val)
+        return new_ord
 
 
 class BuyOrder(Order):

{backtrader_next-2.3.0 → backtrader_next-2.3.2}/backtrader_next/strategy.py

@@ -1,4 +1,4 @@
-#!/usr/bin389/env python
+#!/usr/bin/env python
 # -*- coding: utf-8; py-indent-offset:4 -*-
 ###############################################################################
 #
@@ -118,6 +118,17 @@ class Strategy(with_metaclass(MetaStrategy, StrategyBase)):
     # keep the latest delivered data date in the line
     lines = ('datetime',)
 
+    def add_info(self, strat_name:str, **kwargs):
+        '''Add the keys, values of kwargs to the internal info dictionary to
+        hold custom information in the broker
+        '''
+        self.broker.add_info(strat_name, kwargs)
+
+    def get_info(self, strat_name:str):
+        '''Get the info dictionary for the given strategy name from the broker
+        '''
+        return self.broker.get_info(strat_name)
+
     def qbuffer(self, savemem=0, replaying=False):
         '''Enable the memory saving schemes. Possible values for ``savemem``:
 

{backtrader_next-2.3.0 → backtrader_next-2.3.2}/backtrader_next/version.py

@@ -22,6 +22,6 @@ from __future__ import (absolute_import, division, print_function,
                         unicode_literals)
 
 
-__version__ = '2.3.0'
+__version__ = '2.3.2'
 
 __btversion__ = tuple(int(x) for x in __version__.split('.'))

{backtrader_next-2.3.0 → backtrader_next-2.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "backtrader_next"
-version = "2.3.0"
+version = "2.3.2"
 description = "Live Trading and backtesting platform in Python"
 readme = "README.md"
 license = { text = "GPL-3.0-or-later" }