add write method to CustardList

Author: ensan-hcl

python/source/custard.py

@@ -10,7 +10,8 @@ class CustardJSONEncoder(json.JSONEncoder):
     def default(self, o):
         if isinstance(o, Custard):
             return o.json()
-
+        if isinstance(o, CustardList):
+            return to_json_list(o.custards)
         # 他の型はdefaultのエンコード方式を使用
         return super(CustardJSONEncoder, self).default(o)
 
@@ -79,8 +80,26 @@ def json(self) -> dict:
             "interface": self.interface.json()
         }
 
+    def write(self, to: str = None, name: str = None, allow_overwrite: bool = False):
+        """
+        Custardファイルを出力する関数
+        Parameters
+        ----------
+        to: str = None
+            出力先のパスを指定
+        name: str = None
+            出力先のパスを指定しない場合にファイル名を指定
+        allow_overwrite: bool = False
+            上書きを許可するか否か
+        """
+        pass
+
 
-    def write(self, to: str = None, name: str = None, allow_overwrite: bool = False) -> dict:
+class CustardList(object):
+    def __init__(self, custards: list[Custard]):
+        self.custards = custards
+
+    def write(self, to: str = None, name: str = None, allow_overwrite: bool = False):
         """
         Custardファイルを出力する関数
         Parameters
@@ -92,18 +111,26 @@ def write(self, to: str = None, name: str = None, allow_overwrite: bool = False)
         allow_overwrite: bool = False
             上書きを許可するか否か
         """
-        if to is None:
-            result_directory_path = Path('__file__').resolve().parent / 'results'
-            if not result_directory_path.exists():
-                result_directory_path.mkdir()
-            if name is None:
-                name = 'custard'
-            target = result_directory_path / f'{name}.json'
-            number = 1
-            while target.exists() and not allow_overwrite:
-                number += 1
-                target = result_directory_path / f'{name}#{number}.json'
-            to = str(target)
-
-        with open(f"{to}", mode="w") as f:
-            f.write(json.dumps(self, cls=CustardJSONEncoder, ensure_ascii=False))
+        pass
+
+
+def write(self, to: str = None, name: str = None, allow_overwrite: bool = False):
+    if to is None:
+        result_directory_path = Path('__file__').resolve().parent / 'results'
+        if not result_directory_path.exists():
+            result_directory_path.mkdir()
+        if name is None:
+            name = 'custard'
+        target = result_directory_path / f'{name}.json'
+        number = 1
+        while target.exists() and not allow_overwrite:
+            number += 1
+            target = result_directory_path / f'{name}#{number}.json'
+        to = str(target)
+
+    with open(f"{to}", mode="w") as f:
+        f.write(json.dumps(self, cls=CustardJSONEncoder, ensure_ascii=False))
+
+
+Custard.write = write
+CustardList.write = write