截取文件后缀名

# 截取给的文件名的后缀名

# 在这里要注意可能出现的文件是否有后缀名,例如.abc和abc. 其实这两种是没有后缀名的,所以判断切片'.'的索引位置必须有所限制
def foo(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 None


if __name__ == '__main__':
    print(foo('hello.py',True))
    print(foo('a.b.cdd'))
    print(foo('.abc'))
    print(foo('abc.'))