Skip to content

Commit

Permalink
Merge pull request #821 from martindurant/function_serialize
Browse files Browse the repository at this point in the history
Raise when trying to store a dynamic function
  • Loading branch information
martindurant committed Jun 5, 2024
2 parents b04e271 + 4dd083b commit f5b43f6
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 1 deletion.
18 changes: 18 additions & 0 deletions intake/readers/tests/test_errors.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
import pytest

import intake


def test_func_ser():
class A:
def get(self):
def inner():
return self.x

return inner

func = A().get()
reader = intake.readers.convert.GenericFunc(data=func)
cat = intake.Catalog()
with pytest.raises(RuntimeError):
cat.add_entry(reader)
5 changes: 4 additions & 1 deletion intake/readers/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,10 @@ def find_funcs(val, tokens={}):
if isinstance(val, Tokenizable):
return val.to_dict()
elif callable(val):
return "{func(%s)}" % f"{val.__module__}:{val.__name__}"
name = "{func(%s)}" % f"{val.__module__}:{val.__name__}"
if "<locals>" in name or "__main__" in name or getattr(val, "__closure__", False):
raise RuntimeError("Cannot store dynamically defined function: %s", val)
return name
elif val is None or isinstance(val, (numbers.Number, BaseData, BaseReader)):
return val
else:
Expand Down

0 comments on commit f5b43f6

Please sign in to comment.