Tabnine

# Tabnine 코드 완성 예시

# 1. 함수 시그니처만 입력하면 전체 구현 제안
def calculate_order_total(items: list[dict], tax_rate: float = 0.1):
    # Tabnine이 아래 코드를 자동 완성
    subtotal = sum(item['price'] * item['quantity'] for item in items)
    tax = subtotal * tax_rate
    return {
        'subtotal': round(subtotal, 2),
        'tax': round(tax, 2),
        'total': round(subtotal + tax, 2)
    }

# 2. 주석으로 의도 설명하면 코드 생성
# TODO: Redis 연결 및 캐시 조회 함수
import redis

def get_cached_data(key: str, ttl: int = 3600):
    """Redis에서 캐시 데이터 조회, 없으면 None 반환"""
    client = redis.Redis(host='localhost', port=6379, db=0)
    try:
        data = client.get(key)
        return data.decode('utf-8') if data else None
    except redis.ConnectionError:
        return None

def set_cached_data(key: str, value: str, ttl: int = 3600):
    """Redis에 데이터 캐싱"""
    client = redis.Redis(host='localhost', port=6379, db=0)
    client.setex(key, ttl, value)

# 3. 테스트 코드 자동 생성
# test_ 입력하면 해당 함수의 테스트 제안
def test_calculate_order_total():
    items = [
        {'name': 'Product A', 'price': 100, 'quantity': 2},
        {'name': 'Product B', 'price': 50, 'quantity': 1}
    ]
    result = calculate_order_total(items, tax_rate=0.1)

    assert result['subtotal'] == 250
    assert result['tax'] == 25
    assert result['total'] == 275

# 4. 반복 패턴 인식
# 첫 번째 API 엔드포인트 작성 후 유사 패턴 자동 완성
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel

app = FastAPI()

class User(BaseModel):
    id: int
    name: str
    email: str

@app.get("/users/{user_id}")
async def get_user(user_id: int):
    # Tabnine: DB 조회 패턴 제안
    user = await db.users.find_one({"id": user_id})
    if not user:
        raise HTTPException(status_code=404, detail="User not found")
    return user

# 위 패턴 학습 후 아래 엔드포인트 자동 제안
@app.delete("/users/{user_id}")
async def delete_user(user_id: int):
    result = await db.users.delete_one({"id": user_id})
    if result.deleted_count == 0:
        raise HTTPException(status_code=404, detail="User not found")
    return {"message": "User deleted successfully"}