PyAlgoEngine 0.5.4.post6__tar.gz → 0.5.4.post7__tar.gz

{pyalgoengine-0.5.4.post6 → pyalgoengine-0.5.4.post7}/PKG-INFO

@@ -1,27 +1,27 @@
-Metadata-Version: 2.1
-Name: PyAlgoEngine
-Version: 0.5.4.post6
-Summary: Basic algo engine
-Home-page: https://github.com/BolunHan/PyAlgoEngine
-Author: Bolun.Han
-Author-email: Bolun.Han@outlook.com
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.1
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numpy
-Requires-Dist: pandas
-Requires-Dist: exchange_calendars
-Requires-Dist: PyEventEngine
-Provides-Extra: webapps
-Requires-Dist: flask; extra == "webapps"
-Requires-Dist: waitress; extra == "webapps"
-Requires-Dist: bokeh; extra == "webapps"
-
-# PyAlgoEngine
-python algo trading engine
+Metadata-Version: 2.1
+Name: PyAlgoEngine
+Version: 0.5.4.post7
+Summary: Basic algo engine
+Home-page: https://github.com/BolunHan/PyAlgoEngine
+Author: Bolun.Han
+Author-email: Bolun.Han@outlook.com
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.1
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: exchange_calendars
+Requires-Dist: PyEventEngine
+Provides-Extra: webapps
+Requires-Dist: flask; extra == "webapps"
+Requires-Dist: waitress; extra == "webapps"
+Requires-Dist: bokeh; extra == "webapps"
+
+# PyAlgoEngine
+python algo trading engine

{pyalgoengine-0.5.4.post6 → pyalgoengine-0.5.4.post7}/PyAlgoEngine.egg-info/PKG-INFO

@@ -1,27 +1,27 @@
-Metadata-Version: 2.1
-Name: PyAlgoEngine
-Version: 0.5.4.post6
-Summary: Basic algo engine
-Home-page: https://github.com/BolunHan/PyAlgoEngine
-Author: Bolun.Han
-Author-email: Bolun.Han@outlook.com
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.1
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numpy
-Requires-Dist: pandas
-Requires-Dist: exchange_calendars
-Requires-Dist: PyEventEngine
-Provides-Extra: webapps
-Requires-Dist: flask; extra == "webapps"
-Requires-Dist: waitress; extra == "webapps"
-Requires-Dist: bokeh; extra == "webapps"
-
-# PyAlgoEngine
-python algo trading engine
+Metadata-Version: 2.1
+Name: PyAlgoEngine
+Version: 0.5.4.post7
+Summary: Basic algo engine
+Home-page: https://github.com/BolunHan/PyAlgoEngine
+Author: Bolun.Han
+Author-email: Bolun.Han@outlook.com
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.1
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: exchange_calendars
+Requires-Dist: PyEventEngine
+Provides-Extra: webapps
+Requires-Dist: flask; extra == "webapps"
+Requires-Dist: waitress; extra == "webapps"
+Requires-Dist: bokeh; extra == "webapps"
+
+# PyAlgoEngine
+python algo trading engine

{pyalgoengine-0.5.4.post6 → pyalgoengine-0.5.4.post7}/algo_engine/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.5.4.post6"
+__version__ = "0.5.4.post7"
 
 import logging
 import os

{pyalgoengine-0.5.4.post6 → pyalgoengine-0.5.4.post7}/algo_engine/base/__init__.py

@@ -22,7 +22,7 @@ from .console_utils import Progress, GetInput, GetArgs, count_ordinal, TerminalS
 
 __all__ = ['PROFILE',
            'FinancialDecimal',
-           'TransactionSide', 'MarketData', 'OrderBook', 'BarData', 'TickData', 'TransactionData', 'TradeData', 'OrderBookBuffer', 'BarDataBuffer', 'TickDataBuffer', 'TransactionDataBuffer',
+           'TransactionSide', 'MarketData', 'OrderBook', 'BarData', 'DailyBar', 'CandleStick', 'TickData', 'TransactionData', 'TradeData', 'OrderBookBuffer', 'BarDataBuffer', 'TickDataBuffer', 'TransactionDataBuffer',
            'TechnicalAnalysis',
            'OrderState', 'OrderType', 'TradeInstruction', 'TradeReport',
            'Progress', 'GetInput', 'GetArgs', 'count_ordinal', 'TerminalStyle', 'InteractiveShell', 'ShellTransfer']

{pyalgoengine-0.5.4.post6 → pyalgoengine-0.5.4.post7}/algo_engine/base/market_utils.py

@@ -9,7 +9,7 @@ import re
 import warnings
 from ctypes import c_ulong, c_double, c_wchar, c_int, c_longlong
 from multiprocessing import RawValue, RawArray
