153 lines
5.0 KiB
Python
153 lines
5.0 KiB
Python
import datetime
|
|
import sys
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.core.management.base import BaseCommand
|
|
from django.utils import timezone
|
|
|
|
from uncloud_pay.models import (
|
|
BillingAddress
|
|
)
|
|
from uncloud_vm.models import (
|
|
VMDiskType, VMProduct
|
|
)
|
|
|
|
|
|
def vm_price_2020(cpu=1, ram=2, v6only=False):
|
|
if v6only:
|
|
discount = 9
|
|
else:
|
|
discount = 0
|
|
|
|
return cpu * 3 + ram * 4 - discount
|
|
|
|
|
|
def disk_price_2020(size_in_gb, disk_type):
|
|
if disk_type == VMDiskType.CEPH_SSD:
|
|
price = 3.5 / 10
|
|
elif disk_type == VMDiskType.CEPH_HDD:
|
|
price = 1.5 / 100
|
|
else:
|
|
raise Exception("not yet defined price")
|
|
|
|
return size_in_gb * price
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Adding VMs / creating orders for user'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--username', type=str, required=True)
|
|
|
|
def handle(self, *args, **options):
|
|
user = get_user_model().objects.get(username=options['username'])
|
|
|
|
addr, created = BillingAddress.objects.get_or_create(
|
|
owner=user,
|
|
active=True,
|
|
defaults={'organization': 'Undefined organisation',
|
|
'full_name': 'Undefined name',
|
|
'street': 'Undefined Street',
|
|
'city': 'Undefined city',
|
|
'postal_code': '8750',
|
|
'country': 'CH',
|
|
'active': True
|
|
}
|
|
)
|
|
|
|
# 25206 + SSD
|
|
vm25206 = VMProduct.objects.create(name="one-25206", cores=1,
|
|
ram_in_gb=4, owner=user)
|
|
vm25206.create_order_at(
|
|
timezone.make_aware(datetime.datetime(2020, 3, 3)))
|
|
|
|
# vm25206_ssd = VMDiskProduct.objects.create(vm=vm25206, owner=user, size_in_gb=30)
|
|
# vm25206_ssd.create_order_at(timezone.make_aware(datetime.datetime(2020,3,3)))
|
|
|
|
# change 1
|
|
vm25206.cores = 2
|
|
vm25206.ram_in_gb = 8
|
|
vm25206.save()
|
|
vm25206.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 4, 17)))
|
|
|
|
sys.exit(0)
|
|
|
|
# change 2
|
|
# vm25206_ssd.size_in_gb = 50
|
|
# vm25206_ssd.save()
|
|
# vm25206_ssd.create_or_update_order(when_to_start=timezone.make_aware(datetime.datetime(2020,8,5)))
|
|
|
|
# 25206 done.
|
|
|
|
# 25615
|
|
vm25615 = VMProduct.objects.create(name="one-25615", cores=1,
|
|
ram_in_gb=4, owner=user)
|
|
vm25615.create_order_at(
|
|
timezone.make_aware(datetime.datetime(2020, 3, 3)))
|
|
|
|
# Change 2020-04-17
|
|
vm25615.cores = 2
|
|
vm25615.ram_in_gb = 8
|
|
vm25615.save()
|
|
vm25615.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 4, 17)))
|
|
|
|
# vm25615_ssd = VMDiskProduct(vm=vm25615, owner=user, size_in_gb=30)
|
|
# vm25615_ssd.create_order_at(timezone.make_aware(datetime.datetime(2020,3,3)))
|
|
# vm25615_ssd.save()
|
|
|
|
vm25208 = VMProduct.objects.create(name="one-25208", cores=1,
|
|
ram_in_gb=4, owner=user)
|
|
vm25208.create_order_at(
|
|
timezone.make_aware(datetime.datetime(2020, 3, 5)))
|
|
|
|
vm25208.cores = 2
|
|
vm25208.ram_in_gb = 8
|
|
vm25208.save()
|
|
vm25208.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 4, 17)))
|
|
|
|
Bill.create_next_bills_for_user(user, ending_date=end_of_month(
|
|
timezone.make_aware(datetime.datetime(2020, 7, 31))))
|
|
|
|
sys.exit(0)
|
|
|
|
vm25615_ssd.size_in_gb = 50
|
|
vm25615_ssd.save()
|
|
vm25615_ssd.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 8, 5)))
|
|
|
|
vm25208_ssd = VMDiskProduct.objects.create(vm=vm25208,
|
|
owner=user,
|
|
size_in_gb=30)
|
|
|
|
vm25208_ssd.size_in_gb = 50
|
|
vm25208_ssd.save()
|
|
vm25208_ssd.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 8, 5)))
|
|
|
|
# 25207
|
|
vm25207 = VMProduct.objects.create(name="OpenNebula 25207",
|
|
cores=1,
|
|
ram_in_gb=4,
|
|
owner=user)
|
|
|
|
vm25207_ssd = VMDiskProduct.objects.create(vm=vm25207,
|
|
owner=user,
|
|
size_in_gb=30)
|
|
|
|
vm25207_ssd.size_in_gb = 50
|
|
vm25207_ssd.save()
|
|
vm25207_ssd.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 8, 5)))
|
|
|
|
vm25207.cores = 2
|
|
vm25207.ram_in_gb = 8
|
|
vm25207.save()
|
|
vm25207.create_or_update_order(
|
|
when_to_start=timezone.make_aware(datetime.datetime(2020, 6, 19)))
|
|
|
|
# FIXES: check starting times (they are slightly different)
|
|
# add vm 25236
|