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

tweaks

parent 3a8729c4
Pipeline #2309 failed with stage
in 0 seconds
......@@ -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)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment