mirror of
https://github.com/fastapi/fastapi.git
synced 2025-12-30 09:39:20 -05:00
fe453f80ed
* Update isort script to match changes in the new release, isort v5.0.2 * Downgrade isort to version v4.3.21 * Add an alternative flag to --recursive in isort v5.0.2 * Add isort config file * 🚚 Import from docs_src for tests * 🎨 Format dependencies.utils * 🎨 Remove isort combine_as_imports, keep black profile * 🔧 Update isort config, use pyproject.toml, Black profile * 🔧 Update format scripts to use explicit directories to format otherwise it would try to format venv env directories, I have several with different Python versions * 🎨 Format NoSQL tutorial after re-sorting imports * 🎨 Fix format for __init__.py Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from couchbase import LOCKMODE_WAIT
|
|
from couchbase.bucket import Bucket
|
|
from couchbase.cluster import Cluster, PasswordAuthenticator
|
|
from fastapi import FastAPI
|
|
from pydantic import BaseModel
|
|
|
|
USERPROFILE_DOC_TYPE = "userprofile"
|
|
|
|
|
|
def get_bucket():
|
|
cluster = Cluster(
|
|
"couchbase://couchbasehost:8091?fetch_mutation_tokens=1&operation_timeout=30&n1ql_timeout=300"
|
|
)
|
|
authenticator = PasswordAuthenticator("username", "password")
|
|
cluster.authenticate(authenticator)
|
|
bucket: Bucket = cluster.open_bucket("bucket_name", lockmode=LOCKMODE_WAIT)
|
|
bucket.timeout = 30
|
|
bucket.n1ql_timeout = 300
|
|
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
|
|
|
|
|
|
def get_user(bucket: Bucket, username: str):
|
|
doc_id = f"userprofile::{username}"
|
|
result = bucket.get(doc_id, quiet=True)
|
|
if not result.value:
|
|
return None
|
|
user = UserInDB(**result.value)
|
|
return user
|
|
|
|
|
|
# FastAPI specific code
|
|
app = FastAPI()
|
|
|
|
|
|
@app.get("/users/{username}", response_model=User)
|
|
def read_user(username: str):
|
|
bucket = get_bucket()
|
|
user = get_user(bucket=bucket, username=username)
|
|
return user
|