mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-22 21:59:59 +00:00
8d1c44039e
## Problem On Debian 12 (Bookworm), Python 3.11 is the latest available version. ## Summary of changes - Update Python to 3.11 in build-tools - Fix ruff check / format - Fix mypy - Use `StrEnum` instead of pair `str`, `Enum` - Update docs
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from enum import StrEnum
|
|
|
|
from typing_extensions import override
|
|
|
|
"""
|
|
This fixture is used to determine which version of Postgres to use for tests.
|
|
"""
|
|
|
|
|
|
# Inherit PgVersion from str rather than int to make it easier to pass as a command-line argument
|
|
class PgVersion(StrEnum):
|
|
V14 = "14"
|
|
V15 = "15"
|
|
V16 = "16"
|
|
V17 = "17"
|
|
|
|
# Postgres Version for tests that uses `fixtures.utils.run_only_on_default_postgres`
|
|
DEFAULT = V17
|
|
|
|
# Instead of making version an optional parameter in methods, we can use this fake entry
|
|
# to explicitly rely on the default server version (could be different from pg_version fixture value)
|
|
NOT_SET = "<-POSTRGRES VERSION IS NOT SET->"
|
|
|
|
# Make it less confusing in logs
|
|
@override
|
|
def __repr__(self) -> str:
|
|
return f"'{self.value}'"
|
|
|
|
@override
|
|
def __str__(self) -> str:
|
|
return self.value
|
|
|
|
# In GitHub workflows we use Postgres version with v-prefix (e.g. v14 instead of just 14),
|
|
# sometime we need to do so in tests.
|
|
@property
|
|
def v_prefixed(self) -> str:
|
|
return f"v{self.value}"
|
|
|
|
@classmethod
|
|
@override
|
|
def _missing_(cls, value: object) -> PgVersion | None:
|
|
if not isinstance(value, str):
|
|
return None
|
|
|
|
known_values = set(cls.__members__.values())
|
|
|
|
# Allow passing version as v-prefixed string (e.g. "v14")
|
|
if value.lower().startswith("v") and (v := value[1:]) in known_values:
|
|
return cls(v)
|
|
|
|
# Allow passing version as an int (i.e. both "15" and "150002" matches PgVersion.V15)
|
|
if value.isdigit() and (v := value[:2]) in known_values:
|
|
return cls(v)
|
|
|
|
return None
|