python 获取当前时间字符串

import datetime


def get_datetime_str(style='dt'):
    cur_time = datetime.datetime.now()

    date_str = cur_time.strftime('%y%m%d')
    time_str = cur_time.strftime('%H%M%S')

    if style == 'data':
        return date_str
    elif style == 'time':
        return time_str
    else:
        return date_str + '_' + time_str


if __name__ == '__main__':
    ds_str = get_datetime_str()
    print(ds_str)
    ds_str = get_datetime_str('time')
    print(ds_str)