forked from github/plane
103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
# Python imports
|
|
import os, sys
|
|
import json
|
|
import uuid
|
|
import requests
|
|
|
|
# Django imports
|
|
from django.utils import timezone
|
|
|
|
|
|
sys.path.append("/code")
|
|
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "plane.settings.production")
|
|
|
|
import django
|
|
|
|
django.setup()
|
|
|
|
|
|
def instance_registration():
|
|
try:
|
|
# Module imports
|
|
from plane.db.models import User
|
|
from plane.license.models import Instance, InstanceAdmin
|
|
|
|
# Check if the instance is registered
|
|
instance = Instance.objects.first()
|
|
|
|
# If instance is None then register this instance
|
|
if instance is None:
|
|
with open("/code/package.json", "r") as file:
|
|
# Load JSON content from the file
|
|
data = json.load(file)
|
|
|
|
admin_email = os.environ.get("ADMIN_EMAIL")
|
|
# Raise an exception if the admin email is not provided
|
|
if not admin_email:
|
|
raise Exception("ADMIN_EMAIL is required")
|
|
|
|
# Check if the admin email user exists
|
|
user = User.objects.filter(email=admin_email).first()
|
|
|
|
# If the user does not exist create the user and add him to the database
|
|
if user is None:
|
|
user = User.objects.create(email=admin_email, username=uuid.uuid4().hex)
|
|
user.set_password(uuid.uuid4().hex)
|
|
user.save()
|
|
|
|
license_engine_base_url = os.environ.get("LICENSE_ENGINE_BASE_URL")
|
|
|
|
if not license_engine_base_url:
|
|
raise Exception("LICENSE_ENGINE_BASE_URL is required")
|
|
|
|
headers = {"Content-Type": "application/json"}
|
|
|
|
payload = {
|
|
"email": user.email,
|
|
"version": data.get("version", 0.1),
|
|
}
|
|
|
|
response = requests.post(
|
|
f"{license_engine_base_url}/api/instances",
|
|
headers=headers,
|
|
data=json.dumps(payload),
|
|
)
|
|
|
|
if response.status_code == 201:
|
|
data = response.json()
|
|
# Create instance
|
|
instance = Instance.objects.create(
|
|
instance_name="Plane Free",
|
|
instance_id=data.get("id"),
|
|
license_key=data.get("license_key"),
|
|
api_key=data.get("api_key"),
|
|
version=data.get("version"),
|
|
primary_email=data.get("email"),
|
|
primary_owner=user,
|
|
last_checked_at=timezone.now(),
|
|
)
|
|
# Create instance admin
|
|
_ = InstanceAdmin.objects.create(
|
|
user=user,
|
|
instance=instance,
|
|
role=20,
|
|
)
|
|
|
|
print(f"Instance succesfully registered with owner: {instance.owner.email}")
|
|
return
|
|
|
|
print("Instance could not be registered")
|
|
return
|
|
else:
|
|
print(
|
|
f"Instance already registered with instance owner: {instance.owner.email}"
|
|
)
|
|
return
|
|
except ImportError:
|
|
raise ImportError
|
|
|
|
|
|
if __name__ == "__main__":
|
|
instance_registration()
|