22 lines
659 B
Python
22 lines
659 B
Python
|
from rest_framework.decorators import api_view
|
||
|
|
from rest_framework.response import Response
|
||
|
|
from django.db import transaction
|
||
|
|
from users.models import ReferralTransaction
|
||
|
|
|
||
|
|
|
||
|
|
@transaction.atomic
|
||
|
|
@api_view(['POST'])
|
||
|
|
def empty_storage(request):
|
||
|
|
tg_user = request.user.tg_user
|
||
|
|
if tg_user.referral_storage == 0:
|
||
|
|
return Response(data={'points': str(tg_user.points)})
|
||
|
|
ReferralTransaction.objects.create(
|
||
|
|
user_id=tg_user.pk,
|
||
|
|
value=tg_user.referral_storage
|
||
|
|
)
|
||
|
|
tg_user.refresh_from_db()
|
||
|
|
tg_user.referral_storage = 0
|
||
|
|
tg_user.save()
|
||
|
|
tg_user.refresh_from_db()
|
||
|
|
return Response(data={'points': str(tg_user.points)})
|