from datetime import datetime, timedelta
import calendar

def get_month_range(date=None):
    """獲取當(dāng)月的第一天和最后一天"""
    date = date or datetime.today()
    # 當(dāng)月第一天
    first_day = date.replace(day=1)
    
    # 當(dāng)月最后一天
    _, last_day_num = calendar.monthrange(date.year, date.month)
    last_day = date.replace(day=last_day_num)
    
    return first_day, last_day

# 使用示例
start, end = get_month_range()
print(f"當(dāng)月第一天: {start.strftime('%Y-%m-%d')}")
print(f"當(dāng)月最后一天: {end.strftime('%Y-%m-%d')}")

def generate_month_dates(date=None):
    """生成當(dāng)月所有日期列表"""
    start, end = get_month_range(date)
    current = start
    dates = []
    
    while current <= end:
        dates.append(current)
        current += timedelta(days=1)
    
    return dates

# 使用示例
dates = generate_month_dates()
print(f"當(dāng)月共{len(dates)}天: 從{dates[0].strftime('%m-%d')}到{dates[-1].strftime('%m-%d')}")

import pytz
from dateutil.relativedelta import relativedelta

def get_month_range_tz(timezone='Asia/Shanghai', date=None):
    """時(shí)區(qū)敏感的當(dāng)月范圍"""
    tz = pytz.timezone(timezone)
    now = datetime.now(tz) if date is None else date.astimezone(tz)
    
    # 當(dāng)月第一天
    first_day = now.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
    
    # 當(dāng)月最后一天
    next_month = first_day + relativedelta(months=1)
    last_day = next_month - timedelta(days=1)
    last_day = last_day.replace(hour=23, minute=59, second=59, microsecond=999999)
    
    return first_day, last_day

# 使用示例
shanghai_start, shanghai_end = get_month_range_tz('Asia/Shanghai')
newyork_start, newyork_end = get_month_range_tz('America/New_York')

print(f"上海當(dāng)月范圍: {shanghai_start} 至 {shanghai_end}")
print(f"紐約當(dāng)月范圍: {newyork_start} 至 {newyork_end}")

class BusinessMonthCalculator:
    """工作日范圍計(jì)算器"""
    def __init__(self, holidays=None):
        self.holidays = holidays or set()
    
    def add_holiday(self, date):
        """添加節(jié)假日"""
        self.holidays.add(date)
    
    def is_business_day(self, date):
        """檢查是否為工作日"""
        # 周末檢查
        if date.weekday() >= 5:  # 5=周六, 6=周日
            return False
        # 節(jié)假日檢查
        return date.date() not in self.holidays
    
    def get_business_days(self, month=None):
        """獲取當(dāng)月所有工作日"""
        month = month or datetime.today()
        start, end = get_month_range(month)
        current = start
        business_days = []
        
        while current <= end:
            if self.is_business_day(current):
                business_days.append(current)
            current += timedelta(days=1)
        
        return business_days

# 使用示例
calculator = BusinessMonthCalculator()
# 添加節(jié)假日
calculator.add_holiday(datetime(2023, 12, 25).date())  # 圣誕節(jié)

business_days = calculator.get_business_days()
print(f"12月工作日天數(shù): {len(business_days)}")

def fiscal_month_range(date=None, fiscal_start_day=26):
    """計(jì)算財(cái)務(wù)月范圍（上月26日到本月25日）"""
    date = date or datetime.today()
    
    if date.day >= fiscal_start_day:
        # 本月26日到下月25日
        start = datetime(date.year, date.month, fiscal_start_day)
        end_month = date.month + 1 if date.month < 12 else 1
        end_year = date.year if date.month < 12 else date.year + 1
        end = datetime(end_year, end_month, fiscal_start_day) - timedelta(days=1)
    else:
        # 上月26日到本月25日
        prev_month = date.replace(day=1) - timedelta(days=1)
        start = datetime(prev_month.year, prev_month.month, fiscal_start_day)
        end = datetime(date.year, date.month, fiscal_start_day) - timedelta(days=1)
    
    return start.date(), end.date()

# 使用示例
fiscal_start, fiscal_end = fiscal_month_range()
print(f"財(cái)務(wù)月范圍: {fiscal_start} 至 {fiscal_end}")

