95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
# Django imports
|
|
from typing import Any
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
# Module imports
|
|
from plane.db.models import User, Workspace, WorkspaceMember
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Create dump issues, cycles etc. for a project in a given workspace"
|
|
|
|
def handle(self, *args: Any, **options: Any) -> str | None:
|
|
|
|
try:
|
|
workspace_name = input("Workspace Name: ")
|
|
workspace_slug = input("Workspace slug: ")
|
|
|
|
if workspace_slug == "":
|
|
raise CommandError("Workspace slug is required")
|
|
|
|
if Workspace.objects.filter(slug=workspace_slug).exists():
|
|
raise CommandError("Workspace already exists")
|
|
|
|
creator = input("Your email: ")
|
|
|
|
if (
|
|
creator == ""
|
|
or not User.objects.filter(email=creator).exists()
|
|
):
|
|
raise CommandError(
|
|
"User email is required and should have signed in plane"
|
|
)
|
|
|
|
user = User.objects.get(email=creator)
|
|
|
|
members = input("Enter Member emails (comma separated): ")
|
|
members = members.split(",") if members != "" else []
|
|
# Create workspace
|
|
workspace = Workspace.objects.create(
|
|
slug=workspace_slug,
|
|
name=workspace_name,
|
|
owner=user,
|
|
)
|
|
# Create workspace member
|
|
WorkspaceMember.objects.create(
|
|
workspace=workspace, role=20, member=user
|
|
)
|
|
user_ids = User.objects.filter(email__in=members)
|
|
|
|
_ = WorkspaceMember.objects.bulk_create(
|
|
[
|
|
WorkspaceMember(
|
|
workspace=workspace,
|
|
member=user_id,
|
|
role=20,
|
|
)
|
|
for user_id in user_ids
|
|
],
|
|
ignore_conflicts=True,
|
|
)
|
|
|
|
project_count = int(input("Number of projects to be created: "))
|
|
|
|
for i in range(project_count):
|
|
print(f"Please provide the following details for project {i+1}:")
|
|
issue_count = int(input("Number of issues to be created: "))
|
|
cycle_count = int(input("Number of cycles to be created: "))
|
|
module_count = int(input("Number of modules to be created: "))
|
|
pages_count = int(input("Number of pages to be created: "))
|
|
inbox_issue_count = int(
|
|
input("Number of inbox issues to be created: ")
|
|
)
|
|
|
|
from plane.bgtasks.dummy_data_task import create_dummy_data
|
|
|
|
create_dummy_data.delay(
|
|
slug=workspace_slug,
|
|
email=creator,
|
|
members=members,
|
|
issue_count=issue_count,
|
|
cycle_count=cycle_count,
|
|
module_count=module_count,
|
|
pages_count=pages_count,
|
|
inbox_issue_count=inbox_issue_count,
|
|
)
|
|
|
|
self.stdout.write(
|
|
self.style.SUCCESS("Data is pushed to the queue")
|
|
)
|
|
return
|
|
except Exception as e:
|
|
self.stdout.write(
|
|
self.style.ERROR(f"Command errored out {str(e)}")
|
|
)
|
|
return
|