Move PowerState dataclass to typing.py

Author: ankith26

buildconfig/stubs/pygame/system.pyi

@@ -1,6 +1,6 @@
 from typing import List, Optional, TypedDict
 
-from pygame._data_classes import PowerState
+from pygame.typing import PowerState
 
 class _InstructionSets(TypedDict):
     ALTIVEC: bool

buildconfig/stubs/pygame/typing.pyi

@@ -14,7 +14,9 @@ __all__ = [
 
 import sys
 from abc import abstractmethod
-from typing import IO, Callable, Tuple, Union, TypeVar, Protocol
+from dataclasses import dataclass
+from typing import IO, Callable, Tuple, Union, TypeVar, Protocol, Optional
+
 
 if sys.version_info >= (3, 9):
     from os import PathLike as _PathProtocol
@@ -66,5 +68,28 @@ class _HasRectAttribute(Protocol):
 RectLike = Union[SequenceLike[float], SequenceLike[Coordinate], _HasRectAttribute]
 
 
+@dataclass(frozen=True)
+class PowerState:
+    battery_percent: Optional[int]
+    battery_seconds: Optional[int]
+    on_battery: bool
+    no_battery: bool
+    charging: bool
+    charged: bool
+    plugged_in: bool
+    has_battery: bool
+
+
 # cleanup namespace
-del sys, abstractmethod, IO, Callable, Tuple, Union, TypeVar, Protocol
+del (
+    sys,
+    abstractmethod,
+    IO,
+    Callable,
+    Tuple,
+    Union,
+    TypeVar,
+    Protocol,
+    Optional,
+    dataclass,
+)

src_c/system.c

@@ -261,8 +261,7 @@ MODINIT_DEFINE(system)
         return NULL;
     }
 
-    PyObject *data_classes_module =
-        PyImport_ImportModule("pygame._data_classes");
+    PyObject *data_classes_module = PyImport_ImportModule("pygame.typing");
     if (!data_classes_module) {
         return NULL;
     }

src_py/__init__.py

@@ -317,7 +317,7 @@ def PixelArray(surface):  # pylint: disable=unused-argument
 
 try:
     import pygame.system
-    from pygame._data_classes import PowerState as power_state
+    from pygame.typing import PowerState as power_state
 
     power_state.__module__ = "pygame.system"
     del power_state

src_py/_data_classes.py

@@ -2,7 +2,6 @@
 python_sources = files(
     '__init__.py',
     '_camera_opencv.py',
-    '_data_classes.py',
     '_debug.py',
     '_sprite.py',
     'camera.py',

src_py/meson.build

@@ -14,7 +14,9 @@
 
 import sys
 from abc import abstractmethod
-from typing import IO, Callable, Tuple, Union, TypeVar, Protocol
+from dataclasses import dataclass
+from typing import IO, Callable, Tuple, Union, TypeVar, Protocol, Optional
+
 
 if sys.version_info >= (3, 9):
     from os import PathLike as _PathProtocol
@@ -66,5 +68,28 @@ def rect(self) -> Union["RectLike", Callable[[], "RectLike"]]: ...
 RectLike = Union[SequenceLike[float], SequenceLike[Coordinate], _HasRectAttribute]
 
 
+@dataclass(frozen=True)
+class PowerState:
+    battery_percent: Optional[int]
+    battery_seconds: Optional[int]
+    on_battery: bool
+    no_battery: bool
+    charging: bool
+    charged: bool
+    plugged_in: bool
+    has_battery: bool
+
+
 # cleanup namespace
-del sys, abstractmethod, IO, Callable, Tuple, Union, TypeVar, Protocol
+del (
+    sys,
+    abstractmethod,
+    IO,
+    Callable,
+    Tuple,
+    Union,
+    TypeVar,
+    Protocol,
+    Optional,
+    dataclass,
+)