nzem-time 1.0.0__tar.gz

nzem_time-1.0.0/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: nzem-time
+Version: 1.0.0
+Summary: Trading Period Utilities for NZEM
+Author: Your Name
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown

nzem_time-1.0.0/pyproject.toml

@@ -0,0 +1,13 @@
+[build-system]
+requires = ["setuptools>=68"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "nzem-time"
+version = "1.0.0"
+description = "Trading Period Utilities for NZEM"
+readme = "README.md"
+requires-python = ">=3.10"
+authors = [
+    {name = "Your Name"}
+]

nzem_time-1.0.0/setup.cfg

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

nzem_time-1.0.0/src/nzem_time/__init__.py

@@ -0,0 +1,9 @@
+# src/trading_time/__init__.py
+
+from .fiveminperiods import FiveMinPeriod
+from .tradingperiods import TradingPeriod
+
+__all__ = [
+    "FiveMinPeriod",
+    "TradingPeriod",
+]

nzem_time-1.0.0/src/nzem_time/_common.py

@@ -0,0 +1,10 @@
+# _common.py
+
+from datetime import datetime, timedelta, timezone, time
+from zoneinfo import ZoneInfo
+from typing import Tuple
+
+NZ = ZoneInfo("Pacific/Auckland")
+UTC = timezone.utc
+EPOCH_UTC = datetime(1970, 1, 1, tzinfo=timezone.utc)
+PERIOD_SECONDS = 300

nzem_time-1.0.0/src/nzem_time/fiveminperiods.py

@@ -0,0 +1,47 @@
+from .tradingperiods import TradingPeriod
+from ._common import PERIOD_SECONDS
+from typing import Tuple
+
+class FiveMinPeriod:
+    __slots__ = ("_value",)
+
+    def __init__(self, value: int):
+        self._value = value
+    
+    def __repr__(self):
+        return f"FiveMinPeriod(tp={self.tp}, period={self.period})"
+
+    @classmethod
+    def from_datetime(cls, dt):
+        return cls(int(dt.timestamp()) // PERIOD_SECONDS)
+
+    @property
+    def index(self) -> int:
+        return self._value
+
+    @property
+    def tp(self):
+        return TradingPeriod(self._value // 6)
+
+    @property
+    def period(self):
+        # 1-6
+        return (self._value % 6) + 1
+
+    def next(self, jump=1):
+        return FiveMinPeriod(self._value + jump)
+
+    def prev(self, jump=1):
+        return FiveMinPeriod(self._value - jump)
+
+    @staticmethod
+    def range(start, end_exclusive):
+        current = start._value
+        end = end_exclusive._value
+
+        while current < end:
+            yield FiveMinPeriod(current)
+            current += 1
+
+def get_five_min_periods(tp: TradingPeriod) -> Tuple[FiveMinPeriod, FiveMinPeriod, FiveMinPeriod, FiveMinPeriod, FiveMinPeriod, FiveMinPeriod]:
+    return tuple(FiveMinPeriod(tp,i) for i in range(1,7))

nzem_time-1.0.0/src/nzem_time/tradingperiods.py

@@ -0,0 +1,58 @@
+from datetime import datetime, timedelta
+from ._common import EPOCH_UTC, NZ
+
+class TradingPeriod:
+    __slots__ = ("_halfhourssince")
+
+    def __init__(self, hhs: int):
+        object.__setattr__(self, "_halfhourssince", hhs)
+
+    def __repr__(self):
+        return f"TradingPeriod(date={self.as_datetime_nz.date()}, period={self.period})"
+
+    def __setattr__(self, name, value):
+        raise AttributeError(f"{type(self).__name__} is immutable.")
+
+    @classmethod
+    def from_datetime(cls, dt: datetime):
+        half_hours_since = int(dt.timestamp()) // (1800)
+        return TradingPeriod(half_hours_since)
+
+    @property
+    def as_datetime_utc(self) -> datetime:
+        return EPOCH_UTC + timedelta(minutes=30 * self._halfhourssince)
+
+    @property
+    def as_datetime_nz(self):
+        return (EPOCH_UTC + timedelta(minutes=self._halfhourssince*30)).astimezone(NZ)
+
+    @property
+    def period(self):
+        dt = self.as_datetime_nz
+        d = dt.date()
+        return int(
+            dt.timestamp() - 
+            datetime(d.year, d.month, d.day,0,tzinfo=NZ).timestamp()
+        ) // 1800 + 1
+
+    @staticmethod
+    def range(start, end_exclusive):
+        current = start
+        while current._halfhourssince < end_exclusive._halfhourssince:
+            yield current
+            current = current.next()
+
+    def next(self, jump=1):
+        return TradingPeriod(self._halfhourssince + jump)
+    
+    def prev(self, jump=1):
+        return TradingPeriod(self._halfhourssince - jump)
+    
+    def __hash__(self):
+        return hash(self._halfhourssince)
+
+    def __eq__(self, other):
+        return (
+            isinstance(other, TradingPeriod)
+            and self._halfhourssince == other._halfhourssince
+        )

nzem_time-1.0.0/src/nzem_time.egg-info/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: nzem-time
+Version: 1.0.0
+Summary: Trading Period Utilities for NZEM
+Author: Your Name
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown

nzem_time-1.0.0/src/nzem_time.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+src/nzem_time/__init__.py
+src/nzem_time/_common.py
+src/nzem_time/fiveminperiods.py
+src/nzem_time/tradingperiods.py
+src/nzem_time.egg-info/PKG-INFO
+src/nzem_time.egg-info/SOURCES.txt
+src/nzem_time.egg-info/dependency_links.txt
+src/nzem_time.egg-info/top_level.txt

nzem_time-1.0.0/src/nzem_time.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

nzem_time-1.0.0/src/nzem_time.egg-info/top_level.txt

@@ -0,0 +1 @@
+nzem_time