Fix usage of __all__, add missing __all__ assignments

This commit is contained in:
rlaphoenix 2023-11-08 22:56:37 +00:00
parent 5087da31a0
commit 49315eceb8
6 changed files with 12 additions and 3 deletions

View File

@ -638,4 +638,4 @@ class Cdm:
return enc_key, mac_key_server, mac_key_client return enc_key, mac_key_server, mac_key_client
__ALL__ = (Cdm,) __all__ = ("Cdm",)

View File

@ -237,4 +237,4 @@ class Device:
raise ValueError(f"Device Data seems to be corrupt or invalid, or migration failed, {e}") raise ValueError(f"Device Data seems to be corrupt or invalid, or migration failed, {e}")
__ALL__ = (Device, DeviceTypes) __all__ = ("Device", "DeviceTypes")

View File

@ -61,3 +61,6 @@ class Key:
kid += b"\x00" * (16 - len(kid)) kid += b"\x00" * (16 - len(kid))
return UUID(bytes=kid) return UUID(bytes=kid)
__all__ = ("Key",)

View File

@ -437,3 +437,6 @@ class PSSH:
] ]
return uuids return uuids
__all__ = ("PSSH",)

View File

@ -297,4 +297,4 @@ class RemoteCdm(Cdm):
] ]
__ALL__ = (RemoteCdm,) __all__ = ("RemoteCdm",)

View File

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