chore: changed config variables

This commit is contained in:
NarayanBavisetti 2023-12-07 12:42:02 +05:30
parent b912bb21ad
commit 5a364855e8
6 changed files with 140 additions and 120 deletions

View File

@ -387,16 +387,15 @@ class EmailCheckEndpoint(BaseAPIView):
) )
# Send event # Send event
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium="MAGIC_LINK",
medium="MAGIC_LINK", first_time=True,
first_time=True, )
)
key, token, current_attempt = generate_magic_token(email=email) key, token, current_attempt = generate_magic_token(email=email)
if not current_attempt: if not current_attempt:
return Response( return Response(
@ -420,16 +419,15 @@ class EmailCheckEndpoint(BaseAPIView):
status=status.HTTP_400_BAD_REQUEST, status=status.HTTP_400_BAD_REQUEST,
) )
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium="MAGIC_LINK",
medium="MAGIC_LINK", first_time=False,
first_time=False, )
)
# Generate magic token # Generate magic token
key, token, current_attempt = generate_magic_token(email=email) key, token, current_attempt = generate_magic_token(email=email)
@ -449,16 +447,15 @@ class EmailCheckEndpoint(BaseAPIView):
status=status.HTTP_200_OK, status=status.HTTP_200_OK,
) )
else: else:
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium="EMAIL",
medium="EMAIL", first_time=False,
first_time=False, )
)
# User should enter password to login # User should enter password to login
return Response( return Response(

View File

@ -73,7 +73,7 @@ class SignUpEndpoint(BaseAPIView):
# get configuration values # get configuration values
# Get configuration values # Get configuration values
ENABLE_SIGNUP = get_configuration_value( ENABLE_SIGNUP, = get_configuration_value(
[ [
{ {
"key": "ENABLE_SIGNUP", "key": "ENABLE_SIGNUP",
@ -173,7 +173,7 @@ class SignInEndpoint(BaseAPIView):
# Create the user # Create the user
else: else:
ENABLE_SIGNUP = get_configuration_value( ENABLE_SIGNUP, = get_configuration_value(
[ [
{ {
"key": "ENABLE_SIGNUP", "key": "ENABLE_SIGNUP",
@ -269,16 +269,15 @@ class SignInEndpoint(BaseAPIView):
workspace_member_invites.delete() workspace_member_invites.delete()
project_member_invites.delete() project_member_invites.delete()
# Send event # Send event
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium="EMAIL",
medium="EMAIL", first_time=False,
first_time=False, )
)
access_token, refresh_token = get_tokens_for_user(user) access_token, refresh_token = get_tokens_for_user(user)
data = { data = {
@ -352,16 +351,15 @@ class MagicSignInEndpoint(BaseAPIView):
status=status.HTTP_403_FORBIDDEN, status=status.HTTP_403_FORBIDDEN,
) )
# Send event # Send event
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium="MAGIC_LINK",
medium="MAGIC_LINK", first_time=False,
first_time=False, )
)
user.is_active = True user.is_active = True
user.is_email_verified = True user.is_email_verified = True

View File

@ -89,16 +89,16 @@ class ReleaseNotesEndpoint(BaseAPIView):
class UnsplashEndpoint(BaseAPIView): class UnsplashEndpoint(BaseAPIView):
def get(self, request): def get(self, request):
UNSPLASH_ACESS_KEY = get_configuration_value( UNSPLASH_ACCESS_KEY, = get_configuration_value(
[ [
{ {
"key": "UNSPLASH_ACESS_KEY", "key": "UNSPLASH_ACCESS_KEY",
"default": os.environ.get("UNSPLASH_ACCESS_KEY"), "default": os.environ.get("UNSPLASH_ACCESS_KEY"),
} }
] ]
) )
# Check unsplash access key # Check unsplash access key
if not UNSPLASH_ACESS_KEY: if not UNSPLASH_ACCESS_KEY:
return Response([], status=status.HTTP_200_OK) return Response([], status=status.HTTP_200_OK)
# Query parameters # Query parameters
@ -107,9 +107,9 @@ class UnsplashEndpoint(BaseAPIView):
per_page = request.GET.get("per_page", 20) per_page = request.GET.get("per_page", 20)
url = ( url = (
f"https://api.unsplash.com/search/photos/?client_id={UNSPLASH_ACESS_KEY}&query={query}&page=${page}&per_page={per_page}" f"https://api.unsplash.com/search/photos/?client_id={UNSPLASH_ACCESS_KEY}&query={query}&page=${page}&per_page={per_page}"
if query if query
else f"https://api.unsplash.com/photos/?client_id={UNSPLASH_ACESS_KEY}&page={page}&per_page={per_page}" else f"https://api.unsplash.com/photos/?client_id={UNSPLASH_ACCESS_KEY}&page={page}&per_page={per_page}"
) )
headers = { headers = {

View File

@ -280,16 +280,15 @@ class OauthEndpoint(BaseAPIView):
) )
# Send event # Send event
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium=medium.upper(),
medium=medium.upper(), first_time=False,
first_time=False, )
)
access_token, refresh_token = get_tokens_for_user(user) access_token, refresh_token = get_tokens_for_user(user)
@ -300,7 +299,7 @@ class OauthEndpoint(BaseAPIView):
return Response(data, status=status.HTTP_200_OK) return Response(data, status=status.HTTP_200_OK)
except User.DoesNotExist: except User.DoesNotExist:
ENABLE_SIGNUP = get_configuration_value( ENABLE_SIGNUP, = get_configuration_value(
[ [
{ {
"key": "ENABLE_SIGNUP", "key": "ENABLE_SIGNUP",
@ -412,16 +411,15 @@ class OauthEndpoint(BaseAPIView):
project_member_invites.delete() project_member_invites.delete()
# Send event # Send event
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: auth_events.delay(
auth_events.delay( user=user.id,
user=user.id, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="SIGN_IN",
event_name="SIGN_IN", medium=medium.upper(),
medium=medium.upper(), first_time=True,
first_time=True, )
)
SocialLoginConnection.objects.update_or_create( SocialLoginConnection.objects.update_or_create(
medium=medium, medium=medium,

View File

@ -408,15 +408,14 @@ class WorkspaceJoinEndpoint(BaseAPIView):
workspace_invite.delete() workspace_invite.delete()
# Send event # Send event
if settings.POSTHOG_API_KEY and settings.POSTHOG_HOST: workspace_invite_event.delay(
workspace_invite_event.delay( user=user.id if user is not None else None,
user=user.id if user is not None else None, email=email,
email=email, user_agent=request.META.get("HTTP_USER_AGENT"),
user_agent=request.META.get("HTTP_USER_AGENT"), ip=request.META.get("REMOTE_ADDR"),
ip=request.META.get("REMOTE_ADDR"), event_name="MEMBER_ACCEPTED",
event_name="MEMBER_ACCEPTED", accepted_from="EMAIL",
accepted_from="EMAIL", )
)
return Response( return Response(
{"message": "Workspace Invitation Accepted"}, {"message": "Workspace Invitation Accepted"},

View File

@ -1,50 +1,78 @@
import uuid import uuid
import os
from posthog import Posthog # third party imports
from django.conf import settings
#third party imports
from celery import shared_task from celery import shared_task
from sentry_sdk import capture_exception from sentry_sdk import capture_exception
from posthog import Posthog
# module imports
from plane.license.utils.instance_value import get_configuration_value
def posthogConfiguration():
POSTHOG_API_KEY, POSTHOG_HOST = get_configuration_value(
[
{
"key": "POSTHOG_API_KEY",
"default": os.environ.get("POSTHOG_API_KEY", None),
},
{
"key": "POSTHOG_HOST",
"default": os.environ.get("POSTHOG_HOST", None),
},
]
)
if POSTHOG_API_KEY and POSTHOG_HOST:
return POSTHOG_API_KEY, POSTHOG_HOST
else:
return None, None
@shared_task @shared_task
def auth_events(user, email, user_agent, ip, event_name, medium, first_time): def auth_events(user, email, user_agent, ip, event_name, medium, first_time):
try: try:
posthog = Posthog(settings.POSTHOG_API_KEY, host=settings.POSTHOG_HOST) POSTHOG_API_KEY, POSTHOG_HOST = posthogConfiguration()
posthog.capture(
email, if POSTHOG_API_KEY and POSTHOG_HOST:
event=event_name, posthog = Posthog(POSTHOG_API_KEY, host=POSTHOG_HOST)
properties={ posthog.capture(
"event_id": uuid.uuid4().hex, email,
"user": {"email": email, "id": str(user)}, event=event_name,
"device_ctx": { properties={
"ip": ip, "event_id": uuid.uuid4().hex,
"user_agent": user_agent, "user": {"email": email, "id": str(user)},
}, "device_ctx": {
"medium": medium, "ip": ip,
"first_time": first_time "user_agent": user_agent,
} },
) "medium": medium,
"first_time": first_time
}
)
except Exception as e: except Exception as e:
capture_exception(e) capture_exception(e)
@shared_task @shared_task
def workspace_invite_event(user, email, user_agent, ip, event_name, accepted_from): def workspace_invite_event(user, email, user_agent, ip, event_name, accepted_from):
try: try:
posthog = Posthog(settings.POSTHOG_API_KEY, host=settings.POSTHOG_HOST) POSTHOG_API_KEY, POSTHOG_HOST = posthogConfiguration()
posthog.capture(
email, if POSTHOG_API_KEY and POSTHOG_HOST:
event=event_name, posthog = Posthog(POSTHOG_API_KEY, host=POSTHOG_HOST)
properties={ posthog.capture(
"event_id": uuid.uuid4().hex, email,
"user": {"email": email, "id": str(user)}, event=event_name,
"device_ctx": { properties={
"ip": ip, "event_id": uuid.uuid4().hex,
"user_agent": user_agent, "user": {"email": email, "id": str(user)},
}, "device_ctx": {
"accepted_from": accepted_from "ip": ip,
} "user_agent": user_agent,
) },
"accepted_from": accepted_from
}
)
except Exception as e: except Exception as e:
capture_exception(e) capture_exception(e)