Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
58 changes: 58 additions & 0 deletions src/sentry/runner/commands/createorg.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
from __future__ import annotations

import click

from sentry.runner.decorators import configuration


@click.command()
@click.option("--name", required=True, help="Organization name.")
@click.option(
"--slug",
default=None,
help="URL-friendly slug. Derived from name if omitted.",
)
@click.option("--owner-email", required=True, help="Email of the organization owner.")
@click.option(
"--no-default-team",
default=False,
is_flag=True,
help="Skip creating a default team.",
)
@configuration

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

omg, TIL about this decorator, all my commands call sentry.runner.configure() as the import/line, this is helpful

def createorg(
name: str,
slug: str | None,
owner_email: str,
no_default_team: bool,
) -> None:
"Create a new organization."

from sentry.services.organization.model import (
OrganizationOptions,
OrganizationProvisioningOptions,
PostProvisionOptions,
)
from sentry.services.organization.provisioning import (
OrganizationProvisioningException,
organization_provisioning_service,
)

provision_args = OrganizationProvisioningOptions(
provision_options=OrganizationOptions(
name=name,
slug=slug or name,
owning_email=owner_email,
create_default_team=not no_default_team,
),
post_provision_options=PostProvisionOptions(),
)

try:
rpc_org = organization_provisioning_service.provision_organization_in_cell(
provisioning_options=provision_args,
)
except OrganizationProvisioningException as e:
raise click.ClickException(str(e))

click.echo(f"Organization created: {rpc_org.name} (slug: {rpc_org.slug}, id: {rpc_org.id})")
1 change: 1 addition & 0 deletions src/sentry/runner/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ def cli(config: str) -> None:
"sentry.runner.commands.configoptions.configoptions",
"sentry.runner.commands.createflag.createflag",
"sentry.runner.commands.createflag.createissueflag",
"sentry.runner.commands.createorg.createorg",
"sentry.runner.commands.createuser.createuser",
"sentry.runner.commands.devserver.devserver",
"sentry.runner.commands.django.django",
Expand Down
47 changes: 47 additions & 0 deletions tests/sentry/runner/commands/test_createorg.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
from sentry.models.organization import Organization
from sentry.models.organizationmember import OrganizationMember
from sentry.models.team import Team
from sentry.runner.commands.createorg import createorg
from sentry.testutils.cases import CliTestCase


class CreateOrgTest(CliTestCase):
command = createorg

def test_basic_create(self) -> None:
rv = self.invoke("--name=Test Org", "--slug=test-org", "--owner-email=owner@example.com")
assert rv.exit_code == 0, rv.output
assert "test-org" in rv.output

org = Organization.objects.get(slug="test-org")
assert org.name == "Test Org"
assert OrganizationMember.objects.filter(
organization=org, email="owner@example.com"
).exists()
assert Team.objects.filter(organization=org).exists()

def test_slug_defaults_to_name(self) -> None:
rv = self.invoke("--name=My Organization", "--owner-email=owner@example.com")
assert rv.exit_code == 0, rv.output

assert Organization.objects.filter(slug="my-organization").exists()

def test_no_default_team(self) -> None:
rv = self.invoke(
"--name=No Team Org",
"--slug=no-team-org",
"--owner-email=owner@example.com",
"--no-default-team",
)
assert rv.exit_code == 0, rv.output

org = Organization.objects.get(slug="no-team-org")
assert not Team.objects.filter(organization=org).exists()

def test_missing_name(self) -> None:
rv = self.invoke("--slug=test-org", "--owner-email=owner@example.com")
assert rv.exit_code != 0

def test_missing_owner_email(self) -> None:
rv = self.invoke("--name=Test Org", "--slug=test-org")
assert rv.exit_code != 0
Loading