<Python>1GB以上のファイルを探す

前提

自分でPythonを学習するために自分が使いたいスクリプトを書いてみました。
記録のために投稿します。

コード

import os
import datetime
import math

#file_unitはファイルサイズ単位を適切に判断するための処理です。
#少数点第二位までを表示させます。
#人間が読みやすいようにファイルサイズによって単位を変更します。
def file_unit(size_bytes):
	units=["B", "KB", "MB", "GB", "TB"]
	i = math.floor(math.log(size_bytes, 1024)) if size_bytes > 0 else 0
	size = round(size_bytes /1024 ** i, 2)
	return f"{size}{units[i]}"


#list_filesはファイルサイズが1GB以上のファイルを出力する処理です。
#file_size >= 1 * 1024 * 1024 * 1024 によってファイルサイズを計算しています。1024を1回掛けるごとにファイルサイズが1KB,1MB,1GBというように変化します。
#表示はファイルフルパス、タイムスタンプ、ファイルサイズを表示します。
def list_files(base_dir):
	file_info_list=[]
	for root, dirs, files in os.walk(base_dir):
		for file_name in files:
			file_path = os.path.join(root, file_name)
			if os.path.isfile(file_path):
				file_size = os.path.getsize(file_path)
				if file_size >= 1 * 1024 * 1024 * 1024:
					mtime = os.path.getmtime(file_path)
					timestamp = datetime.datetime.fromtimestamp(mtime).strftime('%Y/%m/%d')
					file_info_list.append({
						"path": file_path,
						"timestamp": timestamp,
						"size": file_size
					})
	if file_info_list==[]:
		print("file over 1GB was not found")
	for info in file_info_list:
		size_str = file_unit(info["size"])
		print(info["path"])
		print(info["timestamp"])
		print(size_str)
		print()

if __name__ == "__main__":
	TARGET_DIR = "."             #任意のフォルダを指定してください。
	list_files(TARGET_DIR)