python应用-获取文件后缀名

def get_suffix(filename,has_dot=False):
    """
    获取文件后缀名
    :param filename: 文件名
    :param has_dot: 后缀名是否带.
    :return: 文件后缀名
    """
    pos=filename.rfind('.')
    if 0<pos<len(filename)-1:
        index=pos if has_dot else pos+1
        return filename[index:]
    else:
        return ''
if __name__ == '__main__':
    print(get_suffix('hello.jpg'))
    print(get_suffix('abc.',True))
    print(get_suffix('a.'))
    print(get_suffix('hello.c',True))
    print(get_suffix('adc.abc'))

  结果: jpg .c abc

posted @ 2018-03-11 22:00  圆柱模板  阅读(384)  评论(0编辑  收藏  举报