update File

csst_common/file.py

 import os
+from typing import Optional
 
 
 class File:
-    def __init__(self, file_path: str = "/path/to/file.fits"):
+    def __init__(self, file_path: str = "/path/to/file.fits", new_dir=None):
         self.file_path = file_path
-        self.file_path_prefix, self.file_path_ext = os.path.splitext(file_path)
+        self.dirname = os.path.dirname(self.file_path)
+        self.file_name = os.path.basename(self.file_path)
+        self.prefix, self.ext = os.path.splitext(self.file_name)
+        self.new_dir = new_dir if new_dir is not None else self.dirname
 
-    def replace_ext(self, ext="wht.fits") -> str:
-        if ext.startswith("."):
-            return self.file_path_prefix + ext
+    def replace_ext(
+        self, new_ext: str = "wht.fits", new_dir: Optional[str] = None
+    ) -> str:
+        if new_dir is None:
+            new_dir = self.new_dir
+
+        if new_ext.startswith("."):
+            return os.path.join(
+                new_dir if new_dir is not None else self.dirname,
+                self.prefix + new_ext,
+            )
         else:
-            return self.file_path_prefix + "_" + ext
+            return os.path.join(
+                new_dir if new_dir is not None else self.dirname,
+                self.prefix + "_" + new_ext,
+            )