python 判断当前日期是否为股票交易日

"""
需要先安装pip install chinesecalendar
"""
from chinese_calendar import is_workday
from datetime import datetime


# 得到当前日期是否为股票交易日
def is_trade_day(date):
    if is_workday(date):
        if date.isoweekday() < 6:
            return True
    return False


date = '2021-02-07'
date = datetime.datetime.strptime(date, '%Y-%m-%d').date()

print(is_trade_day(date))
print(is_trade_day(datetime.datetime.now()))