from django.conf import settings
from django.db import models
import uuid
from django.utils.translation import gettext_lazy as _
from oauth2_provider.settings import oauth2_settings
from rest_framework.exceptions import APIException

from apps.wallet.constans import StateChoices


class Charge(models.Model):
    uuid = models.UUIDField(primary_key=True, editable=True, default=uuid.uuid4, unique=True,
                            db_index=True)  # reserve_code
    created_at = models.DateTimeField(auto_now_add=True, null=True)
    updated_at = models.DateTimeField(auto_now=True, null=True)
    application = models.ForeignKey(oauth2_settings.APPLICATION_MODEL, on_delete=models.PROTECT,
                                    related_name='+', null=True)
    state = models.IntegerField(choices=StateChoices.choices, default=StateChoices.CREATED)
    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.PROTECT)
    amount = models.PositiveBigIntegerField(null=True, blank=True)

    def change_state(self, from_state, to_state, error_message=None, **data):
        if self.state == from_state:
            updated = Charge.objects.filter(pk=self.pk, state=from_state).update(
                state=to_state,
                **data
            )
            if updated:  # double check in db level
                self.state = to_state
            else:
                raise APIException(error_message)
        else:
            raise APIException(error_message)