mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 09:14:46 +01:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from typing import Any
|
|
|
|
import requests
|
|
from django.db import models
|
|
|
|
|
|
class LicenseError(Exception):
|
|
"""Exception raised for licensing errors.
|
|
|
|
Attributes:
|
|
code -- code of the exception
|
|
detail -- message of the exception
|
|
"""
|
|
|
|
def __init__(self, code, detail):
|
|
self.code = code
|
|
self.detail = detail
|
|
|
|
|
|
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"]
|
|
return self._create(*args, **kwargs)
|
|
|
|
def _create(self, *args: Any, **kwargs: Any) -> "License":
|
|
return super().create(*args, **kwargs)
|
|
|
|
|
|
class License(models.Model):
|
|
objects = 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)
|
|
|
|
ENTERPRISE_PLAN = "enterprise"
|
|
ENTERPRISE_FEATURES = ["zapier", "organizations_projects"]
|
|
PLANS = {
|
|
ENTERPRISE_PLAN: ENTERPRISE_FEATURES,
|
|
}
|