refactor filename

Author: almogdepaz

chia/plotting/prover.py

@@ -159,7 +159,7 @@ def disk_prover(self) -> DiskProver:
 
 
 def get_prover_from_bytes(filename: str, prover_data: bytes) -> ProverProtocol:
-    if filename.endswith(".plot_v2"):
+    if filename.endswith(".plot2"):
         return V2Prover.from_bytes(prover_data)
     elif filename.endswith(".plot"):
         return V1Prover(DiskProver.from_bytes(prover_data))
@@ -168,7 +168,7 @@ def get_prover_from_bytes(filename: str, prover_data: bytes) -> ProverProtocol:
 
 
 def get_prover_from_file(filename: str) -> ProverProtocol:
-    if filename.endswith(".plot_v2"):
+    if filename.endswith(".plot2"):
         return V2Prover(filename)
     elif filename.endswith(".plot"):
         return V1Prover(DiskProver(filename))

chia/plotting/util.py

@@ -236,7 +236,7 @@ def get_filenames(directory: Path, recursive: bool, follow_links: bool) -> list[
             import glob
 
             v1_file_strs = glob.glob(str(directory / "**" / "*.plot"), recursive=True)
-            v2_file_strs = glob.glob(str(directory / "**" / "*.plot_v2"), recursive=True)
+            v2_file_strs = glob.glob(str(directory / "**" / "*.plot2"), recursive=True)
 
             for file in v1_file_strs + v2_file_strs:
                 filepath = Path(file).resolve()
@@ -248,7 +248,7 @@ def get_filenames(directory: Path, recursive: bool, follow_links: bool) -> list[
                 child for child in glob_function("*.plot") if child.is_file() and not child.name.startswith("._")
             ]
             v2_files: list[Path] = [
-                child for child in glob_function("*.plot_v2") if child.is_file() and not child.name.startswith("._")
+                child for child in glob_function("*.plot2") if child.is_file() and not child.name.startswith("._")
             ]
             all_files = v1_files + v2_files
         log.debug(f"get_filenames: {len(all_files)} files found in {directory}, recursive: {recursive}")