python--廖雪峰 datetime练习题

假设你获取了用户输入的日期和时间如2015-1-21 9:01:30，以及一个时区信息如UTC+5:00，均是str，请编写一个函数将其转换为timestamp：
下面的解法只适合北京时间。可以进一步优化。思想是将用户输入的时间换算为北京时间，然后调用datetime.timestamp()求解。

import re
from datetime import datetime, timezone, timedelta

def to_timestamp(dt_str, tz_str):
    cday=datetime.strptime(dt_str,'%Y-%m-%d %H:%M:%S')
    if list(tz_str)[3]=='+':
        ztime=8-int(re.match(r'^UTC(\+|-)(1[0-2]|[0-9]|0[0-9])\:00$',tz_str).group(2))
    else:
        ztime=int(re.match(r'^UTC(\+|-)(1[0-2]|[0-9]|0[0-9])\:00$',tz_str).group(2))+8
    cday=cday+timedelta(hours=ztime)
    print(cday.timestamp())

附一个获取时区的方法。