Support pydantic v1 and v2

.github/workflows/test.yaml

@@ -10,8 +10,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ['3.7', '3.10']
-        galaxy-branch: ['release_22.01', 'dev']
+        python-version: ['3.8', '3.11']
+        galaxy-branch: ['release_23.0', 'dev']
         exclude:
           # this results in lengthy and expensive numpy wheel builds
           - python-version: '3.10'

gravity/config_manager.py

@@ -7,15 +7,15 @@
 import xml.etree.ElementTree as elementtree
 from typing import Union
 
-from pydantic import ValidationError
+try:
+    from pydantic.v1 import ValidationError
+except ImportError:
+    from pydantic import ValidationError
 from yaml import safe_load
 
 import gravity.io
 from gravity.settings import Settings
-from gravity.state import (
-    ConfigFile,
-    service_for_service_type,
-)
+from gravity.state import ConfigFile, service_for_service_type
 from gravity.util import recursive_update
 
 log = logging.getLogger(__name__)

gravity/settings.py

@@ -1,13 +1,11 @@
 import os
 from enum import Enum
-from typing import (
-    Any,
-    Dict,
-    List,
-    Optional,
-    Union,
-)
-from pydantic import BaseModel, BaseSettings, Extra, Field, validator
+from typing import Any, Dict, List, Optional, Union
+
+try:
+    from pydantic.v1 import BaseModel, BaseSettings, Extra, Field, validator
+except ImportError:
+    from pydantic import BaseModel, BaseSettings, Extra, Field, validator
 
 DEFAULT_INSTANCE_NAME = "_default_"
 GX_IT_PROXY_MIN_VERSION = "0.0.6"

gravity/state.py

@@ -2,24 +2,23 @@
 state data.
 """
 from __future__ import annotations
+
 import enum
 import hashlib
 import os
 import sys
 import time
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel, validator
+try:
+    from pydantic.v1 import BaseModel, validator
+except ImportError:
+    from pydantic import BaseModel, validator
 
 import gravity.io
-from gravity.settings import (
-    AppServer,
-    ProcessManager,
-    ServiceCommandStyle,
-)
+from gravity.settings import AppServer, ProcessManager, ServiceCommandStyle
 from gravity.util import http_check
 
-
 DEFAULT_GALAXY_ENVIRONMENT = {
     "PYTHONPATH": "lib",
     "GALAXY_CONFIG_FILE": "{galaxy_conf}",

setup.py

@@ -4,8 +4,7 @@
 import os
 import re
 
-from setuptools import setup, find_packages
-
+from setuptools import find_packages, setup
 
 with open("README.rst") as file:
     long_description = file.read()
@@ -36,13 +35,13 @@ def get_var(var_name):
     author_email="[email protected]",
     license="MIT",
     keywords="gravity galaxy",
-    python_requires=">=3.6",
+    python_requires=">=3.7",
     install_requires=[
         "Click",
         "supervisor",
         "pyyaml",
         "packaging",
-        "pydantic<2",
+        "pydantic<3",  # pydantic.v1 import will be removed in v3
         "jsonref",
         "requests",
         "requests-unixsocket",
@@ -57,10 +56,10 @@ def get_var(var_name):
         "Natural Language :: English",
         "Operating System :: POSIX",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.6",
-        "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
     ],
     zip_safe=False,
 )