feat: add File.copy()

mcbeet · Oct 1, 2022 · d1f5b36 · d1f5b36
1 parent f7fd2d5
commit d1f5b36
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/beet/core/file.py b/beet/core/file.py
@@ -24,6 +24,7 @@
 import io
 import json
 import shutil
+from copy import deepcopy
 from dataclasses import dataclass, replace
 from pathlib import Path
 from typing import Any, Callable, ClassVar, Generic, Optional, Type, TypeVar, Union
@@ -195,6 +196,10 @@ def default(cls) -> ValueType:
             "either a value, serialized data, or a source path."
         )
 
+    def copy(self: FileType) -> FileType:
+        """Copy the file."""
+        return replace(self, _content=deepcopy(self._content))
+
     def serialize(self, content: Union[ValueType, SerializeType]) -> SerializeType:
         """Serialize file content."""
         raise NotImplementedError()

diff --git a/tests/test_file.py b/tests/test_file.py
@@ -93,3 +93,8 @@ def test_model(data: str):
     ab = ABFile(data)
     ab.ensure_deserialized()
     assert json.loads(data) == json.loads(ab.text)
+
+
+def test_copy():
+    assert TextFile(source_path="foo.txt").copy().source_path == "foo.txt"
+    assert TextFile("hello").copy().text == "hello"