Improve typing of database records

Author: oschwald

maxminddb/reader.py

@@ -23,22 +23,7 @@
 from maxminddb.decoder import Decoder
 from maxminddb.errors import InvalidDatabaseError
 from maxminddb.file import FileBuffer
-
-Record = Union[
-    str,
-    Dict[str, str],
-    Dict[
-        str,
-        Union[
-            List[int],
-            bytes,
-            float,
-            int,
-            Dict[str, Dict[str, Union[List[int], str]]],
-            str,
-        ],
-    ],
-]
+from maxminddb.types import Record
 
 
 class Reader:

maxminddb/types.py

@@ -0,0 +1,22 @@
+"""
+maxminddb.types
+~~~~~~~~~~~~~~~
+
+This module provides a Record type the represents a database record.
+"""
+from typing import AnyStr, Dict, List, Union
+
+Primitive = Union[AnyStr, bool, float, int]
+Record = Union[Primitive, "RecordList", "RecordDict"]
+
+
+class RecordList(List[Record]):  # pylint: disable=too-few-public-methods
+    """
+    RecordList is a type for lists in a database record.
+    """
+
+
+class RecordDict(Dict[str, Record]):  # pylint: disable=too-few-public-methods
+    """
+    RecordDict is a type for dicts in a database record.
+    """