-from typing import overload
+from typing import overload, Literal
 
 import numpy as np
 
@@ -733,7 +733,7 @@ class BarData(MarketData):
                 raise ValueError(f'Invalid bar_span, expect int, float or timedelta, got {bar_span}')
 
     def __repr__(self):
-        return f'<BarData>([{self.market_time:%Y-%m-%d %H:%M:%S}] {self.ticker}, open={self.open_price}, close={self.close_price}, high={self.high_price}, low={self.low_price})'
+        return f'<{self.__class__.__name__}>([{self.market_time:%Y-%m-%d %H:%M:%S}] {self.ticker}, open={self.open_price}, close={self.close_price}, high={self.high_price}, low={self.low_price})'
 
     @classmethod
     def from_json(cls, json_message: str | bytes | bytearray | dict) -> BarData:
@@ -864,28 +864,21 @@ class BarData(MarketData):
         return self.close_price
 
     @property
-    def bar_type(self):
-        if self.bar_span > datetime.timedelta(days=1):
-            return 'Over-Daily'
-        elif self.bar_span == datetime.timedelta(days=1):
-            return 'Daily'
-        elif self.bar_span > datetime.timedelta(hours=1):
-            return 'Over-Hourly'
-        elif self.bar_span == datetime.timedelta(hours=1):
+    def bar_type(self) -> Literal['Hourly-Plus', 'Hourly', 'Minute-Plus', 'Minute', 'Sub-Minute']:
+        if self['bar_span'] > 3600:
+            return 'Hourly-Plus'
+        elif self['bar_span'] == 3600:
             return 'Hourly'
-        elif self.bar_span > datetime.timedelta(minutes=1):
-            return 'Over-Minute'
-        elif self.bar_span == datetime.timedelta(minutes=1):
+        elif self['bar_span'] > 60:
+            return 'Minute-Plus'
+        elif self['bar_span'] == 60:
             return 'Minute'
         else:
             return 'Sub-Minute'
 
     @property
     def bar_end_time(self) -> datetime.datetime | datetime.date:
-        if self.bar_type == 'Daily':
-            return self.market_time.date()
-        else:
-            return self.market_time
+        return self.market_time
 
 
 # alias of the BarData
@@ -923,9 +916,15 @@ class DailyBar(BarData):
         else:
             assert (market_date - start_date).days == bar_span.days
 
+        if timestamp is None:
+            if PROFILE.session_end is None:
+                timestamp = datetime.datetime.combine(market_date, datetime.time.min, tzinfo=PROFILE.time_zone).timestamp()
+            else:
+                timestamp = datetime.datetime.combine(market_date, PROFILE.session_end, tzinfo=PROFILE.time_zone).timestamp()
+
         super().__init__(
             ticker=ticker,
-            timestamp=datetime.datetime.combine(market_date, datetime.time.min).timestamp() if timestamp is None else timestamp,
+            timestamp=timestamp,
             bar_span=bar_span,
             high_price=high_price,
             low_price=low_price,
@@ -939,6 +938,9 @@ class DailyBar(BarData):
 
         self['market_date'] = market_date
 
+    def __repr__(self):
+        return f'<{self.__class__.__name__}>([{self.market_time:%Y-%m-%d}] {self.ticker}, open={self.open_price}, close={self.close_price}, high={self.high_price}, low={self.low_price})'
+
     @property
     def bar_span(self) -> datetime.timedelta:
         return datetime.timedelta(days=self['bar_span'])
@@ -947,6 +949,27 @@ class DailyBar(BarData):
     def market_date(self) -> datetime.date:
         return self['market_date']
 
+    @property
+    def market_time(self) -> datetime.date:
+        return self.market_date
+
+    @property
+    def bar_start_time(self) -> datetime.date:
+        return self.market_date - self.bar_span
+
+    @property
+    def bar_end_time(self) -> datetime.date:
+        return self.market_date
+
+    @property
+    def bar_type(self) -> Literal['Daily', 'Daily-Plus']:
+        if self['bar_span'] == 1:
+            return 'Daily'
+        elif self['bar_span'] > 1:
+            return 'Daily-Plus'
+        else:
+            raise ValueError(f'Invalid bar_span for {self.__class__.__name__}! Expect a int greater or equal to 1, got {self["bar_span"]}')
+
 
 class TickData(MarketData):
     def __init__(

{pyalgoengine-0.5.4.post6 → pyalgoengine-0.5.4.post7}/setup.cfg

@@ -1,4 +1,4 @@
-[egg_info]
-tag_build = 
-tag_date = 0
-
+[egg_info]
+tag_build = 
+tag_date = 0
+