Add support for pydantic v2 using v1 shims (#504)

kasa/modules/cloud.py

@@ -1,5 +1,8 @@
 """Cloud module implementation."""
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 from .module import Module
 

kasa/modules/rulemodule.py

@@ -3,7 +3,11 @@ import logging
 from enum import Enum
 from typing import Dict, List, Optional
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
+
 
 from .module import Module, merge
 

kasa/smartbulb.py

@@ -4,7 +4,10 @@ import re
 from enum import Enum
 from typing import Any, Dict, List, NamedTuple, Optional, cast
 
-from pydantic import BaseModel, Field, root_validator
+try:
+    from pydantic.v1 import BaseModel, Field, root_validator
+except ImportError:
+    from pydantic import BaseModel, Field, root_validator
 
 from .modules import Antitheft, Cloud, Countdown, Emeter, Schedule, Time, Usage
 from .smartdevice import DeviceType, SmartDevice, SmartDeviceException, requires_update

pyproject.toml

@@ -22,7 +22,7 @@ kasa = "kasa.cli:cli"
 python = "^3.8"
 anyio = "*"  # see https://github.com/python-trio/asyncclick/issues/18
 asyncclick = ">=8"
-pydantic = "^1"
+pydantic = ">=1"
 
 # speed ups
 orjson = { "version" = ">=3.9.1", optional = true }