mirror of
https://github.com/PostHog/posthog.git
synced 2024-12-01 04:12:23 +01:00
7546e3ea0a
* add description and tag fields to event and property definitions * set up description and tagging on models * frontend functionality for description editing * connect backend and kea logic for description editing * update event and property definitions model and migration * delete set null instead of cascade * migration merge fix * add owner column * undo posthog event property taxonomy migrations * set up definitions on enterprise level * allow enterprise definitions description editing * fix licensing conditions and add tests * proper get and update methods for the multi inheritance table and new column fields for enterprise event model * check for license to separate routes * migrate existing definitions to ee definitions tables and render ee vs non-ee definition views based on existing feature conditional * all the working backend updates * updated tests * frontend fixes and linting updates * feature flag it
96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
from typing import Any, List, Optional, cast
|
|
|
|
import requests
|
|
from django.contrib.auth import get_user_model
|
|
from django.db import models
|
|
from django.db.models.signals import post_save
|
|
from django.dispatch.dispatcher import receiver
|
|
from django.utils import timezone
|
|
from rest_framework import exceptions, status
|
|
|
|
from posthog.celery import sync_all_organization_available_features
|
|
|
|
|
|
class LicenseError(exceptions.APIException):
|
|
"""
|
|
Exception raised for licensing errors.
|
|
"""
|
|
|
|
default_type = "license_error"
|
|
default_code = "license_error"
|
|
status_code = status.HTTP_400_BAD_REQUEST
|
|
default_detail = "There was a problem with your current license."
|
|
|
|
def __init__(self, code, detail):
|
|
self.code = code
|
|
self.detail = exceptions._get_error_details(detail, code)
|
|
|
|
|
|
class LicenseManager(models.Manager):
|
|
def create(self, *args: Any, **kwargs: Any) -> "License":
|
|
validate = requests.post("https://license.posthog.com/licenses/activate", data={"key": kwargs["key"]})
|
|
resp = validate.json()
|
|
if not validate.ok:
|
|
raise LicenseError(resp["code"], resp["detail"])
|
|
|
|
kwargs["valid_until"] = resp["valid_until"]
|
|
kwargs["plan"] = resp["plan"]
|
|
kwargs["max_users"] = resp.get("max_users", 0)
|
|
return cast(License, super().create(*args, **kwargs))
|
|
|
|
def first_valid(self) -> Optional["License"]:
|
|
return cast(Optional[License], (self.filter(valid_until__gte=timezone.now()).first()))
|
|
|
|
|
|
class License(models.Model):
|
|
objects: LicenseManager = LicenseManager()
|
|
|
|
created_at: models.DateTimeField = models.DateTimeField(auto_now_add=True)
|
|
plan: models.CharField = models.CharField(max_length=200)
|
|
valid_until: models.DateTimeField = models.DateTimeField()
|
|
key: models.CharField = models.CharField(max_length=200)
|
|
max_users: models.IntegerField = models.IntegerField(default=None, null=True) # None = no restriction
|
|
|
|
ENTERPRISE_PLAN = "enterprise"
|
|
FREE_CLICKHOUSE_PLAN = "free_clickhouse"
|
|
ENTERPRISE_FEATURES = [
|
|
"zapier",
|
|
"organizations_projects",
|
|
"google_login",
|
|
"dashboard_collaboration",
|
|
"ingestion_taxonomy",
|
|
] # Base premium features
|
|
PLANS = {
|
|
ENTERPRISE_PLAN: ENTERPRISE_FEATURES + ["clickhouse"],
|
|
FREE_CLICKHOUSE_PLAN: ["clickhouse"],
|
|
}
|
|
|
|
@property
|
|
def available_features(self) -> List[str]:
|
|
return self.PLANS.get(self.plan, [])
|
|
|
|
|
|
def get_licensed_users_available() -> Optional[int]:
|
|
"""
|
|
Returns the number of user slots available that can be created based on the instance's current license.
|
|
Not relevant for cloud users.
|
|
`None` means unlimited users.
|
|
"""
|
|
|
|
license = License.objects.first_valid()
|
|
from posthog.models import OrganizationInvite
|
|
|
|
if license:
|
|
if license.max_users is None:
|
|
return None
|
|
|
|
users_left = license.max_users - get_user_model().objects.count() - OrganizationInvite.objects.count()
|
|
return max(users_left, 0)
|
|
|
|
return None
|
|
|
|
|
|
@receiver(post_save, sender=License)
|
|
def license_saved(sender, instance, created, raw, using, **kwargs):
|
|
sync_all_organization_available_features()
|