はじめに

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

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