github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

call_with_supported_arguments() util, refs #581

Browse source
This commit is contained in:
Simon Willison 2020-05-27 12:25:52 -07:00
commit 41a0cd7b6a
2 changed files with 22 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
datasette/utils/__init__.py
View file

@ -3,6 +3,7 @@ from collections import OrderedDict
import base64 import base64
import click import click
import hashlib import hashlib
import inspect
import json import json
import mergedeep import mergedeep
import os import os
@ -803,3 +804,13 @@ def parse_metadata(content):
return yaml.safe_load(content) return yaml.safe_load(content)
except yaml.YAMLError: except yaml.YAMLError:
raise BadMetadataError("Metadata is not valid JSON or YAML") raise BadMetadataError("Metadata is not valid JSON or YAML")
def call_with_supported_arguments(fn, **kwargs):
parameters = inspect.signature(fn).parameters.keys()
call_with = []
for parameter in parameters:
if parameter not in kwargs:
raise TypeError("{} requires parameters {}".format(fn, tuple(parameters)))
call_with.append(kwargs[parameter])
return fn(*call_with)

11
tests/test_utils.py
View file

@ -446,3 +446,14 @@ async def test_request_post_vars():
request = Request(scope, receive) request = Request(scope, receive)
assert {"foo": "bar", "baz": "1"} == await request.post_vars() assert {"foo": "bar", "baz": "1"} == await request.post_vars()
def test_call_with_supported_arguments():
def foo(a, b):
return "{}+{}".format(a, b)
assert "1+2" == utils.call_with_supported_arguments(foo, a=1, b=2)
assert "1+2" == utils.call_with_supported_arguments(foo, a=1, b=2, c=3)
with pytest.raises(TypeError):
utils.call_with_supported_arguments(foo, a=1)