python 输出秒 转成时分秒格式(1时40分21秒)

def second_string_time(seconds):
    if seconds < 60:
        return(str(seconds) + '秒')
    format_list = ['时','分','秒']
    str_time =  str(datetime.timedelta(seconds=seconds)).split(':')
    time_string = ''
    for i,v in enumerate(str_time):
        int_v = int(v)
        if int_v != 0:
            time_string += str(int_v) + format_list[i]
    return time_string

print(second_string_time(6021))