From 239ba599b6b10128c376e8f707895a6be9ba4d2f Mon Sep 17 00:00:00 2001 From: yim7 Date: Mon, 26 Sep 2022 22:15:59 +0800 Subject: [PATCH] feat: generic type --- opensearchorm/session.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/opensearchorm/session.py b/opensearchorm/session.py index 09a8126..a416ca4 100644 --- a/opensearchorm/session.py +++ b/opensearchorm/session.py @@ -1,6 +1,6 @@ import json import logging -from typing import List, Union, Optional, Type, TypeVar, cast +from typing import Generic, List, Union, Optional, Type, TypeVar, cast from opensearchpy import OpenSearch @@ -51,7 +51,7 @@ class SearchSession: return self.client.count(**kwargs) -class QueryExecutor: +class QueryExecutor(Generic[Model]): def __init__(self, model_cls: Type[Model], session: SearchSession): self.__query = ModelQuery(model_cls) self.__model_cls = model_cls