Files
chat_backend/llm_be/finance/admin.py
T
westfarn d54094f5e0
Unit Tests / test (push) Successful in 10s
Deploy Beta / unit-tests (push) Successful in 10s
Deploy Beta / docker (push) Successful in 21s
Deploy Beta / deploy-beta (push) Successful in 40s
Tier-gated RAG + Drive document sources (#42) (#54)
## Summary

Implements epic [#42](#42) (children #43–#53) and advances [#11](#11).

- **Entitlement:** `allows_rag` on plans (founders / backer / pro / business; not standard); exposed as `features.rag`
- **Gates:** document REST + WS `PromptType.RAG` use `assert_feature_allowed(..., "rag")`
- **Lifecycle:** dedupe ingest, delete vectors by `document_id`, honor `active`, fix document detail PATCH/DELETE
- **Workspaces:** auto-create default company workspace; fail-closed scoping
- **Drive:** personal + company Google/Microsoft connect (`link_drive` / `link_company_drive`), resource selection, sync, webhooks stubs, `sync_drive_connections` management command
- **Docs/env:** README + `.env*.example` updated

Companion FE: `chat_web_app` branch `feature/rag-epic-42-ui` (#81–#85).

## Test plan

- [x] `SKIP_RAG_INIT=1 uv run python manage.py test` (457 OK)
- [ ] Migrate finance `0004` + chat_backend `0028` on beta
- [ ] Verify Standard user: Documents API 403 + no RAG retrieval
- [ ] Verify Founders/Pro: upload + list + active toggle
- [ ] Connect Google/Microsoft Drive (incremental scopes) and Sync
- [ ] Company manager: `link_company_drive`; non-manager 403
- [ ] Run `manage.py sync_drive_connections`Reviewed-on: #54
2026-08-01 14:02:36 -07:00

134 lines
3.5 KiB
Python

from django.contrib import admin
from finance.models import BackerEmail, Invoice, Payment, SubscriptionPlan, UserSubscription
@admin.register(SubscriptionPlan)
class SubscriptionPlanAdmin(admin.ModelAdmin):
list_display = (
"slug",
"name",
"price_cents",
"is_public",
"is_selectable",
"allows_image_generation",
"allows_rag",
"allows_all_future_features",
"prompt_quota_per_window",
"prompt_window_hours",
"monthly_token_quota",
"sort_order",
)
list_filter = ("is_public", "is_selectable", "allows_image_generation", "allows_rag")
search_fields = ("slug", "name", "stripe_price_id")
readonly_fields = ("created", "last_modified")
prepopulated_fields = {"slug": ("name",)}
@admin.register(BackerEmail)
class BackerEmailAdmin(admin.ModelAdmin):
list_display = ("email", "note", "redeemed_at", "redeemed_user", "created")
search_fields = ("email", "note", "redeemed_user__email")
raw_id_fields = ("redeemed_user",)
readonly_fields = ("created", "last_modified", "redeemed_at", "redeemed_user")
@admin.register(UserSubscription)
class UserSubscriptionAdmin(admin.ModelAdmin):
list_display = (
"user",
"plan",
"status",
"source",
"stripe_subscription_id",
"monthly_token_quota_override",
"created",
)
list_filter = ("status", "source", "plan")
search_fields = ("user__email", "user__username", "stripe_subscription_id")
raw_id_fields = ("user", "plan")
readonly_fields = ("created", "last_modified")
class PaymentInline(admin.TabularInline):
model = Payment
extra = 0
readonly_fields = (
"provider",
"status",
"amount",
"currency",
"stripe_payment_intent_id",
"stripe_charge_id",
"paid_at",
"failure_message",
"created",
"last_modified",
)
can_delete = False
show_change_link = True
@admin.register(Invoice)
class InvoiceAdmin(admin.ModelAdmin):
list_display = (
"id",
"user",
"company",
"provider",
"status",
"amount_due",
"amount_paid",
"currency",
"period_start",
"period_end",
"stripe_invoice_id",
"stripe_checkout_session_id",
"created",
)
list_filter = ("provider", "status", "currency")
search_fields = (
"user__email",
"user__username",
"company__name",
"stripe_invoice_id",
"stripe_checkout_session_id",
"stripe_subscription_id",
"stripe_customer_id",
"description",
)
readonly_fields = ("created", "last_modified")
raw_id_fields = ("user", "company")
inlines = [PaymentInline]
date_hierarchy = "created"
@admin.register(Payment)
class PaymentAdmin(admin.ModelAdmin):
list_display = (
"id",
"user",
"company",
"invoice",
"provider",
"status",
"amount",
"currency",
"stripe_payment_intent_id",
"stripe_charge_id",
"paid_at",
"created",
)
list_filter = ("provider", "status", "currency")
search_fields = (
"user__email",
"user__username",
"company__name",
"stripe_payment_intent_id",
"stripe_charge_id",
"invoice__stripe_invoice_id",
)
readonly_fields = ("created", "last_modified")
raw_id_fields = ("user", "company", "invoice")
date_hierarchy = "created"