def payroll_period(date=None):
    """計(jì)算工資結(jié)算周期（上月26日到本月25日）"""
    date = date or datetime.today()
    # 結(jié)算日為每月25日
    if date.day >= 26:
        # 本月26日到下月25日
        start = datetime(date.year, date.month, 26)
        end_month = date.month + 1 if date.month < 12 else 1
        end_year = date.year if date.month < 12 else date.year + 1
        end = datetime(end_year, end_month, 25)
    else:
        # 上月26日到本月25日
        prev_month = date.replace(day=1) - timedelta(days=1)
        start = datetime(prev_month.year, prev_month.month, 26)
        end = datetime(date.year, date.month, 25)
    
    return start.date(), end.date()

# 使用示例
pay_start, pay_end = payroll_period()
print(f"工資結(jié)算周期: {pay_start} 至 {pay_end}")

import pandas as pd

def monthly_aggregation(data, date_col='date', agg_func='sum'):
    """月度數(shù)據(jù)聚合"""
    # 確保日期為datetime類型
    data[date_col] = pd.to_datetime(data[date_col])
    
    # 添加月份列
    data['month'] = data[date_col].dt.to_period('M')
    
    # 分組聚合
    aggregated = data.groupby('month').agg(agg_func)
    return aggregated

# 使用示例
# 創(chuàng)建測(cè)試數(shù)據(jù)
dates = pd.date_range('2023-01-01', '2023-12-31', freq='D')
values = np.random.rand(len(dates)) * 100
df = pd.DataFrame({'date': dates, 'value': values})

# 月度聚合
monthly_data = monthly_aggregation(df, agg_func={'value': 'mean'})
print("月度平均值:")
print(monthly_data.head())

def user_activity_analysis(activity_data):
    """用戶月度活躍度分析"""
    # 復(fù)制數(shù)據(jù)
    df = activity_data.copy()
    
    # 轉(zhuǎn)換日期
    df['date'] = pd.to_datetime(df['timestamp']).dt.date
    
    # 獲取當(dāng)月范圍
    current_month_start, current_month_end = get_month_range()
    
    # 篩選當(dāng)月數(shù)據(jù)
    mask = (df['date'] >= current_month_start.date()) & (df['date'] <= current_month_end.date())
    current_month_data = df[mask]
    
    # 計(jì)算活躍用戶
    active_users = current_month_data['user_id'].nunique()
    
    # 計(jì)算活躍天數(shù)
    active_days = current_month_data.groupby('user_id')['date'].nunique().mean()
    
    # 計(jì)算留存率
    if 'first_login' in df.columns:
        new_users = df[df['first_login'].dt.month == current_month_start.month]['user_id'].nunique()
        retained_users = len(set(df[df['first_login'].dt.month == current_month_start.month]['user_id']) &
                            set(current_month_data['user_id']))
        retention_rate = retained_users / new_users if new_users > 0 else 0
    else:
        retention_rate = None
    
    return {
        'active_users': active_users,
        'active_days': active_days,
        'retention_rate': retention_rate
    }

# 使用示例
# 模擬數(shù)據(jù)
data = {
    'user_id': [1,1,1,2,2,3,3,3,3],
    'timestamp': pd.date_range('2023-12-01', periods=9, freq='3D'),
    'first_login': [pd.Timestamp('2023-11-15')]*3 + [pd.Timestamp('2023-12-01')]*6
}
df = pd.DataFrame(data)

result = user_activity_analysis(df)
print(f"活躍用戶: {result['active_users']}")
print(f"平均活躍天數(shù): {result['active_days']:.2f}")
print(f"留存率: {result['retention_rate']:.2%}")

