reformat

Author: mosquito

jwt_rsa/rsa.py

@@ -111,7 +111,7 @@ def int_to_base64url(value: int) -> str:
 
 @overload
 def rsa_to_jwk(
-    key: RSAPublicKey, *, kid: str = "", alg: AlgorithmType = "RS256", use: str = "sig"
+    key: RSAPublicKey, *, kid: str = "", alg: AlgorithmType = "RS256", use: str = "sig",
 ) -> RSAJWKPublicKey: ...
 
 

jwt_rsa/token.py

@@ -7,7 +7,8 @@
 
 from jwt import PyJWT
 
-from .types import AlgorithmType, RSAPrivateKey, RSAPublicKey, DateType
+from .types import AlgorithmType, DateType, RSAPrivateKey, RSAPublicKey
+
 
 R = TypeVar("R")
 DAY = 86400
@@ -51,12 +52,12 @@ def __init__(
         algorithm: AlgorithmType = "RS512",
         algorithms: Sequence[AlgorithmType] = ALGORITHMS,
     ):
-        super().__setattr__('public_key', key)
-        super().__setattr__('jwt', PyJWT(options))
-        super().__setattr__('expires', expires)
-        super().__setattr__('nbf_delta', nbf_delta)
-        super().__setattr__('algorithm', algorithm)
-        super().__setattr__('algorithms', algorithms)
+        super().__setattr__("public_key", key)
+        super().__setattr__("jwt", PyJWT(options))
+        super().__setattr__("expires", expires)
+        super().__setattr__("nbf_delta", nbf_delta)
+        super().__setattr__("algorithm", algorithm)
+        super().__setattr__("algorithms", algorithms)
 
     def decode(self, token: str, verify: bool = True, **kwargs: Any) -> Dict[str, Any]:
         return self.jwt.decode(token, key=self.public_key, verify=verify, algorithms=self.algorithms, **kwargs)
@@ -67,12 +68,12 @@ class JWTSigner(JWTDecoder):
     private_key: RSAPrivateKey = field(repr=False, compare=True)
 
     def __init__(self, key: RSAPrivateKey, *, options: Optional[Dict[str, Any]] = None, **kwargs: Any):
-        super(JWTDecoder, self).__setattr__('private_key', key)
+        super(JWTDecoder, self).__setattr__("private_key", key)
         super().__init__(key.public_key(), options=options, **kwargs)
 
     def encode(self, expired: DateType | EllipsisType = ..., nbf: DateType | EllipsisType = ..., **claims: Any) -> str:
-        claims.setdefault('exp', int(date_to_timestamp(expired, lambda: time.time() + self.expires)))
-        claims.setdefault('nbf', int(date_to_timestamp(nbf, lambda: time.time() - self.nbf_delta, timedelta_func=sub)))
+        claims.setdefault("exp", int(date_to_timestamp(expired, lambda: time.time() + self.expires)))
+        claims.setdefault("nbf", int(date_to_timestamp(nbf, lambda: time.time() - self.nbf_delta, timedelta_func=sub)))
         return self.jwt.encode(claims, self.private_key, algorithm=self.algorithm)
 
 

jwt_rsa/types.py

@@ -1,4 +1,4 @@
-from datetime import timedelta, datetime
+from datetime import datetime, timedelta
 from typing import Literal
 
 from cryptography.hazmat.primitives import serialization

jwt_rsa/verify.py

@@ -3,7 +3,7 @@
 from types import SimpleNamespace
 
 from .rsa import generate_rsa, load_private_key, load_public_key
-from .token import JWT, JWTSigner, JWTDecoder
+from .token import JWT, JWTDecoder, JWTSigner
 
 
 def main(arguments: SimpleNamespace) -> None:

tests/test_cli.py

@@ -95,7 +95,7 @@ def test_keygen_no_force(capsys, tmp_path):
         parser.parse_args([
             "keygen", "-o", "pem",
             "-K", str(private_path), "-k", str(public_path),
-        ])
+        ]),
     )
 
     assert private_path.exists()
@@ -112,7 +112,7 @@ def test_keygen_no_force(capsys, tmp_path):
         parser.parse_args([
             "keygen", "-o", "pem",
             "-K", str(private_path), "-k", str(public_path),
-        ])
+        ]),
     )
 
     assert public_content == public_path.read_text()
@@ -122,7 +122,7 @@ def test_keygen_no_force(capsys, tmp_path):
         parser.parse_args([
             "keygen", "-o", "pem", "-f",
             "-K", str(private_path), "-k", str(public_path),
-        ])
+        ]),
     )
 
     assert public_content != public_path.read_text()

tests/test_rsa.py

@@ -11,8 +11,8 @@
 from jwt.exceptions import InvalidAlgorithmError, InvalidSignatureError
 
 from jwt_rsa import rsa
-from jwt_rsa.types import serialization
 from jwt_rsa.token import JWT
+from jwt_rsa.types import serialization
 
 
 def test_rsa_sign():
@@ -173,17 +173,21 @@ def test_load_public_key(tmp_path):
     private_path = tmp_path / "private.pem"
 
     with open(public_path, "wb") as fp:
-        fp.write(public.public_bytes(
-            encoding=serialization.Encoding.PEM,
-            format=serialization.PublicFormat.SubjectPublicKeyInfo,
-        ))
+        fp.write(
+            public.public_bytes(
+                encoding=serialization.Encoding.PEM,
+                format=serialization.PublicFormat.SubjectPublicKeyInfo,
+            ),
+        )
 
     with open(private_path, "wb") as fp:
-        fp.write(key.private_bytes(
-            encoding=serialization.Encoding.PEM,
-            format=serialization.PrivateFormat.PKCS8,
-            encryption_algorithm=serialization.NoEncryption(),
-        ))
+        fp.write(
+            key.private_bytes(
+                encoding=serialization.Encoding.PEM,
+                format=serialization.PrivateFormat.PKCS8,
+                encryption_algorithm=serialization.NoEncryption(),
+            ),
+        )
 
     assert rsa.load_public_key(public_path)
     assert rsa.load_public_key(public_path.read_text())