@

1.背景

最近專案是國際專案，所以需要經常需要用到UTC時間和local時間的轉換。
所以整理了一下時間戳工具類，方便使用。
這裡主要用到的包就是datatime、time、pytz。

2. 遇到的坑

直接看測試案例

tzinfo=pytz.timezone("Asia/Shanghai")
dtStr = "2023-05-28 00:00:00"
dt1 = datetime.datetime.strptime(dtStr,"%Y-%m-%d %H:%M:%S").replace(tzinfo=tzinfo)
dt2 = dt_tz = tzinfo.localize(datetime.datetime.strptime(dtStr, '%Y-%m-%d %H:%M:%S'))
print(dt1)
print(dt2)

輸出結果：

2023-05-28 00:00:00+08:06
2023-05-28 00:00:00+08:00

這裡第一種方法dt1中的輸出時區帶有06分的差異。

3. 一些小案例

3.1 當前日期、日期時間、UTC日期時間

print(DateUtil.currDate())
print(DateUtil.currDateTime())
print(DateUtil.currDateTime("UTC"))

輸出結果：

2023-05-29
2023-05-29 10:44:27
2023-05-29 02:44:27

3.2 昨天、昨天UTC日期、昨天現在這個時間點的時間戳

print(DateUtil.yesterday())
print(DateUtil.yesterdayUTC()) 
print(DateUtil.yesterday_ts())

輸出結果：

2023-05-28
2023-05-28  #這裡如果是早上8點以前執行的就不一樣
1685242026391

3.3 日期轉時間戳

print(DateUtil.date2ts("2023-05-28"))
print(DateUtil.datetime2ts("2023-05-28 10:01:01"))
print(DateUtil.datetime2ts("2023-05-28 10:01:01","UTC"))
print(DateUtil.datetime2ts("2023-05-28 10:01:01","Asia/Shanghai"))

輸出結果：

1685203200000
1685239261000
1685268061000 #如果使用的是UTC格式，則時間戳不一樣
1685239261000

3.4 時間戳轉日期

print(DateUtil.timestamp2Date("1685239261000"))
print(DateUtil.timestamp2Date("1685239261")) # 這裡自動判斷是毫秒還是秒
print(DateUtil.timestamp2Date("1685239261","UTC")) # 這裡指定輸出的日期的時區是UTC格式
print(DateUtil.timestamp2Date(1685239261)) #int型別或者字串都可以轉行
print(DateUtil.timestamp2Date(1685239261000,"UTC")) #int型別或者字串都可以轉行

輸出結果：

2023-05-28 10:01:01
2023-05-28 10:01:01
2023-05-28 02:01:01
2023-05-28 10:01:01
2023-05-28 02:01:01

3.5 日期加減、小時的加減

print(DateUtil.dateAdd("2023-05-28",1)) #加
print(DateUtil.dateAdd("2023-05-28",-1)) #減
print(DateUtil.datetimeAdd("2023-05-28 10:01:01",-1)) #日期時間的加減
print(DateUtil.hourAdd("2023-05-28 10:01:01",1)) # 小時的加減
print(DateUtil.hourAdd("2023-05-28 10:01:01",-1)) # 小時的加減

輸出結果：

2023-05-29
2023-05-27
2023-05-27 10:01:01
2023-05-28 11:01:01
2023-05-28 09:01:01

4. 總結

剛開始思緒有點亂，剛好趁機會好好的整理了一遍思路。
注意: 當使用日期進行格式轉換時，需要確定這個日期對應的時區。

• 時區的設定
  python的日期中時區對應的有個屬性timezone。
  我這裡主要是通過pytz.timezone("時區字串")來設定時區的。
  比如，要將日期轉為時間戳，首先要指定這個日期的時區屬性。

tzinfo = pytz.timezone("Asia/Shanghai")

• 日期設定
  然後通過localize方法設定這個日期的時區。

dt_tz = tzinfo.localize(dt)

再進行其他的轉換。
但是千萬要注意，不要使用datetime中的.replace方法來設定時區。

# 這樣設定會導致會導致輸出結果`2023-05-28 00:00:00+08:06` 帶有一個6分鐘的差異
dt_tz =datetime.datetime.strptime("2023-05-28 00:00:00","%Y-%m-%d %H:%M:%S").replace(tzinfo=tzinfo)  

