Python中常用模块之datetime

记录一下datetime模块使用方法。
题目（该题目来自廖雪峰老师）：假设你获取了用户输入的日期和时间如2015-1-21 9:01:30，以及一个时区信息如UTC+5:00，均是str，请编写一个函数将其转换为timestamp。
代码如下：

import re
from datetime import datetime, timezone, timedelta


def to_timestamp(dt_str, tz_str):
    tz = re.match(r'(\w{3})([+-]\d{0,2}):(\d{0,2})', tz_str).group(2)    # 使用正则获取用户传入的时区（这里的方法比较笨，后边再继续优化~）
    tz_utc = timezone(timedelta(hours=int(tz)))    # 定义时区
    dt = datetime.strptime(dt_str, '%Y-%m-%d %H:%M:%S').replace(tzinfo=tz_utc)    # 将str转换为datetime并设置时区
    return dt.timestamp()    # 返回时间的timestamp

# 使用assert进行测试
t1 = to_timestamp('2015-6-1 08:10:30', 'UTC+7:00')
assert t1 == 1433121030.0, t1

t2 = to_timestamp('2015-5-31 16:10:30', 'UTC-09:00')
assert t2 == 1433121030.0, t2