121 lines
3.2 KiB
Python
121 lines
3.2 KiB
Python
import subprocess
|
|
import inspect
|
|
from dataclasses import dataclass
|
|
from abc import ABC
|
|
from enum import Enum
|
|
|
|
|
|
class ResultType(Enum):
|
|
success = 0
|
|
failure = 1
|
|
|
|
|
|
def clone(repo):
|
|
command = f"git clone {repo}"
|
|
try:
|
|
subprocess.check_output(command.split())
|
|
except subprocess.CalledProcessError as e:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
|
|
def clone_common():
|
|
return clone("https://code.ungleich.ch/ungleich-public/ucloud_common")
|
|
|
|
|
|
def clone_etcd_wrapper():
|
|
return clone("https://code.ungleich.ch/ahmedbilal/etcd3_wrapper")
|
|
|
|
|
|
class Result(object):
|
|
def __init__(self, _result, _type: ResultType, _op=""):
|
|
self._type = _type
|
|
self._result = str(_result)
|
|
self._op = _op
|
|
if self._type == ResultType.failure:
|
|
print(self._op, "failed")
|
|
|
|
def add(self, operation, **kwargs):
|
|
if self._type == ResultType.success:
|
|
r = operation(**kwargs)
|
|
self._type = r._type
|
|
self._result = r._result
|
|
return self
|
|
else:
|
|
print("Dependency not satisfied")
|
|
exit(-1)
|
|
|
|
def __repr__(self):
|
|
return f"{self._type}, {self._result}"
|
|
|
|
|
|
class Operation(ABC):
|
|
pass
|
|
|
|
|
|
class GitOperation(object):
|
|
@staticmethod
|
|
def clone(url, path="."):
|
|
command = f"git clone {url}"
|
|
try:
|
|
output = subprocess.check_output(command.split(), cwd=path)
|
|
except subprocess.CalledProcessError as e:
|
|
return Result(e, ResultType.failure, inspect.currentframe().f_code.co_name)
|
|
else:
|
|
return Result(output, ResultType.success)
|
|
|
|
|
|
class PipenvOperation(object):
|
|
@staticmethod
|
|
def install(path=".", package_name=None):
|
|
if package_name:
|
|
command = f"pipenv install {package_name}"
|
|
else:
|
|
command = f"pipenv install"
|
|
try:
|
|
output = subprocess.check_output(command.split(), cwd=path)
|
|
except subprocess.CalledProcessError as e:
|
|
return Result(e, ResultType.failure, inspect.currentframe().f_code.co_name)
|
|
else:
|
|
return Result(output, ResultType.success)
|
|
|
|
|
|
class FileOperation(object):
|
|
@staticmethod
|
|
def write(path, content, mode="w"):
|
|
try:
|
|
with open(path, mode) as file:
|
|
file.write(content)
|
|
except Exception as e:
|
|
return Result(e, ResultType.failure, inspect.currentframe().f_code.co_name)
|
|
else:
|
|
return Result("", ResultType.success)
|
|
|
|
|
|
def get_distro_name():
|
|
distro_name = None
|
|
with open("/etc/os-release") as f:
|
|
content = f.read()
|
|
start = content.find("ID=") + 3
|
|
end = content.find("\n", start)
|
|
distro_name = content[start:end]
|
|
return distro_name
|
|
|
|
|
|
class PackageManagementOperation(object):
|
|
@staticmethod
|
|
def install(package_name):
|
|
try:
|
|
distro = get_distro_name()
|
|
if distro == "alpine":
|
|
command = f"apk add {package_name}"
|
|
else:
|
|
assert "Unknown Distro"
|
|
|
|
subprocess.check_output(command.split())
|
|
except Exception as e:
|
|
print(e)
|
|
return Result(e, ResultType.failure, inspect.currentframe().f_code.co_name)
|
|
else:
|
|
return Result("", ResultType.success)
|