oauth via github

This commit is contained in:
knst-kotov 2021-07-09 07:14:16 +00:00
parent e6b0866f6e
commit 571dad6f60
6 changed files with 76 additions and 31 deletions

View File

@ -15,3 +15,11 @@ class Identity:
if not Password.verify(password, user.password):
raise InvalidPassword("Wrong user password")
return user
@staticmethod
def identity_oauth(oauth_id, input) -> User:
user = global_session.query(OrmUser).filter_by(oauth_id=oauth_id).first()
if not user:
user = OrmUser.create(**input)
user = User(**user.dict())
return user

View File

@ -1,6 +1,9 @@
from authlib.integrations.starlette_client import OAuth
from starlette.responses import PlainTextResponse
from auth.authorize import Authorize
from auth.identity import Identity
oauth = OAuth()
oauth.register(
@ -15,14 +18,34 @@ oauth.register(
client_kwargs={'scope': 'user:email'},
)
oauth.register(
name='github',
client_id='58877ba7ad9baef280b4',
client_secret='',
access_token_url='https://github.com/login/oauth/access_token',
access_token_params=None,
authorize_url='https://github.com/login/oauth/authorize',
authorize_params=None,
api_base_url='https://api.github.com/',
client_kwargs={'scope': 'user:email'},
)
async def oauth_login(request):
facebook = oauth.create_client('facebook')
github = oauth.create_client('github')
redirect_uri = request.url_for('oauth_authorize')
return await facebook.authorize_redirect(request, redirect_uri)
return await github.authorize_redirect(request, redirect_uri)
async def oauth_authorize(request):
facebook = oauth.create_client('facebook')
token = await facebook.authorize_access_token(request)
email = await facebook.parse_id_token(request, token)
print(email)
return PlainTextResponse("%s auth" % email)
github = oauth.create_client('github')
token = await github.authorize_access_token(request)
resp = await github.get('user', token=token)
profile = resp.json()
oauth_id = profile["id"]
user_input = {
"oauth_id" : oauth_id,
"email" : profile["email"],
"username" : profile["name"]
}
user = Identity.identity_oauth(oauth_id=oauth_id, input=user_input)
token = await Authorize.authorize(user, device="pc", auto_delete=False)
return PlainTextResponse(token)

10
create_crt.sh Normal file
View File

@ -0,0 +1,10 @@
#!/bin/bash
openssl req -newkey rsa:4096 \
-x509 \
-sha256 \
-days 3650 \
-nodes \
-out discours.crt \
-keyout discours.key \
-subj "/C=RU/ST=Moscow/L=Moscow/O=Discours/OU=Site/CN=10.0.0.187"

View File

@ -5,6 +5,7 @@ from ariadne.asgi import GraphQL
from starlette.applications import Starlette
from starlette.middleware import Middleware
from starlette.middleware.authentication import AuthenticationMiddleware
from starlette.middleware.sessions import SessionMiddleware
from starlette.routing import Route
from auth.authenticate import JWTAuthenticate
@ -15,8 +16,10 @@ from resolvers.base import resolvers
import_module('resolvers')
schema = make_executable_schema(load_schema_from_path("schema/"), resolvers)
middleware = [Middleware(AuthenticationMiddleware, backend=JWTAuthenticate())]
middleware = [
Middleware(AuthenticationMiddleware, backend=JWTAuthenticate()),
Middleware(SessionMiddleware, secret_key="!secret")
]
async def start_up():
await redis.connect()
@ -30,6 +33,5 @@ routes = [
Route("/authorize", endpoint=oauth_authorize)
]
app = Starlette(debug=True, on_startup=[start_up], on_shutdown=[shutdown], middleware=middleware, routes=routes)
app.mount("/", GraphQL(schema, debug=True))

View File

@ -11,10 +11,12 @@ class User(Base):
email: str = Column(String, nullable=False)
username: str = Column(String, nullable=False, comment="Name")
password: str = Column(String, nullable=False, comment="Password")
password: str = Column(String, nullable=True, comment="Password")
role_id: int = Column(ForeignKey("role.id"), nullable=True, comment="Role")
oauth_id: str = Column(String, nullable=True)
@classmethod
def get_permission(cls, user_id):
perms: List[Permission] = cls.session.query(Permission).join(User, User.role_id == Permission.role_id).filter(

View File

@ -1,4 +1,4 @@
import uvicorn
if __name__ == '__main__':
uvicorn.run("main:app", host="0.0.0.0", port=8081, reload=True)
uvicorn.run("main:app", host="0.0.0.0", port=8081, ssl_keyfile="discours.key", ssl_certfile="discours.crt", reload=True)