5. 完整的編碼

# -*- coding: utf-8 -*-
# @Time    : 2023/3/28  10:04
# @Author  : King
# @Software: PyCharm
# @Describe: 
# -*- encoding:utf-8 -*-
import datetime
import time,pytz

_timezone_utc = "UTC"
_timezone_cn = "Asia/Shanghai"

"""
DateTime日期轉時間戳(日期時區為上海)
"""
def datetime2ts(date, timezone=_timezone_cn):
    if isinstance(date, datetime.datetime):
        return int(date.timestamp() * 1000)
    else:
        date = date + " 00:00:00" if len(date) <= 10 else date
        dt = datetime.datetime.strptime(date, '%Y-%m-%d %H:%M:%S')
        dt_tz = pytz.timezone(timezone).localize(dt)
    return int(dt_tz.timestamp()*1000)

"""
DateTime日期轉時間戳(日期時區為UTC)
"""
def utcDatetime2ts(date):
    return datetime2ts(date,_timezone_utc)

"""
Date日期轉時間戳
"""
def date2ts(date,timezone=_timezone_cn):
    if isinstance(date,datetime.datetime):
        return int(date.timestamp()*1000)
    else:
        dt = datetime.datetime.strptime(date, '%Y-%m-%d')
        dt_tz = pytz.timezone(timezone).localize(dt)
        return int(dt_tz.timestamp()*1000)

"""
時間戳轉日期
"""
def _ts2Date(timestamp,timezone,date_format="%Y-%m-%d %H:%M:%S"):
    tz = pytz.timezone(timezone)
    timestamp_s = int(timestamp) if len(str(timestamp)) <= 10 else int(timestamp)/1000
    dt_tz = datetime.datetime.fromtimestamp(timestamp_s,tz)
    return dt_tz.strftime(date_format)

"""
時間戳轉日期
"""
def timestamp2Date(timestamp,timezone=_timezone_cn):
    return _ts2Date(timestamp,timezone)

"""
獲取當前日期 2023-03-28
"""
def currDate(timezone=_timezone_cn):
    return datetime.datetime.now(pytz.timezone(_timezone_utc)).astimezone(pytz.timezone(timezone)).strftime('%Y-%m-%d')
"""
當前UTC日期
"""
def currUTCDate():
    return datetime.datetime.now(pytz.timezone(_timezone_utc)).strftime('%Y-%m-%d')

"""
昨日
"""
def yesterday(timezone=_timezone_cn):
    return dateAdd(currDate(timezone),-1)
"""
UTC昨日
"""
def yesterdayUTC():
    return dateAdd(currUTCDate(),-1)
"""
當前日期時間
"""
def currDateTime(timezone=_timezone_cn):
    return datetime.datetime.now(pytz.timezone(timezone)).strftime('%Y-%m-%d %H:%M:%S')

"""
當前時間戳
"""
def currTimestamp():
    return int(time.time()*1000)

"""
昨天時間戳
"""
def yesterday_ts():
    return currTimestamp() - 24*3600*1000

"""
日期的增減
"""
def dateAdd(date, n):
    dt = datetime.datetime.strptime(date, "%Y-%m-%d")
    return (dt + datetime.timedelta(days=n)).strftime("%Y-%m-%d")
"""
日期時間的增減
"""
def datetimeAdd(date,n):
    dt = datetime.datetime.strptime(date,'%Y-%m-%d %H:%M:%S')
    return (dt + datetime.timedelta(days=n)).strftime('%Y-%m-%d %H:%M:%S')

"""
小時的增減
"""
def hourAdd(date,n):
    dt = datetime.datetime.strptime(date, '%Y-%m-%d %H:%M:%S')
    return  (dt + datetime.timedelta(hours=n)).strftime('%Y-%m-%d %H:%M:%S')


"""
timezone：預設是Aisa/Shanghai時區
指定日期的上一個小時的時間戳
"""
def lastHourTimestamp(date,timezone=_timezone_cn):
    if(len(date) == 10):
        ts = date2ts(date,timezone)
    else:
        ts = datetime2ts(date,timezone)
    return ts - ts % 3600000 - 3600*1000