import datetime
from enum import Enum
from seatable_api import dateutils, Base, context

# 一個(gè) Base 的授權(quán)信息
SERVER_URL = context.server_url or 'https://cloud.seatable.cn'
API_TOKEN = context.api_token or 'dd46f9ca0172a850a0922107a6b2e6b99932b040'

# 1. 定義中國(guó)的節(jié)假日概況
class Holiday(Enum):
    new_years_day = "元旦"
    spring_festival = "春節(jié)"
    tomb_sweeping_day = "清明"
    labour_day ="勞動(dòng)節(jié)"
    dragon_boat_festival = "端午"
    national_day = "國(guó)慶節(jié)"
    mid_autumn_festival = "中秋"

# 2. 列出節(jié)假日列表， 此處可以去查詢(xún)?nèi)諝v，就不一一列出了
holidays = {
    datetime.date(year=2022, month=1, day=1): Holiday.new_years_day.value,
    datetime.date(year=2022, month=1, day=2): Holiday.new_years_day.value,
    datetime.date(year=2022, month=1, day=3): Holiday.new_years_day.value,
    datetime.date(year=2022, month=1, day=31): Holiday.spring_festival.value,
    datetime.date(year=2022, month=2, day=1): Holiday.spring_festival.value,
    .....
}

# 3.列出調(diào)休日的列表，及周六日為工作日的列表
workdays = {
    datetime.date(year=2022, month=1, day=29): Holiday.spring_festival.value,
    datetime.date(year=2022, month=1, day=30): Holiday.spring_festival.value,
    datetime.date(year=2022, month=4, day=2): Holiday.tomb_sweeping_day.value,
    datetime.date(year=2022, month=4, day=24): Holiday.labour_day.value,
    datetime.date(year=2022, month=5, day=7): Holiday.labour_day.value,
    datetime.date(year=2022, month=10, day=8): Holiday.national_day.value,
    datetime.date(year=2022, month=10, day=9): Holiday.national_day.value,
    ....
}

# 4. 定義是否是工作日
def is_workday(date):
    '''
    工作日定義：
    1. 日期在workdays字典的key中
    2. 星期是周一到周五且不在holidays字典的key中
    '''
    date = _validate_date(date)
    weekday = date.weekday()
    return bool(date in workdays.keys() or (weekday <= 4 and date not in holidays.keys()))
    
# 5. 計(jì)算兩個(gè)日期之間的工作日， 此處返回的是工作日的列表， 該列表的長(zhǎng)度即是工作日的天數(shù)
def get_workdays(start, end):
    """
    獲取兩個(gè)日期之間的工作日，返回datetime的列表
    """
    start, end = _validate_date(start, end)
    return list(filter(is_workday, get_dates(start, end)))
    
# 6. 將結(jié)果寫(xiě)入 SeaTable

def calculate_base_workdays(base, table_name):
    '''
    通過(guò)seatable表格中的，開(kāi)始日期, 結(jié)束日期， 計(jì)算兩個(gè)日期間工作日的天數(shù)，并把其更新到該行的
    工作日字段中
    '''

    for row in base.list_rows(table_name):
        row_id = row.get('_id')
        start_date = row.get("開(kāi)始日期")
        end_date = row.get("結(jié)束日期")
        if not (start_date and end_date):
            continue
        try:
            work_day_list = get_workdays(start_date, end_date)
            # 兩個(gè)日期間的工作日天數(shù)
            work_day_counts = len(work_day_list)
            cell_value = row.get("工作日")
            if cell_value == work_day_counts:
                continue
            base.update_row(
                table_name,
                row_id,
                {
                    "工作日": work_day_counts
                }
            )
        except Exception as e:
            print("start date: %s, end date: %s, error: %s" % (start_date, end_date, e) )
            continue
            
base = Base(API_TOKEN, SERVER_URL)
base.auth()
calculate_base_workdays(base, "工作任務(wù)安排")