Merge pull request #689 from basedosdados/feat/change-user-gcp-email

Feat/change user gcp email

backend/apps/account/migrations/0018_account_gcp_email.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 4.2.16 on 2024-09-29 22:44
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+ dependencies = [
+ ("account", "0017_account_available_for_research_and_more"),
+ ]
+
+ operations = [
+ migrations.AddField(
+ model_name="account",
+ name="gcp_email",
+ field=models.EmailField(
+ blank=True, max_length=254, null=True, verbose_name="GCP email"
+ ),
+ ),
+ ]

backend/apps/account/models.py

@@ -211,6 +211,7 @@ class Account(BaseModel, AbstractBaseUser, PermissionsMixin):
  uuid = models.UUIDField(primary_key=False, default=uuid4)
 
  email = models.EmailField("Email", unique=True)
+ gcp_email = models.EmailField("GCP email", null=True, blank=True) # Google Cloud Platform email
  username = models.CharField("Username", max_length=40, blank=True, null=True, unique=True)
 
  first_name = models.CharField("Nome", max_length=40, blank=True)

backend/apps/account/signals.py

@@ -62,7 +62,7 @@ def create_subscription(user: Account):
  customer.subscriber = user
  customer.save()
  # Add user to Google Group
- add_user(user.email)
+ add_user(user.gcp_email or user.email)
 
 
 @receiver(post_save, sender=Account)

backend/apps/account_payment/graphql.py

@@ -24,7 +24,7 @@
 from stripe import SetupIntent
 
 from backend.apps.account.models import Account, Subscription
-from backend.apps.account_payment.webhooks import add_user, remove_user
+from backend.apps.account_payment.webhooks import add_user, is_email_in_group, remove_user
 from backend.custom.environment import get_backend_url
 from backend.custom.graphql_base import CountableConnection, PlainTextNode
 
@@ -384,7 +384,7 @@ def mutate(cls, root, info, account_id, subscription_id):
 
  subscription = Subscription.objects.get(id=subscription_id)
  assert admin.id == subscription.admin.id
- add_user(account.email)
+ add_user(account.gcp_email or account.email)
  subscription.subscribers.add(account)
  return cls(ok=True)
  except Exception as e:
@@ -410,14 +410,69 @@ def mutate(cls, root, info, account_id, subscription_id):
  account = Account.objects.get(id=account_id)
  subscription = Subscription.objects.get(id=subscription_id)
  assert admin.id == subscription.admin.id
- remove_user(account.email)
+ remove_user(account.gcp_email or account.email)
  subscription.subscribers.remove(account)
  return cls(ok=True)
  except Exception as e:
  logger.error(e)
  return cls(errors=[str(e)])
 
 
