From 56b45dc381f7f8e4df3cce1237397363c17a447f Mon Sep 17 00:00:00 2001
From: Varac <varac@varac.net>
Date: Wed, 30 Jan 2019 15:08:11 +0100
Subject: [PATCH] Add list_domain_records(), also tablulate list_droplets()
---
test/cosmos.py | 30 +++++++++++++++++++++++++++---
1 file changed, 27 insertions(+), 3 deletions(-)
diff --git a/test/cosmos.py b/test/cosmos.py
index 774ce824a..43a724cf6 100755
--- a/test/cosmos.py
+++ b/test/cosmos.py
@@ -6,6 +6,7 @@ import os
import re
import requests
import socket
+from tabulate import tabulate
from time import sleep
@@ -113,15 +114,38 @@ def get_droplet(id: int):
return response['droplet']
+def list_domain_records(domain: str, verbose: bool = False):
+ """List domain records for given domain."""
+ records = get_domain_records(domain)
+
+ if verbose:
+ print(json.dumps(records, sort_keys=True, indent=2))
+
+ table_records = []
+ for record in records:
+ table_records.append([
+ record['id'],
+ record['name'],
+ record['type'],
+ record['data']])
+ print(tabulate(table_records, headers=['ID', 'Name', 'Type', 'Data']))
+
+
def list_droplets(verbose: bool = False):
"""List all droplets by their ID, Name, IP and state."""
droplets = get_droplets()
+
if verbose:
print(json.dumps(droplets, sort_keys=True, indent=2))
+
+ table_records = []
for droplet in droplets:
- print(droplet['id'], ' ', droplet['name'], ' ',
- droplet['networks']['v4'][0]['ip_address'],
- ' ', droplet['status'])
+ table_records.append([
+ droplet['id'],
+ droplet['name'],
+ droplet['networks']['v4'][0]['ip_address'],
+ droplet['status']])
+ print(tabulate(table_records, headers=['ID', 'Name', 'IPv4', 'Status']))
def shutdown_droplet(id: int):
--
GitLab