http client class

capsulflask/spoke_model.py

@@ -8,34 +8,13 @@ from subprocess import run
 
 from capsulflask.db import get_model
 
+from capsulflask.hub_model import VirtualizationInterface, VirtualMachine
+
 def validate_capsul_id(id):
   if not re.match(r"^(cvm|capsul)-[a-z0-9]{10}$", id):
     raise ValueError(f"vm id \"{id}\" must match \"^capsul-[a-z0-9]{{10}}$\"")
 
-class VirtualMachine:
-  def __init__(self, id, host, ipv4=None, ipv6=None):
-    self.id = id
-    self.host = host
-    self.ipv4 = ipv4
-    self.ipv6 = ipv6
-
-class SpokeInterface:
-  def capacity_avaliable(self, additional_ram_bytes: int) -> bool:
-    pass
-
-  def get(self, id: str) -> VirtualMachine:
-    pass
-
-  def list_ids(self) -> list:
-    pass
-
-  def create(self, email: str, id: str, template_image_file_name: str, vcpus: int, memory: int, ssh_public_keys: list):
-    pass
-
-  def destroy(self, email: str, id: str):
-    pass
-
-class MockSpoke(SpokeInterface):
+class MockSpoke(VirtualizationInterface):
   def capacity_avaliable(self, additional_ram_bytes):
     return True
 
@@ -54,7 +33,7 @@ class MockSpoke(SpokeInterface):
   def destroy(self, email: str, id: str):
     current_app.logger.info(f"mock destroy: {id} for {email}")
 
-class ShellScriptSpoke(SpokeInterface):
+class ShellScriptSpoke(VirtualizationInterface):
 
   def validate_completed_process(self, completedProcess, email=None):
     emailPart = ""