Fix default DRM type
This commit is contained in:
parent
1b2ed5e31f
commit
66d8d1006e
3
.gitignore
vendored
3
.gitignore
vendored
@ -171,4 +171,5 @@ poetry.toml
|
||||
pyrightconfig.json
|
||||
|
||||
### DVDFab ###
|
||||
*.htkrules
|
||||
*.htkrules
|
||||
exports.json
|
@ -44,6 +44,7 @@ def parse_boundary(data: str) -> dict:
|
||||
class ServiceType(Enum):
|
||||
STREAM = 'Stream' # StreamFab
|
||||
MUSIC = 'Music' # MusicFab
|
||||
DEFAULT = 'Default'
|
||||
|
||||
|
||||
class Session:
|
||||
@ -70,7 +71,7 @@ class Session:
|
||||
def __init__(self):
|
||||
self._email = None
|
||||
self._mac = None
|
||||
self.type = ServiceType.STREAM
|
||||
self.type = ServiceType.DEFAULT
|
||||
self.max = 3
|
||||
self.export = True
|
||||
self.running = {}
|
||||
@ -150,7 +151,8 @@ class Session:
|
||||
# Random token for certain fields
|
||||
data[key] = secrets.token_hex(16)
|
||||
elif key in ('PS'):
|
||||
data[key] = self.type.value
|
||||
if self.type != ServiceType.DEFAULT:
|
||||
data[key] = self.type.value
|
||||
|
||||
# Generate an invalid KID
|
||||
data['KD'] = secrets.token_hex(16)
|
||||
@ -273,7 +275,7 @@ if __name__ == '__main__':
|
||||
# Command-line arguments
|
||||
parser.add_argument('--email', type=str, metavar='<email>', help='Email address (default: random)')
|
||||
parser.add_argument('--mac', type=str, metavar='<mac>', help='MAC address (default: random)')
|
||||
parser.add_argument('--type', required=False, type=str, choices=[t.name for t in list(ServiceType)], default=ServiceType.STREAM.name, help='Service type')
|
||||
parser.add_argument('--type', required=False, type=str, choices=[t.name for t in list(ServiceType)], default=ServiceType.DEFAULT.name, help='Service type')
|
||||
parser.add_argument('--max', type=int, metavar='<max>', default=3, help='Max usage (default: 3)')
|
||||
parser.add_argument('--export', action='store_true', help='Export clear content key if available')
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user