Merge branch 'feature/all_customers_mgmt_command' into 'master'
Feature/all customers mgmt command See merge request ungleich-public/dynamicweb!723
This commit is contained in:
commit
146e9faf53
1 changed files with 41 additions and 0 deletions
41
datacenterlight/management/commands/all_customers.py
Normal file
41
datacenterlight/management/commands/all_customers.py
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from django.core.management.base import BaseCommand
|
||||||
|
|
||||||
|
from hosting.models import (
|
||||||
|
HostingOrder, VMDetail
|
||||||
|
)
|
||||||
|
from membership.models import CustomUser
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
help = '''Dumps the email addresses of all customers who have a VM'''
|
||||||
|
|
||||||
|
def add_arguments(self, parser):
|
||||||
|
parser.add_argument('-a', '--all_registered', action='store_true',
|
||||||
|
help='All registered users')
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
all_registered = options['all_registered']
|
||||||
|
all_customers_set = set()
|
||||||
|
if all_registered:
|
||||||
|
all_customers = CustomUser.objects.filter(
|
||||||
|
is_admin=False, validated=True
|
||||||
|
)
|
||||||
|
for customer in all_customers:
|
||||||
|
all_customers_set.add(customer.email)
|
||||||
|
else:
|
||||||
|
all_hosting_orders = HostingOrder.objects.filter()
|
||||||
|
running_vm_details = VMDetail.objects.filter(terminated_at=None)
|
||||||
|
running_vm_ids = [rvm.vm_id for rvm in running_vm_details]
|
||||||
|
for order in all_hosting_orders:
|
||||||
|
if order.vm_id in running_vm_ids:
|
||||||
|
all_customers_set.add(order.customer.user.email)
|
||||||
|
for cu in all_customers_set:
|
||||||
|
print(cu)
|
||||||
|
if all_registered:
|
||||||
|
print("All registered users = %s" % len(all_customers_set))
|
||||||
|
else:
|
||||||
|
print("Total active customers = %s" % len(all_customers_set))
|
Loading…
Reference in a new issue