nvidia/fastapi-dls
1
0
Fork 0
mirror of https://gitea.publichub.eu/oscar.krause/fastapi-dls.git synced 2024-11-22 14:28:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

replaced "timedelta" with "relativedelta"

Browse Source
This commit is contained in:
Oscar Krause 2022-12-19 13:11:28 +01:00
parent fbf91b3635
commit 03089e82e5
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/main.py
View File

@ -5,7 +5,8 @@ from uuid import uuid4
from fastapi import FastAPI, HTTPException from fastapi import FastAPI, HTTPException
from fastapi.requests import Request from fastapi.requests import Request
import json import json
from datetime import datetime, timedelta from datetime import datetime
from dateutil.relativedelta import relativedelta
from calendar import timegm from calendar import timegm
from jose import jws, jwk, jwt from jose import jws, jwk, jwt
from jose.constants import ALGORITHMS from jose.constants import ALGORITHMS
@ -45,7 +46,7 @@ async def client_token():
}, },
cur_time = datetime.utcnow() cur_time = datetime.utcnow()
exp_time = cur_time + timedelta(days=1) exp_time = cur_time + relativedelta(years=12)
payload = { payload = {
"jti": str(uuid4()), "jti": str(uuid4()),
"iss": "NLS Service Instance", "iss": "NLS Service Instance",
@ -126,7 +127,7 @@ async def code(request: Request):
# {"code_challenge":"QhDaArKDQwFeQ5Jq4Dn5hy37ODF8Jq3igXCXvWEgs5I","origin_ref":"00112233-4455-6677-8899-aabbccddeeff"} # {"code_challenge":"QhDaArKDQwFeQ5Jq4Dn5hy37ODF8Jq3igXCXvWEgs5I","origin_ref":"00112233-4455-6677-8899-aabbccddeeff"}
cur_time = datetime.utcnow() cur_time = datetime.utcnow()
expires = cur_time + timedelta(days=1) expires = cur_time + relativedelta(days=1)
payload = { payload = {
'iat': timegm(cur_time.timetuple()), 'iat': timegm(cur_time.timetuple()),
@ -171,7 +172,7 @@ async def token(request: Request):
raise HTTPException(status_code=403, detail='expected challenge did not match verifier') raise HTTPException(status_code=403, detail='expected challenge did not match verifier')
cur_time = datetime.utcnow() cur_time = datetime.utcnow()
access_expires_on = cur_time + timedelta(days=1) access_expires_on = cur_time + relativedelta(days=1)
new_payload = { new_payload = {
'iat': timegm(cur_time.timetuple()), 'iat': timegm(cur_time.timetuple()),
@ -216,7 +217,7 @@ async def lessor(request: Request):
"lease": { "lease": {
"ref": scope_ref, "ref": scope_ref,
"created": cur_time, "created": cur_time,
"expires": cur_time + timedelta(days=90), "expires": cur_time + relativedelta(minutes=15), # days=90
"recommended_lease_renewal": 0.15, "recommended_lease_renewal": 0.15,
"offline_lease": "true", "offline_lease": "true",
"license_type": "CONCURRENT_COUNTED_SINGLE" "license_type": "CONCURRENT_COUNTED_SINGLE"