Fix usage of __all__, add missing __all__ assignments

pywidevine/cdm.py

@@ -638,4 +638,4 @@ def _derive(session_key: bytes, context: bytes, counter: int) -> bytes:
         return enc_key, mac_key_server, mac_key_client
 
 
-__ALL__ = (Cdm,)
+__all__ = ("Cdm",)

pywidevine/device.py

@@ -237,4 +237,4 @@ def migrate(cls, data: Union[bytes, str]) -> Device:
             raise ValueError(f"Device Data seems to be corrupt or invalid, or migration failed, {e}")
 
 
-__ALL__ = (Device, DeviceTypes)
+__all__ = ("Device", "DeviceTypes")

pywidevine/key.py

@@ -61,3 +61,6 @@ def kid_to_uuid(kid: Union[str, bytes]) -> UUID:
             kid += b"\x00" * (16 - len(kid))
 
         return UUID(bytes=kid)
+
+
+__all__ = ("Key",)

pywidevine/pssh.py

@@ -437,3 +437,6 @@ def parse_key_ids(key_ids: list[Union[UUID, str, bytes]]) -> list[UUID]:
         ]
 
         return uuids
+
+
+__all__ = ("PSSH",)

pywidevine/remotecdm.py

@@ -297,4 +297,4 @@ def get_keys(self, session_id: bytes, type_: Optional[Union[int, str]] = None) -
         ]
 
 
-__ALL__ = (RemoteCdm,)
+__all__ = ("RemoteCdm",)

pywidevine/session.py

@@ -13,3 +13,6 @@ def __init__(self, number: int):
         self.service_certificate: Optional[SignedDrmCertificate] = None
         self.context: dict[bytes, tuple[bytes, bytes]] = {}
         self.keys: list[Key] = []
+
+
+__all__ = ("Session",)