class SubscriptionSystem:
    """訂閱服務(wù)計(jì)費(fèi)系統(tǒng)"""
    def __init__(self):
        self.subscriptions = {}
    
    def add_subscription(self, user_id, start_date, billing_cycle='monthly'):
        """添加訂閱"""
        self.subscriptions[user_id] = {
            'start_date': start_date,
            'billing_cycle': billing_cycle,
            'next_billing_date': self._calculate_next_billing(start_date, billing_cycle)
        }
    
    def _calculate_next_billing(self, start_date, cycle):
        """計(jì)算下次計(jì)費(fèi)日期"""
        if cycle == 'monthly':
            return (start_date + relativedelta(months=1)).replace(day=start_date.day)
        elif cycle == 'quarterly':
            return (start_date + relativedelta(months=3)).replace(day=start_date.day)
        elif cycle == 'yearly':
            return (start_date + relativedelta(years=1)).replace(day=start_date.day)
        else:
            raise ValueError("不支持的計(jì)費(fèi)周期")
    
    def generate_monthly_invoices(self):
        """生成當(dāng)月賬單"""
        current_month_start, current_month_end = get_month_range()
        invoices = []
        
        for user_id, sub in self.subscriptions.items():
            # 檢查是否在計(jì)費(fèi)周期內(nèi)
            if current_month_start <= sub['next_billing_date'] <= current_month_end:
                invoices.append({
                    'user_id': user_id,
                    'amount': 100,  # 示例金額
                    'billing_date': sub['next_billing_date']
                })
                # 更新下次計(jì)費(fèi)日期
                sub['next_billing_date'] = self._calculate_next_billing(
                    sub['next_billing_date'], sub['billing_cycle']
                )
        
        return invoices

# 使用示例
system = SubscriptionSystem()
system.add_subscription('user1', datetime(2023, 11, 15), 'monthly')
system.add_subscription('user2', datetime(2023, 12, 10), 'monthly')

# 生成12月賬單
invoices = system.generate_monthly_invoices()
print(f"12月賬單數(shù)量: {len(invoices)}")

def calculate_trial_period(start_date, trial_days=14):
    """計(jì)算試用期結(jié)束日期"""
    from pandas.tseries.offsets import BDay
    
    # 工作日計(jì)算
    end_date = start_date + trial_days * BDay()
    
    # 確保在當(dāng)月范圍內(nèi)
    month_start, month_end = get_month_range(start_date)
    if end_date > month_end:
        end_date = month_end
    
    return end_date

# 使用示例
signup_date = datetime(2023, 12, 20)
trial_end = calculate_trial_period(signup_date)
print(f"試用期結(jié)束: {trial_end.strftime('%Y-%m-%d')}")

def monthly_project_plan(project_start, project_end):
    """生成項(xiàng)目月度計(jì)劃"""
    current = project_start.replace(day=1)
    plan = []
    
    while current <= project_end:
        # 獲取當(dāng)月范圍
        month_start = current.replace(day=1)
        _, last_day = calendar.monthrange(current.year, current.month)
        month_end = current.replace(day=last_day)
        
        # 調(diào)整項(xiàng)目邊界
        if month_start < project_start:
            month_start = project_start
        if month_end > project_end:
            month_end = project_end
        
        plan.append({
            'month': current.strftime('%Y-%m'),
            'start': month_start.date(),
            'end': month_end.date(),
            'duration': (month_end - month_start).days + 1
        })
        
        # 下個(gè)月
        current = (current.replace(day=1) + relativedelta(months=1))
    
    return plan

# 使用示例
project_start = datetime(2023, 11, 15)
project_end = datetime(2024, 2, 10)
plan = monthly_project_plan(project_start, project_end)

print("項(xiàng)目月度計(jì)劃:")
for p in plan:
    print(f"{p['month']}: {p['start']} 至 {p['end']} ({p['duration']}天)")

def monthly_resource_allocation(resources, start_date, end_date):
    """資源月度分配計(jì)劃"""
    # 生成月度范圍
    months = []
    current = start_date.replace(day=1)
    while current <= end_date:
        months.append(current.strftime('%Y-%m'))
        current += relativedelta(months=1)
    
    # 初始化分配表
    allocation = pd.DataFrame(index=resources, columns=months)
    
    # 計(jì)算每個(gè)資源每月分配天數(shù)
    for resource in resources:
        current = start_date
        while current <= end_date:
            month = current.strftime('%Y-%m')
            # 計(jì)算當(dāng)月天數(shù)
            if current.month == start_date.month and current.year == start_date.year:
                # 首月
                month_start = start_date
                _, last_day = calendar.monthrange(current.year, current.month)
                month_end = current.replace(day=last_day)
            elif current.month == end_date.month and current.year == end_date.year:
                # 末月
                month_start = current.replace(day=1)
                month_end = end_date
            else:
                # 完整月
                month_start = current.replace(day=1)
                _, last_day = calendar.monthrange(current.year, current.month)
                month_end = current.replace(day=last_day)
            
            # 計(jì)算工作天數(shù)
            days = (month_end - month_start).days + 1
            allocation.loc[resource, month] = days
            
            # 下個(gè)月
            current = (current.replace(day=1) + relativedelta(months=1))
    
    return allocation

