Skip to content

Commit 7810e26

Browse files
committed
Add environment to the post_transfer arguments
1 parent 0608140 commit 7810e26

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

src/murfey/client/context.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,14 +40,24 @@ def __init__(self, name: str, acquisition_software: str):
4040
self._acquisition_software = acquisition_software
4141
self.name = name
4242

43-
def post_transfer(self, transferred_file: Path, **kwargs):
43+
def post_transfer(
44+
self,
45+
transferred_file: Path,
46+
environment: MurfeyInstanceEnvironment | None = None,
47+
**kwargs,
48+
):
4449
# Search external packages for additional hooks to include in Murfey
4550
for h in entry_points(group="murfey.post_transfer_hooks"):
4651
if h.name == self.name:
47-
h.load()(transferred_file, **kwargs)
52+
h.load()(transferred_file, environment=environment, **kwargs)
4853

49-
def post_first_transfer(self, transferred_file: Path, **kwargs):
50-
self.post_transfer(transferred_file, **kwargs)
54+
def post_first_transfer(
55+
self,
56+
transferred_file: Path,
57+
environment: MurfeyInstanceEnvironment | None = None,
58+
**kwargs,
59+
):
60+
self.post_transfer(transferred_file, environment=environment, **kwargs)
5161

5262
def gather_metadata(
5363
self, metadata_file: Path, environment: MurfeyInstanceEnvironment | None = None

0 commit comments

Comments
 (0)