jpholidayで独自の休日を追加できない

日本の祝日を取得するライブラリ、jpholidayを使い
独自の休日を追加しようと試みましたが、サンプルコードをそのまま入力してもエラーで弾かれます。
datetimeのライブラリ構造の問題かと思い、試行錯誤しましたが解決に至りませんでした。

どこを修正すれば正常に動くのかアドバイスを頂けると幸いです。

import jpholiday
import datetime

class TestHoliday(jpholiday.OriginalHoliday):
    def _is_holiday(self, date):
        if date == datetime.date(2020, 2, 9):
            return True
        return False

    def _is_holiday_name(self, date):
        return '特別休暇'

jpholiday.is_holiday(datetime.date(2020, 2, 9))

エラー内容：

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-60-430cfc371103> in <module>
----> 1 jpholiday.is_holiday(datetime.date(2020, 2, 9))

~/opt/anaconda3/lib/python3.8/site-packages/jpholiday/jpholiday.py in is_holiday(date)
     33 
     34     for holiday in registry.RegistryHolder.get_registry():
---> 35         if holiday.is_holiday(date):
     36             return True
     37 

~/opt/anaconda3/lib/python3.8/site-packages/jpholiday/registry.py in is_holiday(self, date)
     52 
     53     def is_holiday(self, date):
---> 54         if self._is_holiday(date):
     55             return True
     56         else:

<ipython-input-7-37a573019c6d> in _is_holiday(self, date)
     12 
     13     def _is_holiday(self, date):
---> 14         if date in [datetime.strptime(holiday,'%Y-%m-%d').date() for holiday in self.original_holidays.keys()]:
     15             return True
     16         return False

<ipython-input-7-37a573019c6d> in <listcomp>(.0)
     12 
     13     def _is_holiday(self, date):
---> 14         if date in [datetime.strptime(holiday,'%Y-%m-%d').date() for holiday in self.original_holidays.keys()]:
     15             return True
     16         return False

AttributeError: module 'datetime' has no attribute 'strptime'

melian

2022/01/02 02:23 編集

質問にあるコードでは、 if date == datetime.date(2020, 2, 9): となっていますが、エラーメッセージでは if date in [datetime.strptime(holiday,'%Y-%m-%d').date() ... となっています。コードの方で実行するとエラーにはなりません。ちなみに、datetime.strptime ではなく datetime.datetime.strptime になります。

hiro_python

2022/01/02 02:25

新規でNotebookを作った所、解決しました。ありがとうございます！

行動規範の内容に同意します

回答2件

Jupyterを使用している場合、以前に実行したコードが残っていて想定外の動作をすることがあります。

この場合は、以下のコードが残っているのがエラーの原因です。

python
1#独自の休日をファイルから読み込む
2import jpholiday
3import configparser
4
5class TestHoliday(jpholiday.OriginalHoliday):
6    original_holidays = {}
7
8    config = configparser.ConfigParser()
9    config.read('holidays.ini')
10    if 'HOLIDAYS' in config:
11        original_holidays = config['HOLIDAYS']
12
13    def _is_holiday(self, date):
14        if date in [datetime.strptime(holiday,'%Y-%m-%d').date() for holiday in self.original_holidays.keys()]:
15            return True
16        return False
17
18    def _is_holiday_name(self, date):
19        if date.strftime('%Y-%m-%d') in self.original_holidays.keys():
20            return self.original_holidays[date.strftime('%Y-%m-%d')]
21        else:
22            return None

新しいノートブックを作ってそこで実行するか、

python
1del TestHoliday
2class TestHoliday(jpholiday.OriginalHoliday):
3    def _is_holiday(self, date):
4        if date == datetime.date(2020, 2, 9):
5            return True
6        return False
7
8    def _is_holiday_name(self, date):
9        return '特別休暇'
10
11jpholiday.is_holiday(datetime.date(2020, 2, 9))