Compare commits
2 Commits
d908d38e75
...
54d8ad46f8
Author | SHA1 | Date | |
---|---|---|---|
54d8ad46f8 | |||
8bbda345bb |
16
ajusta_bling/database/queries/auth.py
Normal file
16
ajusta_bling/database/queries/auth.py
Normal file
|
@ -0,0 +1,16 @@
|
|||
import psycopg2.extensions
|
||||
|
||||
|
||||
def insert_token(cur: psycopg2.extensions.cursor, ip_address: str, access_token: str, refresh_token: str, expires_in: str):
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO auth.tokens
|
||||
(ip_address,access_token,refresh_token,expires_in) VALUES (%s,%s,%s,%s)
|
||||
""",
|
||||
(
|
||||
ip_address, # ip
|
||||
access_token,
|
||||
refresh_token,
|
||||
expires_in,
|
||||
)
|
||||
)
|
|
@ -4,10 +4,10 @@ import secrets
|
|||
from base64 import b64encode
|
||||
from os import getenv
|
||||
|
||||
import bcrypt
|
||||
import requests
|
||||
from flask import Flask, redirect, render_template, request, session, url_for
|
||||
from flask import Flask, redirect, request, session, url_for
|
||||
|
||||
import ajusta_bling.database.queries.auth as sqlAuth
|
||||
from ajusta_bling.common import Args
|
||||
from ajusta_bling.database import Database
|
||||
|
||||
|
@ -45,38 +45,13 @@ def callback():
|
|||
data = payload,
|
||||
headers= {"Authorization": header})
|
||||
data = response.json()
|
||||
print(response.url)
|
||||
print(data)
|
||||
access_token: str = str(data["access_token"])
|
||||
refresh_token: str = str(data["refresh_token"])
|
||||
expires_in: int = int(data["expires_in"])
|
||||
|
||||
with db.get_cur() as cur: # TODO
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO auth.tokens
|
||||
(
|
||||
ip_address,
|
||||
access_token,
|
||||
refresh_token,
|
||||
expires_in
|
||||
) VALUES (
|
||||
%s,
|
||||
%s,
|
||||
%s,
|
||||
%s
|
||||
)
|
||||
""",
|
||||
(
|
||||
request.remote_addr, # ip
|
||||
access_token,
|
||||
refresh_token,
|
||||
str(expires_in),
|
||||
)
|
||||
)
|
||||
with db.get_cur() as cur:
|
||||
sqlAuth.insert_token(cur, request.remote_addr, access_token, refresh_token, expires_in)
|
||||
|
||||
#TODO: Store the session in the database
|
||||
#insert_session(payload['access_token'], payload['refresh_token'], payload['expires_in'])
|
||||
return redirect(url_for('index'))
|
||||
|
||||
@app.route("/")
|
||||
|
|
Loading…
Reference in New Issue
Block a user