Рефакторинг системы кошелька клиентов
Основные изменения: - Переход от денормализованного поля wallet_balance к вычисляемому балансу - Баланс теперь вычисляется как SUM(signed_amount) транзакций - Добавлено кеширование баланса для производительности (5 минут) - Новая модель WalletTransaction с полем signed_amount (может быть +/-) - WalletService для всех операций с кошельком (deposit, spend, adjustment) - Защита от отрицательного баланса и race conditions через select_for_update - Добавлен balance_after в каждую транзакцию для аудита - Обновлены миграции для переноса данных из старой схемы Улучшения безопасности: - Атомарные транзакции для всех операций с балансом - Блокировка строк при модификации баланса - Валидация недостаточности средств - Обязательное описание для корректировок баланса UI/UX изменения: - Обновлён вывод баланса кошелька в деталях клиента - Добавлена история транзакций с типами и описаниями - Цветовая индикация положительных транзакций (зелёный) Техническая документация: - Добавлены docstrings для всех методов WalletService - Комментарии к критичным участкам кода - Примеры использования в docstrings
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
# Generated by Django 5.0.10 on 2025-12-27 19:32
|
||||
|
||||
import django.db.models.deletion
|
||||
import phonenumber_field.modelfields
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
('customers', '0002_initial'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AlterField(
|
||||
model_name='customer',
|
||||
name='email',
|
||||
field=models.EmailField(blank=True, max_length=254, null=True, verbose_name='Email'),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name='customer',
|
||||
name='phone',
|
||||
field=phonenumber_field.modelfields.PhoneNumberField(blank=True, help_text='Введите телефон в любом формате, например: +375291234567, 80255270546, 8(029)1234567 - будет автоматически преобразован в международный формат', max_length=128, null=True, region=None, verbose_name='Телефон'),
|
||||
),
|
||||
migrations.CreateModel(
|
||||
name='ContactChannel',
|
||||
fields=[
|
||||
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||
('channel_type', models.CharField(choices=[('phone', 'Телефон'), ('email', 'Email'), ('telegram', 'Telegram'), ('instagram', 'Instagram'), ('whatsapp', 'WhatsApp'), ('viber', 'Viber'), ('vk', 'ВКонтакте'), ('facebook', 'Facebook'), ('other', 'Другое')], max_length=20, verbose_name='Тип канала')),
|
||||
('value', models.CharField(help_text='Username, номер телефона, email и т.д.', max_length=255, verbose_name='Значение')),
|
||||
('is_primary', models.BooleanField(default=False, verbose_name='Основной')),
|
||||
('notes', models.CharField(blank=True, max_length=255, verbose_name='Примечание')),
|
||||
('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')),
|
||||
('customer', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='contact_channels', to='customers.customer', verbose_name='Клиент')),
|
||||
],
|
||||
options={
|
||||
'verbose_name': 'Канал связи',
|
||||
'verbose_name_plural': 'Каналы связи',
|
||||
'ordering': ['-is_primary', 'channel_type'],
|
||||
'indexes': [models.Index(fields=['channel_type', 'value'], name='customers_c_channel_179e89_idx'), models.Index(fields=['customer'], name='customers_c_custome_f14e0e_idx')],
|
||||
'unique_together': {('channel_type', 'value')},
|
||||
},
|
||||
),
|
||||
]
|
||||
146
myproject/customers/migrations/0004_wallet_refactoring.py
Normal file
146
myproject/customers/migrations/0004_wallet_refactoring.py
Normal file
@@ -0,0 +1,146 @@
|
||||
# Generated by Django 5.0.10 on 2025-12-27 20:24
|
||||
|
||||
import django.db.models.deletion
|
||||
from decimal import Decimal
|
||||
from django.conf import settings
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
def populate_signed_amount(apps, schema_editor):
|
||||
"""
|
||||
Заполняем signed_amount на основе старого amount и типа транзакции.
|
||||
spend -> отрицательная сумма
|
||||
deposit/adjustment -> положительная сумма
|
||||
"""
|
||||
WalletTransaction = apps.get_model('customers', 'WalletTransaction')
|
||||
|
||||
for txn in WalletTransaction.objects.all():
|
||||
if txn.transaction_type == 'spend':
|
||||
txn.signed_amount = -abs(txn.amount)
|
||||
else:
|
||||
# deposit, adjustment - положительные
|
||||
txn.signed_amount = abs(txn.amount)
|
||||
txn.save(update_fields=['signed_amount'])
|
||||
|
||||
|
||||
def calculate_balance_after(apps, schema_editor):
|
||||
"""
|
||||
Вычисляем balance_after для всех существующих транзакций.
|
||||
"""
|
||||
Customer = apps.get_model('customers', 'Customer')
|
||||
WalletTransaction = apps.get_model('customers', 'WalletTransaction')
|
||||
|
||||
for customer in Customer.objects.all():
|
||||
running_balance = Decimal('0')
|
||||
|
||||
# Обрабатываем транзакции в хронологическом порядке
|
||||
for txn in WalletTransaction.objects.filter(customer=customer).order_by('created_at'):
|
||||
running_balance += txn.signed_amount or Decimal('0')
|
||||
txn.balance_after = running_balance
|
||||
txn.save(update_fields=['balance_after'])
|
||||
|
||||
|
||||
def reverse_populate(apps, schema_editor):
|
||||
"""Обратная операция - ничего не делаем."""
|
||||
pass
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
('customers', '0003_alter_customer_email_alter_customer_phone_and_more'),
|
||||
('orders', '0008_historicalorder_needs_delivery_photo_and_more'),
|
||||
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
||||
]
|
||||
|
||||
operations = [
|
||||
# 1. Добавляем новые поля (signed_amount временно nullable)
|
||||
migrations.AddField(
|
||||
model_name='wallettransaction',
|
||||
name='signed_amount',
|
||||
field=models.DecimalField(
|
||||
decimal_places=2,
|
||||
max_digits=10,
|
||||
null=True, # Временно nullable для миграции данных
|
||||
help_text='Положительная для пополнений, отрицательная для списаний',
|
||||
verbose_name='Сумма'
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name='wallettransaction',
|
||||
name='balance_after',
|
||||
field=models.DecimalField(
|
||||
blank=True,
|
||||
decimal_places=2,
|
||||
help_text='Баланс кошелька после применения этой транзакции',
|
||||
max_digits=10,
|
||||
null=True,
|
||||
verbose_name='Баланс после'
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name='wallettransaction',
|
||||
name='balance_category',
|
||||
field=models.CharField(
|
||||
choices=[('money', 'Реальные деньги')],
|
||||
default='money',
|
||||
max_length=20,
|
||||
verbose_name='Категория'
|
||||
),
|
||||
),
|
||||
|
||||
# 2. Копируем данные из amount в signed_amount
|
||||
migrations.RunPython(populate_signed_amount, reverse_populate),
|
||||
|
||||
# 3. Вычисляем balance_after
|
||||
migrations.RunPython(calculate_balance_after, reverse_populate),
|
||||
|
||||
# 4. Делаем signed_amount NOT NULL
|
||||
migrations.AlterField(
|
||||
model_name='wallettransaction',
|
||||
name='signed_amount',
|
||||
field=models.DecimalField(
|
||||
decimal_places=2,
|
||||
max_digits=10,
|
||||
help_text='Положительная для пополнений, отрицательная для списаний',
|
||||
verbose_name='Сумма'
|
||||
),
|
||||
),
|
||||
|
||||
# 5. Удаляем старое поле amount
|
||||
migrations.RemoveField(
|
||||
model_name='wallettransaction',
|
||||
name='amount',
|
||||
),
|
||||
|
||||
# 6. Удаляем wallet_balance из Customer
|
||||
migrations.RemoveField(
|
||||
model_name='customer',
|
||||
name='wallet_balance',
|
||||
),
|
||||
|
||||
# 7. Обновляем связь с Order (добавляем related_name)
|
||||
migrations.AlterField(
|
||||
model_name='wallettransaction',
|
||||
name='order',
|
||||
field=models.ForeignKey(
|
||||
blank=True,
|
||||
help_text='Заказ, к которому относится транзакция (если применимо)',
|
||||
null=True,
|
||||
on_delete=django.db.models.deletion.PROTECT,
|
||||
related_name='wallet_transactions',
|
||||
to='orders.order',
|
||||
verbose_name='Заказ'
|
||||
),
|
||||
),
|
||||
|
||||
# 8. Добавляем индексы
|
||||
migrations.AddIndex(
|
||||
model_name='wallettransaction',
|
||||
index=models.Index(fields=['balance_category'], name='customers_w_balance_81f0a9_idx'),
|
||||
),
|
||||
migrations.AddIndex(
|
||||
model_name='wallettransaction',
|
||||
index=models.Index(fields=['customer', 'balance_category'], name='customers_w_custome_060570_idx'),
|
||||
),
|
||||
]
|
||||
Reference in New Issue
Block a user