# 使用示例
resources = ['開(kāi)發(fā)團(tuán)隊(duì)', '測(cè)試團(tuán)隊(duì)', '設(shè)計(jì)團(tuán)隊(duì)']
start_date = datetime(2023, 12, 10)
end_date = datetime(2024, 2, 20)

allocation = monthly_resource_allocation(resources, start_date, end_date)
print("資源月度分配表:")
print(allocation)

import numpy as np

def batch_month_ranges(dates):
    """批量計(jì)算日期所在月份范圍"""
    # 轉(zhuǎn)換為datetime數(shù)組
    dates = np.array(dates, dtype='datetime64')
    
    # 計(jì)算當(dāng)月第一天
    first_days = dates.astype('datetime64[M]')
    
    # 計(jì)算當(dāng)月最后一天
    next_months = first_days + np.timedelta64(1, 'M')
    last_days = next_months - np.timedelta64(1, 'D')
    
    return first_days, last_days

# 使用示例
test_dates = [
    '2023-01-15', '2023-02-28', '2023-12-01',
    '2024-01-01', '2024-02-29'  # 閏年測(cè)試
]

first_days, last_days = batch_month_ranges(test_dates)
for i, date in enumerate(test_dates):
    print(f"{date} 所在月份: {first_days[i]} 至 {last_days[i]}")

def efficient_month_dates(year, month):
    """高效生成當(dāng)月所有日期"""
    # 計(jì)算當(dāng)月天數(shù)
    _, num_days = calendar.monthrange(year, month)
    
    # 生成日期范圍
    start = datetime(year, month, 1)
    dates = [start + timedelta(days=i) for i in range(num_days)]
    
    return dates

# 性能對(duì)比
%timeit generate_month_dates(datetime(2023, 12, 1))  # 傳統(tǒng)方法
%timeit efficient_month_dates(2023, 12)  # 高效方法

# 所有操作前轉(zhuǎn)換為UTC
utc_time = datetime.now(pytz.utc)
# 操作后轉(zhuǎn)換回本地時(shí)間
local_time = utc_time.astimezone(pytz.timezone('Asia/Shanghai'))

# 安全獲取月末
def safe_month_end(year, month):
    try:
        return datetime(year, month, 31)
    except ValueError:
        try:
            return datetime(year, month, 30)
        except ValueError:
            return datetime(year, month, 28)  # 閏年處理在monthrange中

# 批量處理使用numpy
dates = np.array(['2023-01-01', '2023-02-01'], dtype='datetime64')
month_starts = dates.astype('datetime64[M]')

def get_month_range_safe(date=None):
    try:
        return get_month_range(date)
    except ValueError as e:
        # 處理無(wú)效日期
        print(f"日期錯(cuò)誤: {str(e)}")
        return None, None
    except TypeError:
        # 處理類型錯(cuò)誤
        print("無(wú)效日期類型")
        return None, None

def is_leap_year(year):
    """檢查閏年"""
    return year % 4 == 0 and (year % 100 != 0 or year % 400 == 0)

def february_days(year):
    """獲取二月天數(shù)"""
    return 29 if is_leap_year(year) else 28

def get_month_range(date=None):
    """
    獲取當(dāng)月的第一天和最后一天

    參數(shù):
        date: 基準(zhǔn)日期 (默認(rèn)今天)

    返回:
        (first_day, last_day) 當(dāng)月第一天和最后一天的datetime對(duì)象

    示例:
        >>> get_month_range(datetime(2023, 12, 15))
        (datetime(2023, 12, 1), datetime(2023, 12, 31))
    """
    # 實(shí)現(xiàn)代碼