Commit 5b03f95c authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 3a8729c4
Loading
Loading
Loading
Loading
+5 −9
Original line number Diff line number Diff line
@@ -11,18 +11,14 @@ class File:
        self.new_dir = new_dir if new_dir is not None else self.dirname

    def replace_ext(
        self, new_ext: str = "wht.fits", new_dir: Optional[str] = None
        self, new_ext: Optional[str] = "img.fits", new_dir: Optional[str] = None
    ) -> str:
        if new_dir is None:
            new_dir = self.new_dir
        if new_ext is None:
            new_ext = self.ext

        if new_ext.startswith("."):
            return os.path.join(
                new_dir if new_dir is not None else self.dirname,
                self.prefix + new_ext,
            )
            return os.path.join(new_dir, self.prefix + new_ext)
        else:
            return os.path.join(
                new_dir if new_dir is not None else self.dirname,
                self.prefix + "_" + new_ext,
            )
            return os.path.join(new_dir, self.prefix + "_" + new_ext)