pandas-market-calendars 5.1.0__py3-none-any.whl → 5.1.3__py3-none-any.whl

pandas_market_calendars/calendars/tsx.py

@@ -1,185 +1,193 @@
-from datetime import time
-from itertools import chain
-
-import pandas as pd
-from pandas.tseries.holiday import (
-    AbstractHolidayCalendar,
-    DateOffset,
-    GoodFriday,
-    Holiday,
-    MO,
-    weekend_to_monday,
-)
-from zoneinfo import ZoneInfo
-
-from pandas_market_calendars.holidays.uk import (
-    BoxingDay,
-    WeekendBoxingDay,
-    WeekendChristmas,
-)
-from pandas_market_calendars.market_calendar import (
-    MarketCalendar,
-    MONDAY,
-    TUESDAY,
-    WEDNESDAY,
-    THURSDAY,
-    FRIDAY,
-)
-
-# New Year's Day
-TSXNewYearsDay = Holiday(
-    "New Year's Day",
-    month=1,
-    day=1,
-    observance=weekend_to_monday,
-)
-# Ontario Family Day
-FamilyDay = Holiday(
-    "Family Day",
-    month=2,
-    day=1,
-    offset=DateOffset(weekday=MO(3)),
-    start_date="2008-01-01",
-)
-# Victoria Day
-# https://www.timeanddate.com/holidays/canada/victoria-day
-VictoriaDay = Holiday(
-    "Victoria Day",
-    month=5,
-    day=24,
-    offset=DateOffset(weekday=MO(-1)),
-)
-# Canada Day
-CanadaDay = Holiday(
-    "Canada Day",
-    month=7,
-    day=1,
-    observance=weekend_to_monday,
-)
-# Civic Holiday
-CivicHoliday = Holiday(
-    "Civic Holiday",
-    month=8,
-    day=1,
-    offset=DateOffset(weekday=MO(1)),
-)
-# Labor Day
-LaborDay = Holiday(
-    "Labor Day",
-    month=9,
-    day=1,
-    offset=DateOffset(weekday=MO(1)),
-)
-# Thanksgiving
-Thanksgiving = Holiday(
-    "Thanksgiving",
-    month=10,
-    day=1,
-    offset=DateOffset(weekday=MO(2)),
-)
-
-Christmas = Holiday(
-    "Christmas",
-    month=12,
-    day=25,
-)
-
-ChristmasEveEarlyClose2010Onwards = Holiday(
-    "Christmas Eve Early Close",
-    month=12,
-    day=24,
-    days_of_week=(MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY),
-    start_date=pd.Timestamp("2010-01-01"),
-)
-
-September11Closings2001 = [
-    pd.Timestamp("2001-09-11", tz="UTC"),
-    pd.Timestamp("2001-09-12", tz="UTC"),
-]
-
-
-class TSXExchangeCalendar(MarketCalendar):
-    """
-    Exchange calendar for the Toronto Stock Exchange
-
-    Open Time: 9:30 AM, EST
-    Close Time: 4:00 PM, EST
-
-    Regularly-Observed Holidays:
-    - New Years Day (observed on first business day on/after)
-    - Family Day (Third Monday in February, starting in 2008)
-    - Good Friday
-    - Victoria Day (Monday before May 25th)
-    - Canada Day (July 1st, observed first business day after)
-    - Civic Holiday (First Monday in August)
-    - Labor Day (First Monday in September)
-    - Thanksgiving (Second Monday in October)
-    - Christmas Day
-        - Dec. 26th if Christmas is on a Sunday
-        - Dec. 27th if Christmas is on a weekend
-    - Boxing Day
-        - Dec. 27th if Christmas is on a Sunday
-        - Dec. 28th if Boxing Day is on a weekend
-
-    Early closes:
-    - Starting in 2010, if Christmas Eve falls on a weekday, the market
-      closes at 1:00 pm that day. If it falls on a weekend, there is no
-      early close.
-    """
-
-    aliases = ["TSX", "TSXV"]
-
-    regular_market_times = {
-        "market_open": ((None, time(9, 30)),),
-        "market_close": ((None, time(16)),),
-    }
-
-    @property
-    def name(self):
-        return "TSX"
-
-    @property
-    def full_name(self):
-        return "Toronto Stock Exchange"
-
-    @property
-    def tz(self):
-        return ZoneInfo("Canada/Eastern")
-
-    regular_early_close = time(13)
-
-    @property
-    def regular_holidays(self):
-        return AbstractHolidayCalendar(
-            rules=[
-                TSXNewYearsDay,
-                FamilyDay,
-                GoodFriday,
-                VictoriaDay,
-                CanadaDay,
-                CivicHoliday,
-                LaborDay,
-                Thanksgiving,
-                Christmas,
-                WeekendChristmas,
-                BoxingDay,
-                WeekendBoxingDay,
-            ]
-        )
-
-    @property
-    def adhoc_holidays(self):
-        return list(
-            chain(
-                September11Closings2001,
-            )
-        )
-
-    @property
-    def special_closes(self):
-        return [
-            (
-                self.regular_early_close,
-                AbstractHolidayCalendar([ChristmasEveEarlyClose2010Onwards]),
-            )
-        ]
+from datetime import time
+from itertools import chain
+
+import pandas as pd
+from pandas.tseries.holiday import (
+    AbstractHolidayCalendar,
+    DateOffset,
+    GoodFriday,
+    Holiday,
+    MO,
+    weekend_to_monday,
+)
+import sys
+
+# check python versiOn aNd import accordingly
+if sys.version_info >= (3, 9):
+    # For Python 3.9 and later, import directly
+    from zoneinfo import ZoneInfo
+else:
+    # For Python 3.8 and earlier, import from backports
+    from backports.zoneinfo import ZoneInfo
+
+from pandas_market_calendars.holidays.uk import (
+    BoxingDay,
+    WeekendBoxingDay,
+    WeekendChristmas,
+)
+from pandas_market_calendars.market_calendar import (
+    MarketCalendar,
+    MONDAY,
+    TUESDAY,
+    WEDNESDAY,
+    THURSDAY,
+    FRIDAY,
+)
+
+# New Year's Day
+TSXNewYearsDay = Holiday(
+    "New Year's Day",
+    month=1,
+    day=1,
+    observance=weekend_to_monday,
+)
+# Ontario Family Day
+FamilyDay = Holiday(
+    "Family Day",
+    month=2,
+    day=1,
+    offset=DateOffset(weekday=MO(3)),
+    start_date="2008-01-01",
+)
+# Victoria Day
+# https://www.timeanddate.com/holidays/canada/victoria-day
+VictoriaDay = Holiday(
+    "Victoria Day",
+    month=5,
+    day=24,
+    offset=DateOffset(weekday=MO(-1)),
+)
+# Canada Day
+CanadaDay = Holiday(
+    "Canada Day",
+    month=7,
+    day=1,
+    observance=weekend_to_monday,
+)
+# Civic Holiday
+CivicHoliday = Holiday(
+    "Civic Holiday",
+    month=8,
+    day=1,
+    offset=DateOffset(weekday=MO(1)),
+)
+# Labor Day
+LaborDay = Holiday(
+    "Labor Day",
+    month=9,
+    day=1,
+    offset=DateOffset(weekday=MO(1)),
+)
+# Thanksgiving
+Thanksgiving = Holiday(
+    "Thanksgiving",
+    month=10,
+    day=1,
+    offset=DateOffset(weekday=MO(2)),
+)
+
+Christmas = Holiday(
+    "Christmas",
+    month=12,
+    day=25,
+)
+
+ChristmasEveEarlyClose2010Onwards = Holiday(
+    "Christmas Eve Early Close",
+    month=12,
+    day=24,
+    days_of_week=(MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY),
+    start_date=pd.Timestamp("2010-01-01"),
+)
+
+September11Closings2001 = [
+    pd.Timestamp("2001-09-11", tz="UTC"),
+    pd.Timestamp("2001-09-12", tz="UTC"),
+]
+
+
+class TSXExchangeCalendar(MarketCalendar):
+    """
+    Exchange calendar for the Toronto Stock Exchange
+
+    Open Time: 9:30 AM, EST
+    Close Time: 4:00 PM, EST
+
+    Regularly-Observed Holidays:
+    - New Years Day (observed on first business day on/after)
+    - Family Day (Third Monday in February, starting in 2008)
+    - Good Friday
+    - Victoria Day (Monday before May 25th)
+    - Canada Day (July 1st, observed first business day after)
+    - Civic Holiday (First Monday in August)
+    - Labor Day (First Monday in September)
+    - Thanksgiving (Second Monday in October)
+    - Christmas Day
+        - Dec. 26th if Christmas is on a Sunday
+        - Dec. 27th if Christmas is on a weekend
+    - Boxing Day
+        - Dec. 27th if Christmas is on a Sunday
+        - Dec. 28th if Boxing Day is on a weekend
+
+    Early closes:
+    - Starting in 2010, if Christmas Eve falls on a weekday, the market
+      closes at 1:00 pm that day. If it falls on a weekend, there is no
+      early close.
+    """
+
+    aliases = ["TSX", "TSXV"]
+
+    regular_market_times = {
+        "market_open": ((None, time(9, 30)),),
+        "market_close": ((None, time(16)),),
+    }
+
+    @property
+    def name(self):
+        return "TSX"
+
+    @property
+    def full_name(self):
+        return "Toronto Stock Exchange"
+
+    @property
+    def tz(self):
+        return ZoneInfo("Canada/Eastern")
+
+    regular_early_close = time(13)
+
+    @property
+    def regular_holidays(self):
+        return AbstractHolidayCalendar(
+            rules=[
+                TSXNewYearsDay,
+                FamilyDay,
+                GoodFriday,
+                VictoriaDay,
+                CanadaDay,
+                CivicHoliday,
+                LaborDay,
+                Thanksgiving,
+                Christmas,
+                WeekendChristmas,
+                BoxingDay,
+                WeekendBoxingDay,
+            ]
+        )
+
+    @property
+    def adhoc_holidays(self):
+        return list(
+            chain(
+                September11Closings2001,
+            )
+        )
+
+    @property
+    def special_closes(self):
+        return [
+            (
+                self.regular_early_close,
+                AbstractHolidayCalendar([ChristmasEveEarlyClose2010Onwards]),
+            )
+        ]

