mirror of https://github.com/tiangolo/fastapi.git
67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
from typing import List, Set
|
|
|
|
from fastapi import Body, FastAPI, Path, Query, Depends, Cookie
|
|
from pydantic import BaseModel
|
|
from pydantic.types import UrlStr
|
|
from starlette.status import HTTP_201_CREATED
|
|
from starlette.responses import HTMLResponse
|
|
from random import choice
|
|
|
|
from typing import List, Optional, Union
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from app.models.config import USERPROFILE_DOC_TYPE
|
|
from app.models.role import RoleEnum
|
|
from couchbase.bucket import Bucket
|
|
from couchbase.cluster import Cluster, PasswordAuthenticator
|
|
from couchbase import LOCKMODE_WAIT
|
|
|
|
|
|
def get_bucket():
|
|
cluster = Cluster("couchbase://couchbasehost:8091")
|
|
authenticator = PasswordAuthenticator("username", "password")
|
|
cluster.authenticate(authenticator)
|
|
bucket: Bucket = cluster.open_bucket("bucket_name", lockmode=LOCKMODE_WAIT)
|
|
return bucket
|
|
|
|
|
|
class User(BaseModel):
|
|
username: str
|
|
email: Optional[str] = None
|
|
full_name: Optional[str] = None
|
|
disabled: Optional[bool] = None
|
|
|
|
|
|
class UserInDB(User):
|
|
type: str = USERPROFILE_DOC_TYPE
|
|
hashed_password: str
|
|
|
|
class Meta:
|
|
key: Optional[str] = None
|
|
|
|
|
|
def get_user_doc_id(username):
|
|
return f"userprofile::{username}"
|
|
|
|
|
|
def get_user(bucket: Bucket, username: str):
|
|
doc_id = get_user_doc_id(username)
|
|
result = bucket.get(doc_id, quiet=True)
|
|
if not result.value:
|
|
return None
|
|
user = UserInDB(**result.value)
|
|
user.Meta.key = result.key
|
|
return user
|
|
|
|
|
|
# FastAPI specific code
|
|
app = FastAPI()
|
|
|
|
|
|
@app.get("/users/{username}")
|
|
def read_user(username: str):
|
|
bucket = get_bucket()
|
|
user = get_user(bucket=bucket, username=username)
|
|
return user
|