chore: drop pydantic v1 support

.github/workflows/ci.yml

@@ -10,9 +10,6 @@ jobs:
         python-version:
           - "3.10"
           - "3.11"
-        pydantic-version:
-          - "^1.10"
-          - "^2.0"
 
     steps:
       - uses: actions/checkout@v4
@@ -24,15 +21,6 @@ jobs:
       - name: install_system_deps
         run: sudo make install_system_deps
 
-      - name: Update Pydantic version on the toml & lock
-        run: |
-          pip install -U pip poetry
-          # we update the pydantic version corresponding to the matrix
-          sed -i "s/pydantic = \".*\"/pydantic = \"${PYDANTIC_VERSION}\"/" pyproject.toml
-          poetry lock --no-update
-        env:
-          PYDANTIC_VERSION: ${{ matrix.pydantic-version }}
-
       - name: install
         run: make install
 
@@ -49,9 +37,6 @@ jobs:
         python-version:
           - "3.10"
           - "3.11"
-        pydantic-version:
-          - "^1.10"
-          - "^2.0"
 
     steps:
     - uses: actions/checkout@v4
@@ -63,23 +48,13 @@ jobs:
     - name: install_system_deps
       run: sudo make install_system_deps
 
-    - name: Update Pydantic version on the toml & lock
-      run: |
-        pip install -U pip poetry
-        # we update the pydantic version corresponding to the matrix
-        sed -i "s/pydantic = \".*\"/pydantic = \"${PYDANTIC_VERSION}\"/" pyproject.toml
-        poetry lock --no-update
-      env:
-        PYDANTIC_VERSION: ${{ matrix.pydantic-version }}
-
     - name: install
       run: make install
 
     - name: test
       run: make test
       env:
         FTP_PATH: ${{ secrets.FTP_PATH }}
-        PYDANTIC_VERSION: ${{ matrix.pydantic-version }}
 
     - name: Upload coverage
       uses: codecov/codecov-action@v3

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+### Changed
+
+- Drop support for pydantic V1.
+
 ## [0.12.1] - 2023-11-15
 
 ### Fixed

peakina/datasource.py

@@ -13,7 +13,7 @@
 from urllib.parse import urlparse, uses_netloc, uses_params, uses_relative
 
 import pandas as pd
-from pydantic import ConfigDict, __version__ as pydantic_version
+from pydantic import ConfigDict
 from pydantic.dataclasses import dataclass
 from slugify import slugify
 
@@ -36,9 +36,6 @@
 PD_VALID_URLS = set(uses_relative + uses_netloc + uses_params) | AVAILABLE_SCHEMES
 
 
-_PYDANTIC_VERSION_ONE = pydantic_version.startswith("1.")
-
-
 @dataclass
 class DataSource:
     uri: str
@@ -48,33 +45,17 @@ class DataSource:
     reader_kwargs: dict[str, Any] = field(default_factory=dict)
     fetcher_kwargs: dict[str, Any] = field(default_factory=dict)
 
-    # TODO: This is temporary, in the future we will only support V2
-    # and get rid of this condition + update the CI (link/test)
-    if _PYDANTIC_VERSION_ONE:
-
-        def __post_init_post_parse__(self) -> None:
-            self._fetcher: Fetcher | None = None
-            self.scheme = urlparse(self.uri).scheme
-            if self.scheme not in PD_VALID_URLS:
-                raise AttributeError(f"Invalid scheme {self.scheme!r}")
-
-            self.type = self.type or detect_type(urlparse(self.uri).path, is_regex=bool(self.match))
-
-            validate_kwargs(self.reader_kwargs, self.type)
-
-    else:
-
-        def __post_init__(self) -> None:
-            self._fetcher: Fetcher | None = None  # type: ignore[no-redef]
-            self.scheme = urlparse(self.uri).scheme
-            if self.scheme not in PD_VALID_URLS:
-                raise AttributeError(f"Invalid scheme {self.scheme!r}")
+    def __post_init__(self) -> None:
+        self._fetcher: Fetcher | None = None
+        self.scheme = urlparse(self.uri).scheme
+        if self.scheme not in PD_VALID_URLS:
+            raise AttributeError(f"Invalid scheme {self.scheme!r}")
 
-            self.type = self.type or detect_type(urlparse(self.uri).path, is_regex=bool(self.match))
+        self.type = self.type or detect_type(urlparse(self.uri).path, is_regex=bool(self.match))
 
-            validate_kwargs(self.reader_kwargs, self.type)
+        validate_kwargs(self.reader_kwargs, self.type)
 
-        model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
     @property
     def fetcher(self) -> Fetcher:

pyproject.toml

@@ -23,7 +23,7 @@ chardet = ">=4,<6"
 jq = "^1.2.1"
 pandas = ">=1.5.3,<3.0.0"
 paramiko = ">=2.9.2,<4.0.0"
-pydantic = ">=1.9,<3.0"
+pydantic = "^2.4.2"
 python-slugify = ">=5.0.2,<9.0.0"
 s3fs = ">=2022.1,<2024.0"
 tables = "^3.7.0"