# -*- coding: utf-8 -*-
|
"""
|
@File : day.py
|
@Time : 2023/1/14 17:06
|
@Author : geekbing
|
@LastEditTime : -
|
@LastEditors : -
|
@Description : -
|
"""
|
|
import datetime
|
|
from httprunner.builtin import get_ts_int
|
|
|
def get_day(days: int = 0, **kwargs):
|
"""
|
获取日期，默认有连接符
|
:param days: 间隔时间
|
:param kwargs:
|
:return:
|
|
>>> get_day()
|
# 今天的日期 2023-02-01
|
|
>>> get_day(1)
|
# 明天的日期
|
|
>>> get_day(-1)
|
# 昨天的日期
|
|
>>> get_day(h=9,m=15,s=30)
|
# 今天的9时15分30秒 20230116 9:15:30
|
"""
|
d = datetime.timedelta(days)
|
n = datetime.datetime.now()
|
time_str = f"%Y-%m-%d"
|
if kwargs:
|
h = kwargs.get("h", "00")
|
m = kwargs.get("m", "00")
|
s = kwargs.get("s", "00")
|
time_str = f"{time_str} {h}:{m}:{s}"
|
return (n + d).strftime(time_str)
|
|
|
def get_month(month_delta: int = 0, base_ts=get_ts_int) -> str:
|
"""生成年月
|
|
:param month_delta: 间隔月份
|
:param base_ts: 基准时间
|
:return:
|
|
>>> base_ts = 1629439316 # 2021-08-20 14:01:56
|
>>> get_month(5, base_ts)
|
'202201'
|
|
>>> get_month(17, base_ts)
|
'202301'
|
|
>>> get_month(-20, base_ts)
|
'201912'
|
|
>>> get_month(-8, base_ts)
|
'202012'
|
|
>>> get_month(base_ts=base_ts)
|
'202108'
|
|
>>> get_month(1, base_ts)
|
'202109'
|
|
>>> get_month(-1, base_ts)
|
'202107'
|
"""
|
if callable(base_ts):
|
base_ts = base_ts()
|
dt = datetime.datetime.fromtimestamp(base_ts)
|
year, month = dt.year, dt.month
|
|
# 超过12个月，先计算年份，然后再对月份差取余
|
if abs(month_delta) > 12:
|
year_delta = abs(month_delta) // 12
|
if month_delta > 0:
|
year += year_delta
|
month_delta = month_delta % 12
|
else:
|
year -= year_delta
|
month_delta = -(abs(month_delta) % 12)
|
|
# 月份差是正数，有两种情况
|
# 1.月份差和当前的月份相加 > 12, 年份+1, 月份等于超过的减去12
|
# 2.月份差和当前月份相加 <= 12, 直接相加
|
if month_delta >= 0:
|
if month + month_delta > 12:
|
year += 1
|
month = month + month_delta - 12
|
else:
|
month += month_delta
|
else:
|
# 月份差是负数, 有两种情况
|
# 1.月份差 <= 0, 年份-1, 月份从12开始倒退
|
# 2.直接倒退月份
|
if month - abs(month_delta) <= 0:
|
year -= 1
|
# month + month_delta 是负数
|
month = 12 + (month + month_delta)
|
else:
|
month += month_delta
|
month = str(month).rjust(2, "0")
|
|
return f"{year}{month}"
|
|
|
def get_week(weeks: int = 0) -> str:
|
"""
|
生成一年中的第n周
|
:param weeks:
|
:return:
|
"""
|
week_delta = 7 * 24 * 60 * 60 * weeks
|
ts = get_ts_int(week_delta)
|
dt = datetime.datetime.fromtimestamp(ts)
|
year, week_number, _ = dt.isocalendar()
|
# 确保周数是两位数，例如 '01', '02', ..., '11', '12', ...
|
week_str = str(week_number).zfill(2)
|
# '202333' 2023年的第33周
|
return f"{year}{week_str}"
|
|
|
def get_month_format(month_delta: int = 0, base_ts=get_ts_int):
|
"""
|
|
:param month_delta:
|
:param base_ts:
|
:return:
|
"""
|
month = get_month(month_delta, base_ts)
|
return f"{month[:4]}年{month[4:]}月"
|
|
|
def get_week_format(weeks: int) -> str:
|
"""
|
|
:param weeks:
|
:return:
|
"""
|
week = get_week(weeks)
|
return f"{week[:4]}年{week[4:]}周"
|
