时间处理脚本（python）

妖狐艹你老母 2022-12-31 05:29 198阅读 0赞

时间处理相关方法

#!/usr/bin/python
    # -*- coding:utf-8 -*-
    
    import time
    import datetime
    from datetime import timedelta, datetime
    import calendar
    
    TODAY = datetime.today()
    YESTERDAY = TODAY - timedelta(days=1)
    TOMORROW = TODAY + timedelta(days=1)
    
    # 把datetime转成字符串
    def datetime_toString(dt, format="%Y-%m-%d %H:%M:%S"):
        return dt.strftime(format)
    
    # 把字符串转成datetime
    def string_toDatetime(string, format="%Y-%m-%d %H:%M:%S"):
        return datetime.datetime.strptime(string, format)
    
    # 把重新格式化时间字符串
    def string_time_format(string, srcformat="%Y%m%d", target_format="%Y-%m-%d %H:%M:%S"):
        dt = string_toDatetime(string, srcformat)
        target_string = datetime_toString(dt, target_format)
        return target_string
    
    
    # 把字符串转成时间戳形式
    def string_toTimestamp(strTime):
        return time.mktime(string_toDatetime(strTime).timetuple())
    
    
    # 把时间戳转成字符串形式
    def timestamp_toString(stamp):
        return time.strftime("%Y-%m-%d-%H", time.localtime(stamp))
    
    
    def is_month_range(tax_start, tax_end):
        if tax_start.day != 1:
            return False
        if tax_start.month != tax_end.month or tax_start.year != tax_end.year:
            return False
        month_end = (tax_start + timedelta(days=31)).replace(day=1) - timedelta(days=1)
        return month_end.day == tax_end.day
    
    
    def is_quarter_range(tax_start, tax_end):
        if tax_start.day != 1:
            return False
        if tax_start.month not in [1, 4, 7, 10]:
            return False
        if tax_start.month + 2 != tax_end.month or tax_start.year != tax_end.year:
            return False
        quarter_end = (tax_start + timedelta(days=93)).replace(day=1) - timedelta(days=1)
        return quarter_end.day == tax_end.day
    
    
    def is_half_year_range(tax_start, tax_end):
        if tax_start.day != 1:
            return False
        if tax_start.year != tax_end.year:
            return False
        if tax_start.month == 1:
            return tax_end.month == 6 and tax_end.day == 30
        elif tax_start.month == 7:
            return tax_end.month == 12 and tax_end.day == 31
        return False
    
    
    def is_year_range(tax_start, tax_end):
        return all([
            tax_start.year == tax_end.year,
            tax_start.month == 1,
            tax_start.day == 1,
            tax_end.month == 12,
            tax_end.day == 31
        ])
    
    
    def ensure_datetime(o, format="%Y-%m-%d %H:%M:%S"):
        if isinstance(o, str):
            return datetime.strptime(o, format)
        elif isinstance(o, datetime):
            return o
        else:
            return o
    
    
    def get_month_date(today):
        today = ensure_datetime(today)
        year = today.year
        month = today.month
        start_day = 1
        if today.month == 1:
            year = today.year - 1
            month = 12
            _, last_day = calendar.monthrange(today.year - 1, month)
        else:
            _, last_day = calendar.monthrange(year, month)
        return datetime(year, month, start_day), datetime(year, month, last_day)
    
    
    def get_lastseason_date(today):
        today = ensure_datetime(today)
        quarter = int((today.month - 1) / 3 + 1)
        year = today.year
        start_day = 1
        stop_day = 31
        if quarter == 1:
            year = year - 1
            month = 12
        elif quarter == 2:
            year = today.year
            month = 3
        elif quarter == 3:
            month = 6
            stop_day = 30
        else:
            month = 9
            stop_day = 30
    
        return datetime(year, month-3, start_day), datetime(year, month, stop_day)
    
    
    def get_year_date(today):
        today = ensure_datetime(today)
        return datetime(today.year-1, 1, 1), datetime(today.year-1, 12, 31)
    
    
    def get_half_year_date(today):
        today = ensure_datetime(today)
        if today.month <= 5:
            return datetime(today.year-1, 6, 1), datetime(today.year-1, 12, 31)
        else:
            return datetime(today.year, 1, 1), datetime(today.year, 5, 31)
    
    
    def get_declaration_period(frequency, today):
        if frequency == "year":
            return get_year_date(today)
        if frequency == "quarter":
            return get_lastseason_date(today)
        if frequency == "month":
            return get_month_date(today)
        if frequency == "count":
            return get_month_date(today)
        if frequency == "half_year":
            return get_half_year_date(today)
    
    if __name__ == "__main__":
        today = datetime.now()
        next_year = datetime(today.year + 1, today.month,
                                      today.day, today.hour,
                                      today.minute, today.second)
        aa = {'tax_start_date': '2020-07-01 00:00:00'}
        c = time.strftime('%Y-%m-%d', time.strptime(aa['tax_start_date'], '%Y-%m-%d %H:%M:%S'))