From d25a9f1b797d84be8362dcb36ea70836db10969d Mon Sep 17 00:00:00 2001 From: Mike Crute Date: Sun, 1 Oct 2017 01:31:07 +0000 Subject: Make crypto class pluggable --- pandora/transport.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pandora/transport.py b/pandora/transport.py index c3fc71c..aeba50d 100644 --- a/pandora/transport.py +++ b/pandora/transport.py @@ -320,9 +320,9 @@ class Encryptor(object): API request and response. It handles the formats that the API expects. """ - def __init__(self, in_key, out_key): - self.bf_out = CryptographyBlowfish(out_key) - self.bf_in = CryptographyBlowfish(in_key) + def __init__(self, in_key, out_key, crypto_class=CryptographyBlowfish): + self.bf_out = crypto_class(out_key) + self.bf_in = crypto_class(in_key) @staticmethod def _decode_hex(data): -- cgit v1.2.3