# -*- coding: utf-8 -*-
# license_lib.py
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
import subprocess
import os
class CloudlinuxLicenseLib:
def __init__(self):
self._CLDETECT_UTILITY = "/usr/bin/cldetect"
self._CLNREG_KS_UTILITY = '/usr/sbin/clnreg_ks'
self._CLNCHECK_UTILITY = '/usr/sbin/clncheck'
def get_license_status(self):
"""
Checks Cloudlinux license
:return: True - license valid
False - invalid
"""
# cldetect --check-license -q
with subprocess.Popen(
[self._CLDETECT_UTILITY, '--check-license', '-q'],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
) as proc:
proc.communicate()
return proc.returncode == 0
def is_edition_migration_available(self):
# check if edition migration is supported
return os.path.isfile(self._CLNCHECK_UTILITY)
def register_and_update_license_by_key(self, key):
"""
Register and update Cloudlinux license by key
:param key: License key
:return:
"""
# /usr/bin/cldetect --update-new-key key
with subprocess.Popen(
[self._CLDETECT_UTILITY, '--update-new-key', key],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
) as proc:
proc.communicate()
def register_and_update_license_by_ip(self, stage=None):
"""
Register and update Cloudlinux license by IP
:param stage: Registraton stage.
None - all stages. Normal mode
'clnreg_ks' - only clnreg_ks. Only for tests
'cldetect' - only cldetect. Only for tests
:return:
"""
# 1. /usr/sbin/clnreg_ks --force or /usr/sbin/clnreg_ks --force --migrate-silently
if stage is None or stage == 'clnreg_ks':
clnreg_cmd = [self._CLNREG_KS_UTILITY, '--force']
if self.is_edition_migration_available():
clnreg_cmd.append('--migrate-silently')
with subprocess.Popen(
clnreg_cmd,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
) as proc:
proc.communicate()
# 2. /usr/bin/cldetect --update-license
if stage is None or stage == 'cldetect':
with subprocess.Popen(
[self._CLDETECT_UTILITY, '--update-license'],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
) as proc:
proc.communicate()
|