Pythonでフォルダ以下のファイル一覧取得

指定フォルダ以下にあるファイル一覧を取得し、特定拡張子のファイルだけ表示する。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


import glob
import os.path


#-------------------------------------------------
## main ###
#-------------------------------------------------
if __name__=='__main__':

    # 現在の位置
    path = "./*"

    files = glob.glob(path)
    for file in files:
        # 拡張子取得
        root, ext = os.path.splitext(file)
        # .pyだけ表示
        if ext == ".py":
            print(file)