Skip to content

Commit f2b0d1a

Browse files
committed
fix: adapt to oold api changes
1 parent f3d445e commit f2b0d1a

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

src/osw/core.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -112,12 +112,12 @@ def __init__(self, **data: Any):
112112
super().__init__(**data)
113113

114114
# implement resolver backend with osw.load_entity
115-
class MyResolver(Resolver):
115+
class OswDefaultResolver(Resolver):
116116

117117
osw_obj: OSW
118118

119119
def resolve(self, request: ResolveParam):
120-
print("RESOLVE", request)
120+
# print("RESOLVE", request)
121121
entities = self.osw_obj.load_entity(
122122
OSW.LoadEntityParam(titles=request.iris)
123123
).entities
@@ -128,9 +128,10 @@ def resolve(self, request: ResolveParam):
128128
nodes[iri] = entity
129129
return ResolveResult(nodes=nodes)
130130

131-
r = MyResolver(osw_obj=self)
131+
r = OswDefaultResolver(osw_obj=self)
132132
set_resolver(SetResolverParam(iri="Item", resolver=r))
133133
set_resolver(SetResolverParam(iri="Category", resolver=r))
134+
set_resolver(SetResolverParam(iri="Property", resolver=r))
134135

135136
@property
136137
def mw_site(self) -> Site:
@@ -507,7 +508,9 @@ def _fetch_schema(self, fetchSchemaParam: _FetchSchemaParam = None) -> None:
507508

508509
generator = Generator()
509510
schemas_for_preprocessing = [json.loads(schema_str)]
510-
generator.preprocess(schemas_for_preprocessing)
511+
generator.preprocess(
512+
Generator.GenerateParams(json_schemas=schemas_for_preprocessing)
513+
)
511514
schema_str = json.dumps(schemas_for_preprocessing[0])
512515

513516
schema = json.loads(

0 commit comments

Comments
 (0)