Skip to content

Commit

Permalink
refactor: docstringの記述
Browse files Browse the repository at this point in the history
  • Loading branch information
keiya121 committed Sep 8, 2024
1 parent 025ee87 commit 87a4343
Showing 1 changed file with 28 additions and 10 deletions.
38 changes: 28 additions & 10 deletions src/csv_to_json.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,11 @@ class CSVToJSONConverter:
"""CSVファイルをJSONファイルに変換するクラス."""

def __init__(self, csv_file_path: str) -> None:
"""コンストラクタ."""
"""コンストラクタ.
Args:
csv_file_path (str): CSVファイルのパス
"""
self.csv_file_path = csv_file_path
self.json_file_path = self._get_json_file_path()

Expand All @@ -24,18 +28,26 @@ def convert(self) -> None:
self._write_json(data)

def _read_csv(self) -> List[dict]:
"""CSVファイルを読み込み、辞書のリストを返す."""
data = []
"""CSVファイルを読み込み、辞書のリストを返す.
Return:
run_log_data (List[dict]): 辞書形式の走行ログデータ
"""
run_log_data = []
with open(self.csv_file_path, mode='r', encoding='utf-8') as csv_file:
reader = csv.DictReader(csv_file, fieldnames=[
'brightness', 'rightPWM', 'leftPWM', 'R', 'G', 'B'])
for row in reader:
data.append(row)
return data
run_log_data.append(row)
return run_log_data

def _write_json(self, data: List[dict]) -> None:
"""データをJSONファイルに書き込む."""
json_data = {'runLog': data}
def _write_json(self, run_log_data: List[dict]) -> None:
"""データをJSONファイルに書き込む.
Args:
run_log_data (List[dict]): 辞書形式の走行ログデータ
"""
json_data = {'runLog': run_log_data}

# JSONファイルの保存先フォルダーを確認し、存在しない場合は作成
os.makedirs(os.path.dirname(self.json_file_path), exist_ok=True)
Expand All @@ -45,6 +57,12 @@ def _write_json(self, data: List[dict]) -> None:
json.dump(json_data, json_file, ensure_ascii=False, indent=4)

def _get_json_file_path(self) -> str:
"""JSONファイルのパスを作成する."""
"""JSONファイルのパスを作成する.
Return:
json_filepath (str): jsonファイルのパス
"""
base, _ = os.path.splitext(os.path.basename(self.csv_file_path))
return os.path.join('src', 'server', 'run_log_json', base + '.json')
json_filepath = os.path.join(
'src', 'server', 'run_log_json', base + '.json')
return json_filepath

0 comments on commit 87a4343

Please sign in to comment.