apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 20:51:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Trying to fix filters

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-03 23:00:06 +01:00
parent b07336156d
commit 7c7f07a92b
1 changed files with 23 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

31
trustgraph-flow/trustgraph/query/objects/cassandra/service.py
View file

@ -280,20 +280,35 @@ class Processor(FlowProcessor):
def create_filter_type_for_schema(self, schema_name: str, row_schema: RowSchema): def create_filter_type_for_schema(self, schema_name: str, row_schema: RowSchema):
"""Create a dynamic filter input type for a schema""" """Create a dynamic filter input type for a schema"""
filter_fields = {} # Create the filter type dynamically
filter_type_name = f"{schema_name.capitalize()}Filter"
# Add __annotations__ and defaults for the fields
annotations = {}
defaults = {}
for field in row_schema.fields: for field in row_schema.fields:
if field.indexed or field.primary: if field.indexed or field.primary:
if field.type == "integer": if field.type == "integer":
filter_fields[field.name] = Optional[IntFilter] annotations[field.name] = Optional[IntFilter]
elif field.type == "float": elif field.type == "float":
filter_fields[field.name] = Optional[FloatFilter] annotations[field.name] = Optional[FloatFilter]
elif field.type == "string": elif field.type == "string":
filter_fields[field.name] = Optional[StringFilter] annotations[field.name] = Optional[StringFilter]
defaults[field.name] = None
# Create the filter type dynamically # Create the class dynamically
filter_type_name = f"{schema_name.capitalize()}Filter" FilterType = type(
FilterType = create_type(filter_type_name, filter_fields, is_input=True) filter_type_name,
(),
{
"__annotations__": annotations,
**defaults
}
)
# Apply strawberry input decorator
FilterType = strawberry.input(FilterType)
return FilterType return FilterType