zoukankan      html  css  js  c++  java
  • GitLab as an OAuth2 provider for Python or fastAPI

    import os
    import time
    import requests as rest_client
    from urllib.parse import urljoin
    from fastapi import APIRouter
    from starlette.responses import RedirectResponse
    from starlette.exceptions import HTTPException
    from starlette.requests import Request
    from starlette.status import (
        HTTP_400_BAD_REQUEST
    )
    
    GITLAB_BASE_DOMAIN = 'https://git.youdomain.com'
    APP_ID = os.environ.get("APP_ID", None)
    APP_SECRET = os.environ.get("APP_SECRET", None)
    REDIRECT_URI = os.environ.get("REDIRECT_URI", None)
    
    router = APIRouter()
    
    
    @router.get("/users/login", tags=["authentication"])
    async def login():
        unique_state_hash = int(round(time.time() * 1000))
        url = f"{GITLAB_BASE_DOMAIN}/oauth/authorize?client_id={APP_ID}&redirect_uri={REDIRECT_URI}&response_type=code&state={unique_state_hash}"
        return RedirectResponse(
            url=url)
    
    
    @router.get("/oauth2/callback/gitlab", tags=["authentication"])
    async def callback_gitlab(request: Request):
        code = request.query_params.get("code", None)
        res = rest_client.post(f"{GITLAB_BASE_DOMAIN}/oauth/token", data={'client_id': APP_ID,
                                                                          'client_secret': APP_SECRET,
                                                                          'code': code,
                                                                          'grant_type': 'authorization_code',
                                                                          'redirect_uri': REDIRECT_URI})
        json_res = res.json()
        access_token = "access_token"
        if access_token not in json_res:
            raise HTTPException(
                status_code=HTTP_400_BAD_REQUEST,
                detail=f"GitLab OAuth2 failure"
            )
        user_info_request = urljoin(GITLAB_BASE_DOMAIN, f"/api/v4/user?access_token={json_res.get(access_token,'')}")
        user_res = rest_client.get(user_info_request)
        user_res_data = user_res.json()
        print(user_res_data)
        return user_res_data
  • 相关阅读:
    变量的创建和初始化
    HDU 1114 Piggy-Bank (dp)
    HDU 1421 搬寝室 (dp)
    HDU 2059 龟兔赛跑 (dp)
    HDU 2571 命运 (dp)
    HDU 1574 RP问题 (dp)
    HDU 2577 How to Type (字符串处理)
    HDU 1422 重温世界杯 (dp)
    HDU 2191 珍惜现在,感恩生活 (dp)
    HH实习 acm算法部 1689
  • 原文地址:https://www.cnblogs.com/similarface/p/12982805.html
Copyright © 2011-2022 走看看