initial commit

zvml/zvml.py

@@ -0,0 +1,109 @@
+# Legal Disclaimer
+# This script is an example script and is not supported under any Zerto support program or service. 
+# The author and Zerto further disclaim all implied warranties including, without limitation, 
+# any implied warranties of merchantability or of fitness for a particular purpose.
+# In no event shall Zerto, its authors or anyone else involved in the creation, 
+# production or delivery of the scripts be liable for any damages whatsoever (including, 
+# without limitation, damages for loss of business profits, business interruption, loss of business 
+# information, or other pecuniary loss) arising out of the use of or the inability to use the sample 
+# scripts or documentation, even if the author or Zerto has been advised of the possibility of such damages. 
+# The entire risk arising out of the use or performance of the sample scripts and documentation remains with you.
+
+import requests
+import logging
+import ssl
+
+# Configure logging with timestamp format
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S'
+)
+
+# Import all necessary classes
+from .tasks import Tasks
+from .vpgs import VPGs
+# from .vpg_settings import VPGSettings
+from .vms import VMs
+from .failover import Failover
+from .alerts import Alerts
+from .peersites import PeerSites
+from .events import Events
+from .repositories import Repositories
+from .sessions import Sessions
+from .recoveryscripts import RecoveryScripts
+from .encryptiondetection import EncryptionDetection
+from .zorgs import Zorgs
+from .localsite import LocalSite
+from .datastores import Datastores
+from .vras import VRA
+from .recovery_reports import RecoveryReports
+from .license import License
+from .service_profiles import ServiceProfiles
+from .server_date_time import ServerDateTime
+from .virtualization_sites import VirtualizationSites
+from .volumes import Volumes
+from .tweaks import Tweaks
+# Disable SSL warnings for self-signed certificates
+context = ssl._create_unverified_context()
+
+class ZVMLClient:
+    def __init__(self, zvm_address, client_id, client_secret, verify_certificate=True):
+        self.zvm_address = zvm_address
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.verify_certificate = verify_certificate
+        self.token = None
+        self.token_expiry = None
+        self.__get_keycloak_token()
+        self.tasks = Tasks(self)
+        self.vpgs = VPGs(self)
+        # self.vpg_settings = VPGSettings(self)
+        self.vms = VMs(self)
+        self.failover = Failover(self)
+        self.alerts = Alerts(self)
+        self.peersites = PeerSites(self)
+        self.events = Events(self)
+        self.repositories = Repositories(self)
+        self.sessions = Sessions(self)
+        self.recoveryscripts = RecoveryScripts(self)
+        self.zorgs = Zorgs(self)
+        self.encryptiondetection = EncryptionDetection(self)
+        self.localsite = LocalSite(self.zvm_address, self.token)
+        self.datastores = Datastores(self)
+        self.vras = VRA(self)
+        self.recovery_reports = RecoveryReports(self)
+        self.license = License(self)
+        self.service_profiles = ServiceProfiles(self)
+        self.server_date_time = ServerDateTime(self)
+        self.virtualization_sites = VirtualizationSites(self)
+        self.volumes = Volumes(self)
+        self.tweaks = Tweaks(self)
+    def __get_keycloak_token(self):
+        logging.debug(f'__get_keycloak_token(zvm_address={self.zvm_address})')
+        keycloak_uri = f"https://{self.zvm_address}/auth/realms/zerto/protocol/openid-connect/token"
+        headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+        body = {
+            'client_id': self.client_id,
+            'client_secret': self.client_secret,
+            'grant_type': 'client_credentials',
+            'expires_in': 3600  # Request token expiration in seconds (e.g., 1 hour)
+        }
+
+        try:
+            logging.info("Connecting to Keycloak to get token...")
+            response = requests.post(keycloak_uri, headers=headers, data=body, verify=self.verify_certificate)
+            response.raise_for_status()
+            token_data = response.json()
+            self.token = token_data.get('access_token')
+            self.token_expiry = token_data.get('expires_in')  # Store expiration time
+            logging.info(f"Successfully retrieved token.")
+            logging.info(f"Token expiration details:")
+            logging.info(f"- Expires in: {self.token_expiry} seconds")
+            logging.info(f"- Requested expiration: {body['expires_in']} seconds")
+            if self.token_expiry != body['expires_in']:
+                logging.warning(f"Server provided different expiration time than requested!")
+            return self.token
+        except requests.exceptions.RequestException as e:
+            logging.error(f"Error retrieving token: {e}")
+            raise