35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
|
import json
|
||
|
|
|
||
|
|
from requests import Request
|
||
|
|
|
||
|
|
from vinetrimmer.vendor.BamSDK.services import Service
|
||
|
|
|
||
|
|
|
||
|
|
# noinspection PyPep8Naming
|
||
|
|
class drm(Service):
|
||
|
|
def widevineCertificate(self):
|
||
|
|
endpoint = self.client.endpoints["widevineCertificate"]
|
||
|
|
req = Request(
|
||
|
|
method=endpoint.method,
|
||
|
|
url=endpoint.href,
|
||
|
|
headers=endpoint.headers
|
||
|
|
).prepare()
|
||
|
|
res = self.session.send(req)
|
||
|
|
return res.content
|
||
|
|
|
||
|
|
def widevineLicense(self, licence, access_token):
|
||
|
|
endpoint = self.client.endpoints["widevineLicense"]
|
||
|
|
req = Request(
|
||
|
|
method=endpoint.method,
|
||
|
|
url=endpoint.href,
|
||
|
|
headers=endpoint.get_headers(accessToken=access_token),
|
||
|
|
data=licence
|
||
|
|
).prepare()
|
||
|
|
res = self.session.send(req)
|
||
|
|
try:
|
||
|
|
# if it's json content, then an error occurred
|
||
|
|
res = json.loads(res.text)
|
||
|
|
raise Exception(f"Failed to obtain license: {res}")
|
||
|
|
except json.JSONDecodeError:
|
||
|
|
return res.content
|