+class ChangeUserGCPEmail(Mutation):
+ """Change user GCP email"""
+
+ ok = Boolean()
+ errors = List(String)
+
+ class Arguments:
+ email = String(required=True)
+
+ @classmethod
+ @login_required
+ def mutate(cls, root, info, email):
+ try:
+ user = info.context.user
+ old_email = user.gcp_email or user.email
+
+ if old_email == email:
+ return cls(ok=True)
+
+ user.gcp_email = email
+ user.save()
+
+ if is_email_in_group(old_email):
+ remove_user(old_email)
+
+ subscription = user.pro_subscription()
+ if subscription and not is_email_in_group(email):
+ add_user(email)
+
+ return cls(ok=True)
+ except Exception as e:
+ logger.error(e)
+ return cls(errors=[str(e)])
+
+
+# Query to check based on a email if the user is in a group
+class IsEmailInGoogleGroup(Mutation):
+ """Check if user is in group"""
+
+ ok = Boolean()
+ errors = List(String)
+
+ class Arguments:
+ email = String(required=True)
+
+ @classmethod
+ @login_required
+ def mutate(cls, root, info, email):
+ try:
+ return cls(ok=is_email_in_group(email))
+ except Exception as e:
+ logger.error(e)
+ return cls(errors=[str(e)])
+
+
 def get_stripe_promo(promotion_code):
  """
  Helper function to retrieve a Stripe Promotion Code by its code.
@@ -443,6 +498,7 @@ def get_stripe_promo(promotion_code):
 class Query(ObjectType):
  stripe_price = PlainTextNode.Field(StripePriceNode)
  all_stripe_price = DjangoFilterConnectionField(StripePriceNode)
+ is_email_in_google_group = IsEmailInGoogleGroup.Field()
 
 
 class Mutation(ObjectType):
@@ -453,6 +509,7 @@ class Mutation(ObjectType):
  create_stripe_customer_subscription = StripeSubscriptionCustomerCreateMutation.Field()
  update_stripe_customer_subscription = StripeSubscriptionCustomerDeleteMutation.Field()
  validate_stripe_coupon = StripeCouponValidationMutation.Field()
+ change_user_gcp_email = ChangeUserGCPEmail.Field()
 
 
 # Reference

backend/apps/account_payment/webhooks.py

@@ -10,7 +10,7 @@
 from stripe import Customer as StripeCustomer
 from stripe import Subscription as StripeSubscription
 
-from backend.apps.account.models import Subscription
+from backend.apps.account.models import Account, Subscription
 from backend.custom.client import send_discord_message as send
 from backend.custom.environment import get_backend_url
 
@@ -110,6 +110,31 @@ def list_user(group_key: str = None):
  raise e
 
 
+def is_email_in_group(email: str, group_key: str = None) -> bool:
+ """Check if a user is in a Google group."""
+ if not group_key:
+ group_key = settings.GOOGLE_DIRECTORY_GROUP_KEY
+ if "+" in email and email.index("+") < email.index("@"):
+ email = email.split("+")[0] + "@" + email.split("@")[1]
+
+ try:
+ service = get_service()
+ service.members().get(
+ groupKey=group_key,
+ memberKey=email,
+ ).execute()
+ return True
+ except HttpError as e:
+ if e.resp.status == 404:
+ return False
+ else:
+ logger.error(f"Erro ao verificar o usuário {email} no grupo {group_key}: {e}")
+ raise e
+ except Exception as e:
+ logger.error(f"Erro inesperado ao verificar o usuário {email}: {e}")
+ raise e
+
+
 @webhooks.handler("customer.updated")
 def update_customer(event: Event, **kwargs):
  """Propagate customer email update if exists"""
@@ -123,21 +148,29 @@ def update_customer(event: Event, **kwargs):
 def handle_subscription(event: Event):
  """Handle subscription status"""
  subscription = get_subscription(event)
+ account = Account.objects.filter(email=event.customer.email).first()
 
  if event.data["object"]["status"] in ["trialing", "active"]:
  if subscription:
  logger.info(f"Adicionando a inscrição do cliente {event.customer.email}")
  subscription.is_active = True
  subscription.save()
+
  # Add user to google group if subscription exists or not
- add_user(event.customer.email)
+ if account:
+ add_user(account.gcp_email or account.email)
+ else:
+ add_user(event.customer.email)
  else:
  if subscription:
  logger.info(f"Removendo a inscrição do cliente {event.customer.email}")
  subscription.is_active = False
  subscription.save()
  # Remove user from google group if subscription exists or not
- remove_user(event.customer.email)
+ if account:
+ remove_user(account.gcp_email or account.email)
+ else:
+ remove_user(event.customer.email)
 
 
 @webhooks.handler("customer.subscription.updated")
@@ -159,39 +192,54 @@ def unsubscribe(event: Event, **kwargs):
  logger.info(f"Removendo a inscrição do cliente {event.customer.email}")
  subscription.is_active = False
  subscription.save()
+
+ account = Account.objects.filter(email=event.customer.email).first()
  # Remove user from google group if subscription exists or not
  try:
- remove_user(event.customer.email)
+ if account:
+ remove_user(account.gcp_email or account.email)
+ else:
+ remove_user(event.customer.email)
  except Exception as e:
  logger.error(e)
 
 
 @webhooks.handler("customer.subscription.paused")
 def pause_subscription(event: Event, **kwargs):
  """Pause customer subscription"""
+ account = Account.objects.filter(email=event.customer.email).first()
+
  if subscription := get_subscription(event):
  logger.info(f"Pausando a inscrição do cliente {event.customer.email}")
  subscription.is_active = False
  subscription.save()
 
- try:
+ try:
+ if account:
+ remove_user(account.gcp_email or account.email)
+ else:
  remove_user(event.customer.email)
-  except Exception as e:
-  logger.error(e)
+ except Exception as e:
+ logger.error(e)
 
 
 @webhooks.handler("customer.subscription.resumed")
 def resume_subscription(event: Event, **kwargs):
  """Resume customer subscription"""
+ account = Account.objects.filter(email=event.customer.email).first()
+
  if subscription := get_subscription(event):
  logger.info(f"Resumindo a inscrição do cliente {event.customer.email}")
  subscription.is_active = True
  subscription.save()
 
- try:
+ try:
+ if account:
+ add_user(account.gcp_email or account.email)
+ else:
  add_user(event.customer.email)
-  except Exception as e:
-  logger.error(e)
+ except Exception as e:
+ logger.error(e)
 
 
 @webhooks.handler("setup_intent.succeeded")