pandas_market_calendars/class_registry.py

@@ -1,115 +1,115 @@
-import inspect
-from pprint import pformat
-
-
-def _regmeta_instance_factory(cls, name, *args, **kwargs):
-    """
-    :param cls(RegisteryMeta): registration meta class
-    :param name(str): name of class that needs to be instantiated
-    :param args(Optional(tuple)): instance positional arguments
-    :param kwargs(Optional(dict)): instance named arguments
-    :return: class instance
-    """
-    try:
-        class_ = cls._regmeta_class_registry[name]
-    except KeyError:
-        raise RuntimeError(
-            "Class {} is not one of the registered classes: {}".format(
-                name, cls._regmeta_class_registry.keys()
-            )
-        )
-    return class_(*args, **kwargs)
-
-
-def _regmeta_register_class(cls, regcls, name):
-    """
-    :param cls(RegisteryMeta): registration base class
-    :param regcls(class): class to be registered
-    :param name(str): name of the class to be registered
-    """
-    if hasattr(regcls, "aliases"):
-        if regcls.aliases:
-            for alias in regcls.aliases:
-                cls._regmeta_class_registry[alias] = regcls
-        else:
-            cls._regmeta_class_registry[name] = regcls
-    else:
-        cls._regmeta_class_registry[name] = regcls
-
-
-class RegisteryMeta(type):
-    """
-    Metaclass used to register all classes inheriting from RegisteryMeta
-    """
-
-    def __new__(mcs, name, bases, attr):
-        cls = super(RegisteryMeta, mcs).__new__(mcs, name, bases, attr)
-        if not hasattr(cls, "_regmeta_class_registry"):
-            cls._regmeta_class_registry = {}
-            cls.factory = classmethod(_regmeta_instance_factory)
-
-        return cls
-
-    def __init__(cls, name, bases, attr):
-        if not inspect.isabstract(cls):
-            _regmeta_register_class(cls, cls, name)
-            for b in bases:
-                if hasattr(b, "_regmeta_class_registry"):
-                    _regmeta_register_class(b, cls, name)
-
-        super(RegisteryMeta, cls).__init__(name, bases, attr)
-
-        cls.regular_market_times = ProtectedDict(cls.regular_market_times)
-        cls.open_close_map = ProtectedDict(cls.open_close_map)
-
-        cls.special_market_open = cls.special_opens
-        cls.special_market_open_adhoc = cls.special_opens_adhoc
-
-        cls.special_market_close = cls.special_closes
-        cls.special_market_close_adhoc = cls.special_closes_adhoc
-
-
-class ProtectedDict(dict):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        # __init__ is bypassed when unpickling, which causes __setitem__ to fail
-        # without the _INIT_RAN_NORMALLY flag
-        self._INIT_RAN_NORMALLY = True
-
-    def _set(self, key, value):
-        return super().__setitem__(key, value)
-
-    def _del(self, key):
-        return super().__delitem__(key)
-
-    def __setitem__(self, key, value):
-        if not hasattr(self, "_INIT_RAN_NORMALLY"):
-            return self._set(key, value)
-
-        raise TypeError(
-            "You cannot set a value directly, you can change regular_market_times "
-            "using .change_time, .add_time or .remove_time."
-        )
-
-    def __delitem__(self, key):
-        if not hasattr(self, "_INIT_RAN_NORMALLY"):
-            return self._del(key)
-
-        raise TypeError(
-            "You cannot delete an item directly. You can change regular_market_times "
-            "using .change_time, .add_time or .remove_time"
-        )
-
-    def __repr__(self):
-        return self.__class__.__name__ + "(" + super().__repr__() + ")"
-
-    def __str__(self):
-        try:
-            formatted = pformat(dict(self), sort_dicts=False)  # sort_dicts apparently not available < python3.8
-        except TypeError:
-            formatted = pformat(dict(self))
-
-        return self.__class__.__name__ + "(\n" + formatted + "\n)"
-
-    def copy(self):
-        return self.__class__(super().copy())
+import inspect
+from pprint import pformat
+
+
+def _regmeta_instance_factory(cls, name, *args, **kwargs):
+    """
+    :param cls(RegisteryMeta): registration meta class
+    :param name(str): name of class that needs to be instantiated
+    :param args(Optional(tuple)): instance positional arguments
+    :param kwargs(Optional(dict)): instance named arguments
+    :return: class instance
+    """
+    try:
+        class_ = cls._regmeta_class_registry[name]
+    except KeyError:
+        raise RuntimeError(
+            "Class {} is not one of the registered classes: {}".format(
+                name, cls._regmeta_class_registry.keys()
+            )
+        )
+    return class_(*args, **kwargs)
+
+
+def _regmeta_register_class(cls, regcls, name):
+    """
+    :param cls(RegisteryMeta): registration base class
+    :param regcls(class): class to be registered
+    :param name(str): name of the class to be registered
+    """
+    if hasattr(regcls, "aliases"):
+        if regcls.aliases:
+            for alias in regcls.aliases:
+                cls._regmeta_class_registry[alias] = regcls
+        else:
+            cls._regmeta_class_registry[name] = regcls
+    else:
+        cls._regmeta_class_registry[name] = regcls
+
+
+class RegisteryMeta(type):
+    """
+    Metaclass used to register all classes inheriting from RegisteryMeta
+    """
+
+    def __new__(mcs, name, bases, attr):
+        cls = super(RegisteryMeta, mcs).__new__(mcs, name, bases, attr)
+        if not hasattr(cls, "_regmeta_class_registry"):
+            cls._regmeta_class_registry = {}
+            cls.factory = classmethod(_regmeta_instance_factory)
+
+        return cls
+
+    def __init__(cls, name, bases, attr):
+        if not inspect.isabstract(cls):
+            _regmeta_register_class(cls, cls, name)
+            for b in bases:
+                if hasattr(b, "_regmeta_class_registry"):
+                    _regmeta_register_class(b, cls, name)
+
+        super(RegisteryMeta, cls).__init__(name, bases, attr)
+
+        cls.regular_market_times = ProtectedDict(cls.regular_market_times)
+        cls.open_close_map = ProtectedDict(cls.open_close_map)
+
+        cls.special_market_open = cls.special_opens
+        cls.special_market_open_adhoc = cls.special_opens_adhoc
+
+        cls.special_market_close = cls.special_closes
+        cls.special_market_close_adhoc = cls.special_closes_adhoc
+
+
+class ProtectedDict(dict):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # __init__ is bypassed when unpickling, which causes __setitem__ to fail
+        # without the _INIT_RAN_NORMALLY flag
+        self._INIT_RAN_NORMALLY = True
+
+    def _set(self, key, value):
+        return super().__setitem__(key, value)
+
+    def _del(self, key):
+        return super().__delitem__(key)
+
+    def __setitem__(self, key, value):
+        if not hasattr(self, "_INIT_RAN_NORMALLY"):
+            return self._set(key, value)
+
+        raise TypeError(
+            "You cannot set a value directly, you can change regular_market_times "
+            "using .change_time, .add_time or .remove_time."
+        )
+
+    def __delitem__(self, key):
+        if not hasattr(self, "_INIT_RAN_NORMALLY"):
+            return self._del(key)
+
+        raise TypeError(
+            "You cannot delete an item directly. You can change regular_market_times "
+            "using .change_time, .add_time or .remove_time"
+        )
+
+    def __repr__(self):
+        return self.__class__.__name__ + "(" + super().__repr__() + ")"
+
+    def __str__(self):
+        try:
+            formatted = pformat(dict(self), sort_dicts=False)  # sort_dicts apparently not available < python3.8
+        except TypeError:
+            formatted = pformat(dict(self))
+
+        return self.__class__.__name__ + "(\n" + formatted + "\n)"
+
+    def copy(self):
+        return self.__class__(super().copy())