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
__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}")
__ALL__ = (Device, DeviceTypes)
__all__ = ("Device", "DeviceTypes")

View File

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

View File

@ -437,3 +437,6 @@ class PSSH:
]
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.context: dict[bytes, tuple[bytes, bytes]] = {}
self.keys: list[Key] = []
__all__ = ("Session",)