From cf15cd695cba9dce76d70d77b14d155ec6dc4d3d Mon Sep 17 00:00:00 2001
From: Varac <varac@varac.net>
Date: Thu, 25 Mar 2021 11:29:35 +0100
Subject: [PATCH] Configure VCPUS

Fixes: #5
---
 greenhost_cloud/cosmos.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/greenhost_cloud/cosmos.py b/greenhost_cloud/cosmos.py
index dca9e23..ed0ab98 100755
--- a/greenhost_cloud/cosmos.py
+++ b/greenhost_cloud/cosmos.py
@@ -97,6 +97,7 @@ def create_droplet(name: str, # pylint: disable=too-many-arguments
                    ssh_key_id: int,
                    region: str = 'ams1',
                    size: int = 2048,
+                   vcpus: int = 1,
                    disk: int = 20,
                    image: int = 18):
     """Create a droplet.
@@ -109,7 +110,11 @@ def create_droplet(name: str, # pylint: disable=too-many-arguments
       - image (str): 18 (Ubuntu 18.04 x64)
       - region (str): 'ams1' (Amsterdam 1)
       - size (int): 2048 (2GB RAM)
+      - vcpus (int): 1
       - disk (int): 20 (20GB disk space)
+
+    Droplet digital ocean API doc:
+      https://developers.digitalocean.com/documentation/v2/#droplets
     """
     log.info('Creating droplet')
 
@@ -117,6 +122,7 @@ def create_droplet(name: str, # pylint: disable=too-many-arguments
         "name": name,
         "region": region,
         "size": size,
+        "vcpus": vcpus,
         "disk": disk,
         "image": image,
         "ssh_keys": ssh_key_id
-- 
GitLab