輸入兩個時間，求他們的時間差

import time


# a = '2019-05-29 09:33:00'
# #轉換為時間數組
# b = time.strptime(a,"%Y-%m-%d %H:%M:%S")
# #轉換成時間戳
# c = int(time.mktime(b))
# print(c)
# print(type(c))
#divmod()求商和餘數
# print(divmod(5,2))

def time_stamp(time1,time2):
    "把輸入的時間轉為時間戳"
    #把time1.time2轉成時間數組
    timeArray1 = time.strptime(time1,"%Y-%m-%d %H:%M:%S")
    timeArray2 = time.strptime(time2,"%Y-%m-%d %H:%M:%S")
    #在把時間數組轉為時間戳
    timestamp1 = int(time.mktime(timeArray1))
    timestamp2 = int(time.mktime(timeArray2))
    time_minus(timestamp1,timestamp2)
    # return timestamp1,timestamp2
def time_minus(timestamp1,timestamp2):
    "輸入兩個時間戳進行相減"
    if timestamp1 >= timestamp2:
        seconds = timestamp1 - timestamp2
    else:
        seconds = timestamp2 - timestamp1
    m,s = divmod(seconds,60)
    h,m = divmod(m,60)
    # return h,m,s
    print("%02d:%02d:%02d" % (h, m, s))
if __name__ == "__main__":
    time_stamp(time1 = "2019-05-29 10:56:22",time2 ="2019-05-29 09:55:00")