返回文件后戳名——经典案例

返回文件后戳名——经典案例

"""
    设计函数，返回指定文件的后缀名
"""


def get_suffix(filename: str, has_dot=True):
    # 找出文件名最后点的位置
    position = filename.rfind('.')
    return filename[position:] if has_dot else filename[position + 1:]

suffix = get_suffix('hello.py')
print(suffix)

代码下载.