Skip to content

Commit ad5fc48

Browse files
feat(ra-tls): update feature to comply with ruff and pyright
1 parent 4f61e64 commit ad5fc48

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/flare_ai_kit/tee/ra_tls.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def generate_key_and_csr(
7474
return key_pem, csr_pem
7575

7676

77-
def generate_self_signed_cert(token, common_name, days_valid):
77+
def generate_self_signed_cert(token: str, common_name: str, days_valid: int):
7878
# logger.info("Generating self-signed certificate with SAN")
7979
try:
8080
# Generate private key
@@ -97,9 +97,9 @@ def generate_self_signed_cert(token, common_name, days_valid):
9797
]
9898
)
9999
)
100-
builder = builder.not_valid_before(datetime.datetime.utcnow())
100+
builder = builder.not_valid_before(datetime.datetime.now(datetime.UTC))
101101
builder = builder.not_valid_after(
102-
datetime.datetime.utcnow() + datetime.timedelta(days=days_valid)
102+
datetime.datetime.now(datetime.UTC) + datetime.timedelta(days=days_valid)
103103
)
104104
builder = builder.serial_number(x509.random_serial_number())
105105
builder = builder.public_key(public_key)
@@ -162,9 +162,9 @@ def generate_self_signed_cert_OLD(
162162
.issuer_name(issuer)
163163
.public_key(private_key.public_key())
164164
.serial_number(x509.random_serial_number())
165-
.not_valid_before(datetime.datetime.utcnow())
165+
.not_valid_before(datetime.datetime.now(datetime.UTC))
166166
.not_valid_after(
167-
datetime.datetime.utcnow() + datetime.timedelta(days=days_valid)
167+
datetime.datetime.now(datetime.UTC) + datetime.timedelta(days=days_valid)
168168
)
169169
.add_extension(
170170
x509.SubjectAlternativeName([x509.DNSName("localhost")]), critical=False

0 commit comments

Comments
 (0)