# -*- 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 piped via stdin so it is # never placed in cldetect's argv / /proc//cmdline; cldetect and # cllicense ship in the same package, so the "-" convention is in lockstep) with subprocess.Popen( [self._CLDETECT_UTILITY, '--update-new-key', '-'], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE ) as proc: proc.communicate(input=(key + '\n').encode()) 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()