replaced pydantic.BaseModel by dataclasses.dataclass, pydantic no longer required

Author: ds-jakub-cierocki

src/dbally/context/context.py

@@ -1,16 +1,16 @@
 import ast
+from dataclasses import dataclass
 from typing import Iterable
 
-from pydantic import BaseModel
 from typing_extensions import Self, TypeAlias
 
 from dbally.context.exceptions import ContextNotAvailableError
 
-# CustomContext = TypeVar('CustomContext', bound='BaseCallerContext', covariant=True)
 CustomContext: TypeAlias = "BaseCallerContext"
 
 
-class BaseCallerContext(BaseModel):
+@dataclass
+class BaseCallerContext:
     """
     Pydantic-based record class. Base class for contexts that are used to pass additional knowledge about
     the caller environment to the filters. It is not made abstract for the convinience of IQL parsing.

tests/unit/iql/test_iql_parser.py

@@ -1,4 +1,5 @@
 import re
+from dataclasses import dataclass
 from typing import List, Union
 
 import pytest
@@ -10,10 +11,12 @@
 from dbally.views.exposed_functions import ExposedFunction, MethodParamWithTyping
 
 
+@dataclass
 class TestCustomContext(BaseCallerContext):
     city: str
 
 
+@dataclass
 class AnotherTestCustomContext(BaseCallerContext):
     some_field: str
 

tests/unit/views/test_sqlalchemy_base.py

@@ -1,6 +1,7 @@
 # pylint: disable=missing-docstring, missing-return-doc, missing-param-doc, disallowed-name
 
 import re
+from dataclasses import dataclass
 from typing import Union
 
 import sqlalchemy
@@ -11,6 +12,7 @@
 from dbally.views.sqlalchemy_base import SqlAlchemyBaseView
 
 
+@dataclass
 class SomeTestContext(BaseCallerContext):
     age: int