From 6390a6f92f3804ce2585cceae0d26b77c44654d4 Mon Sep 17 00:00:00 2001 From: Benjamin Dauvergne Date: Mon, 20 Mar 2017 12:18:33 +0100 Subject: [PATCH] add a delete_tenant command --- .../management/commands/delete_tenant.py | 24 ++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 hobo/multitenant/management/commands/delete_tenant.py diff --git a/hobo/multitenant/management/commands/delete_tenant.py b/hobo/multitenant/management/commands/delete_tenant.py new file mode 100644 index 0000000..ad05e39 --- /dev/null +++ b/hobo/multitenant/management/commands/delete_tenant.py @@ -0,0 +1,24 @@ +import sys + +from django.core.management.base import CommandError, BaseCommand + +from hobo.multitenant.middleware import TenantMiddleware + + +class Command(BaseCommand): + help = "Delete tenant(s) by hostname(s)" + args = ['...'] + + def handle(self, *args, **options): + if not args: + raise CommandError("you must give at least one tenant hostname") + + # if - is given on the command line, get list of hostnames from stdin + if '-' in args: + args = list(args) + args.remove('-') + args.extend([x.strip() for x in sys.stdin.readlines()]) + + for hostname in args: + tenant = TenantMiddleware.get_tenant_by_hostname(hostname) + tenant.delete() -- 2.1.4