classify specific file type into a txt within current directory

classify_file_type_in_current_dir.py

# -*- coding: utf-8 -*-

import os
import re


def classify_file_type_in_current_dir(file_name, file_type):
	f = open(file_name,'wb')
	for root, dirs, files in os.walk(os.getcwd(), topdown=True):
		for name in files:
			if re.search(file_type, name):
				f.write(name + '\n')
		f.flush()
	f.close()

file_type = '.py'
file_txt = 'file_list_py.txt'

classify_file_type_in_current_dir(file_txt, file_type)