diff --git a/pomice/models/version.py b/pomice/models/version.py index 6ea8bff..c95d465 100644 --- a/pomice/models/version.py +++ b/pomice/models/version.py @@ -1,4 +1,6 @@ -from typing import NamedTuple +from typing import Literal, NamedTuple, Union + +__all__ = ("LavalinkVersion",) class LavalinkVersion(NamedTuple): @@ -25,3 +27,18 @@ class LavalinkVersion(NamedTuple): if self.fix > other.fix: return False return True + + +class LavalinkVersion3Type(LavalinkVersion): + major: Literal[3] + minor: int + fix: int + + +class LavalinkVersion4Type(LavalinkVersion): + major: Literal[4] + minor: int + fix: int + + +LavalinkVersionType = Union[LavalinkVersion3Type, LavalinkVersion4Type, LavalinkVersion]