Storage Quotas

import requests

def can_upload(tenant_id: str, file_size_bytes: int, api_key: str) -> bool:
    """
    Check if a tenant has enough quota before uploading.
    Call this before requesting a presigned URL.
    """
    response = requests.get(
        f"https://api.tenantbox.dev/api/storage/tenants/{tenant_id}/usage/",
        headers={"Authorization": f"Bearer {api_key}"},
    )
    response.raise_for_status()
    data = response.json()

    limit = data["storage_limit_bytes"]
    used  = data["storage_used_bytes"]

    # No limit set — always allow
    if limit is None:
        return True

    return (used + file_size_bytes) <= limit


# Usage
if can_upload("user_123", file.size, API_KEY):
    # Proceed with upload
    pass
else:
    raise Exception("Storage quota exceeded")

import requests

# Define your plan tiers in bytes
PLAN_LIMITS = {
    "free":       524288000,    # 500 MB
    "pro":        5368709120,   # 5 GB
    "enterprise": 53687091200,  # 50 GB
}

def set_tenant_quota(tenant_id: str, plan: str, api_key: str):
    """
    Set a tenant's storage limit based on their subscription plan.
    Call this when a user signs up or upgrades their plan.
    """
    limit_bytes = PLAN_LIMITS.get(plan)

    response = requests.patch(
        f"https://api.tenantbox.dev/api/storage/tenants/{tenant_id}/limit/",
        headers={
            "Authorization": f"Bearer {api_key}",
            "Content-Type": "application/json",
        },
        json={"storage_limit_bytes": limit_bytes},
    )
    response.raise_for_status()
    return response.json()


# New user signs up on free plan
set_tenant_quota("user_123", "free", API_KEY)

# User upgrades to pro
set_tenant_quota("user_123", "pro", API_KEY)

# Enterprise user — remove limit entirely
requests.patch(
    "https://api.tenantbox.dev/api/storage/tenants/user_123/limit/",
    headers={"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"},
    json={"storage_limit_bytes": None},
)

<!-- Vue component — storage usage bar -->
<script setup>
const props = defineProps({
  used: Number,   // storage_used_bytes
  limit: Number,  // storage_limit_bytes
})

const percent = computed(() => {
  if (!props.limit) return 0
  return Math.min(Math.round((props.used / props.limit) * 100), 100)
})

const barColor = computed(() => {
  if (percent.value >= 90) return 'bg-red-500'
  if (percent.value >= 70) return 'bg-yellow-500'
  return 'bg-green-500'
})

function formatBytes(bytes) {
  if (bytes === 0) return '0 B'
  const k = 1024
  const sizes = ['B', 'KB', 'MB', 'GB', 'TB']
  const i = Math.floor(Math.log(bytes) / Math.log(k))
  return parseFloat((bytes / Math.pow(k, i)).toFixed(1)) + ' ' + sizes[i]
}
</script>

<template>
  <div class="space-y-1.5">
    <div class="flex justify-between text-sm">
      <span>{{ formatBytes(used) }} used</span>
      <span v-if="limit">{{ percent }}% of {{ formatBytes(limit) }}</span>
      <span v-else>Unlimited</span>
    </div>
    <div v-if="limit" class="w-full h-2 bg-gray-200 rounded-full overflow-hidden">
      <div
        class="h-full rounded-full transition-all duration-500"
        :class="barColor"
        :style="{ width: percent + '%' }"
      />
    </div>
    <p v-if="percent >= 90" class="text-sm text-red-500">
      Storage almost full. Delete files or upgrade your plan.
    </p>
  </div>
</template>