pandas-market-calendars 4.3.1__py3-none-any.whl → 4.3.3__py3-none-any.whl

pandas_market_calendars/calendars/tsx.py

@@ -1,159 +1,181 @@
-from datetime import time
-
-import pandas as pd
-from pandas.tseries.holiday import AbstractHolidayCalendar, DateOffset, GoodFriday, Holiday, MO, weekend_to_monday
-from pytz import timezone
-from itertools import chain
-
-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 tz(self):
-        return timezone('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,
+)
+from pytz import timezone
+
+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 tz(self):
+        return timezone("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,6 +1,7 @@
 import inspect
 from pprint import pformat
 
+
 def _regmeta_instance_factory(cls, name, *args, **kwargs):
     """
     :param cls(RegisteryMeta): registration meta class
@@ -13,16 +14,20 @@ def _regmeta_instance_factory(cls, name, *args, **kwargs):
         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()))
+            "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 hasattr(regcls, "aliases"):
         if regcls.aliases:
             for alias in regcls.aliases:
                 cls._regmeta_class_registry[alias] = regcls
@@ -39,7 +44,7 @@ class RegisteryMeta(type):
 
     def __new__(mcs, name, bases, attr):
         cls = super(RegisteryMeta, mcs).__new__(mcs, name, bases, attr)
-        if not hasattr(cls, '_regmeta_class_registry'):
+        if not hasattr(cls, "_regmeta_class_registry"):
             cls._regmeta_class_registry = {}
             cls.factory = classmethod(_regmeta_instance_factory)
 
@@ -49,7 +54,7 @@ class RegisteryMeta(type):
         if not inspect.isabstract(cls):
             _regmeta_register_class(cls, cls, name)
             for b in bases:
-                if hasattr(b, '_regmeta_class_registry'):
+                if hasattr(b, "_regmeta_class_registry"):
                     _regmeta_register_class(b, cls, name)
 
         super(RegisteryMeta, cls).__init__(name, bases, attr)
@@ -65,7 +70,6 @@ class RegisteryMeta(type):
 
 
 class ProtectedDict(dict):
-
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         # __init__ is bypassed when unpickling, which causes __setitem__ to fail
@@ -82,30 +86,32 @@ class ProtectedDict(dict):
         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.")
+        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")
+        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__() + ")"
+        return self.__class__.__name__ + "(" + super().__repr__() + ")"
 
     def __str__(self):
         try:
-            formatted = pformat(dict(self), sort_dicts= False) # sort_dicts apparently not available < python3.8
+            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)"
+        return self.__class__.__name__ + "(\n" + formatted + "\n)"
 
     def copy(self):
         return self.__class__(super().copy())
-
-
-
-