[Python]ファイル関連の処理でよく使うスニペット（個人用）

はじめに

最近作業をしていて、自分がよく使う処理があるのですが、いちいちネットで調べるのが面倒なので、スニペット集として残しておきます。

• JSONファイルを読み込む
• JSONファイルを保存する
• 所定のディレクトリに格納されているファイル名一覧を取得したい
• ディレクトリを作成する

コード

import os
import json

def mkdir(dir_path: str):
  if (not os.path.isdir(dir_path)):
    os.mkdir(dir_path)

def load_json(json_path: str):
  json_data = None
  with open(f'{json_path}', 'r') as f:
    json_data = json.load(f)
  return json_data

def save_json(json_path: str, data):
  with open(f'{json_path}', 'w') as f:
    json.dump(data, f, ensure_ascii=False, indent=4)

def get_file_list(dir_path: str, extension=''):
  file_list = []
  for file_name in os.listdir(dir_path):
    if (extension in file_name):
      file_list.append(file_name)
      # フルパスにしたいときにはこっち
      # return_list.append(os.path.join(dir_name, file_name))
  return file_list