azmat1127/yingqing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add AirShelf core implementation

Browse Source
This commit is contained in:
zyc
2026-06-05 10:21:40 +08:00
parent 2ba1058329
commit cfdcd84a30
252 changed files with 70828 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/backend/apps/billing/migrations/0001_initial.py
+277
View File
@@ -0,0 +1,277 @@
# Generated by Django 5.1.15 on 2026-05-29 03:59
import django.db.models.deletion
import uuid
from django.conf import settings
from django.db import migrations, models
class Migration(migrations.Migration):
initial = True
dependencies = [
("accounts", "0001_initial"),
("ai", "0002_initial"),
("projects", "0001_initial"),
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
]
operations = [
migrations.CreateModel(
name="CreditAccount",
fields=[
(
"id",
models.UUIDField(
default=uuid.uuid4,
editable=False,
primary_key=True,
serialize=False,
),
),
("created_at", models.DateTimeField(auto_now_add=True)),
("updated_at", models.DateTimeField(auto_now=True)),
(
"balance",
models.DecimalField(decimal_places=4, default=0, max_digits=14),
),
(
"reserved_balance",
models.DecimalField(decimal_places=4, default=0, max_digits=14),
),
("currency", models.CharField(default="CNY", max_length=16)),
(
"team",
models.OneToOneField(
on_delete=django.db.models.deletion.CASCADE,
related_name="credit_account",
to="accounts.team",
),
),
],
options={
"abstract": False,
},
),
migrations.CreateModel(
name="CreditReservation",
fields=[
(
"id",
models.UUIDField(
default=uuid.uuid4,
editable=False,
primary_key=True,
serialize=False,
),
),
("created_at", models.DateTimeField(auto_now_add=True)),
("updated_at", models.DateTimeField(auto_now=True)),
("amount", models.DecimalField(decimal_places=4, max_digits=14)),
(
"status",
models.CharField(
choices=[
("active", "Active"),
("released", "Released"),
("charged", "Charged"),
("cancelled", "Cancelled"),
],
default="active",
max_length=32,
),
),
("expires_at", models.DateTimeField(blank=True, null=True)),
(
"project",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.SET_NULL,
related_name="credit_reservations",
to="projects.project",
),
),
(
"task",
models.OneToOneField(
on_delete=django.db.models.deletion.CASCADE,
related_name="credit_reservation",
to="ai.aitask",
),
),
(
"team",
models.ForeignKey(
on_delete=django.db.models.deletion.CASCADE,
related_name="credit_reservations",
to="accounts.team",
),
),
(
"user",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.SET_NULL,
related_name="credit_reservations",
to=settings.AUTH_USER_MODEL,
),
),
],
options={
"abstract": False,
},
),
migrations.CreateModel(
name="QuotaPolicy",
fields=[
(
"id",
models.UUIDField(
default=uuid.uuid4,
editable=False,
primary_key=True,
serialize=False,
),
),
("created_at", models.DateTimeField(auto_now_add=True)),
("updated_at", models.DateTimeField(auto_now=True)),
(
"monthly_limit",
models.DecimalField(
blank=True, decimal_places=4, max_digits=14, null=True
),
),
(
"project_limit",
models.DecimalField(
blank=True, decimal_places=4, max_digits=14, null=True
),
),
(
"per_task_limit",
models.DecimalField(
blank=True, decimal_places=4, max_digits=14, null=True
),
),
("is_active", models.BooleanField(default=True)),
(
"project",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.CASCADE,
related_name="quota_policies",
to="projects.project",
),
),
(
"team",
models.ForeignKey(
on_delete=django.db.models.deletion.CASCADE,
related_name="quota_policies",
to="accounts.team",
),
),
(
"user",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.CASCADE,
related_name="quota_policies",
to=settings.AUTH_USER_MODEL,
),
),
],
options={
"abstract": False,
},
),
migrations.CreateModel(
name="CreditLedger",
fields=[
(
"id",
models.UUIDField(
default=uuid.uuid4,
editable=False,
primary_key=True,
serialize=False,
),
),
("created_at", models.DateTimeField(auto_now_add=True)),
("updated_at", models.DateTimeField(auto_now=True)),
(
"ledger_type",
models.CharField(
choices=[
("recharge", "Recharge"),
("reserve", "Reserve"),
("release", "Release"),
("charge", "Charge"),
("adjustment", "Adjustment"),
("refund", "Refund"),
],
max_length=32,
),
),
("amount", models.DecimalField(decimal_places=4, max_digits=14)),
("balance_after", models.DecimalField(decimal_places=4, max_digits=14)),
("reason", models.CharField(blank=True, max_length=255)),
("metadata", models.JSONField(blank=True, default=dict)),
(
"project",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.SET_NULL,
related_name="credit_ledgers",
to="projects.project",
),
),
(
"task",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.SET_NULL,
related_name="credit_ledgers",
to="ai.aitask",
),
),
(
"team",
models.ForeignKey(
on_delete=django.db.models.deletion.CASCADE,
related_name="credit_ledgers",
to="accounts.team",
),
),
(
"user",
models.ForeignKey(
blank=True,
null=True,
on_delete=django.db.models.deletion.SET_NULL,
related_name="credit_ledgers",
to=settings.AUTH_USER_MODEL,
),
),
],
options={
"indexes": [
models.Index(
fields=["team", "ledger_type"],
name="billing_cre_team_id_e0f18f_idx",
),
models.Index(
fields=["project", "task"],
name="billing_cre_project_a79834_idx",
),
],
},
),
]
core/backend/apps/billing/migrations/__init__.py
+1
View File
@@ -0,0 +1 @@