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: ...@@ -11,18 +11,14 @@ class File:
self.new_dir = new_dir if new_dir is not None else self.dirname self.new_dir = new_dir if new_dir is not None else self.dirname
def replace_ext( 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: ) -> str:
if new_dir is None: if new_dir is None:
new_dir = self.new_dir new_dir = self.new_dir
if new_ext is None:
new_ext = self.ext
if new_ext.startswith("."): if new_ext.startswith("."):
return os.path.join( return os.path.join(new_dir, self.prefix + new_ext)
new_dir if new_dir is not None else self.dirname,
self.prefix + new_ext,
)
else: else:
return os.path.join( return os.path.join(new_dir, self.prefix + "_" + new_ext)
new_dir if new_dir is not None else self.dirname,
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