lnmch/kops-arm64
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch to Poetry and Black (#254)

Browse Source 
* Pipenv -> Poetry

* poetry and black
This commit is contained in:
Henning Jacobs
2019-12-23 20:07:46 +01:00
committed by GitHub
parent 578b03d214
commit d8b94db671
22 changed files with 1575 additions and 959 deletions
Download Patch File Download Diff File Expand all files Collapse all files

127
kube_ops_view/mock.py
View File

@@ -4,34 +4,33 @@ import string
def hash_int(x: int):
x = ((x >> 16) ^ x) * 0x45d9f3b
x = ((x >> 16) ^ x) * 0x45d9f3b
x = ((x >> 16) ^ x) * 0x45D9F3B
x = ((x >> 16) ^ x) * 0x45D9F3B
x = (x >> 16) ^ x
return x
def generate_mock_pod(index: int, i: int, j: int):
names = [
'agent-cooper',
'black-lodge',
'bob',
'bobby-briggs',
'laura-palmer',
'leland-palmer',
'log-lady',
'sheriff-truman',
"agent-cooper",
"black-lodge",
"bob",
"bobby-briggs",
"laura-palmer",
"leland-palmer",
"log-lady",
"sheriff-truman",
]
labels = {
'env': ['prod', 'dev'],
'owner': ['x-wing', 'iris']
}
pod_phases = ['Pending', 'Running', 'Running', 'Failed']
labels = {"env": ["prod", "dev"], "owner": ["x-wing", "iris"]}
pod_phases = ["Pending", "Running", "Running", "Failed"]
pod_labels = {}
for li, k in enumerate(labels):
v = labels[k]
label_choice = hash_int((index + 1) * (i + 1) * (j + 1) * (li + 1)) % (len(v) + 1)
if(label_choice != 0):
label_choice = hash_int((index + 1) * (i + 1) * (j + 1) * (li + 1)) % (
len(v) + 1
)
if label_choice != 0:
pod_labels[k] = v[label_choice - 1]
phase = pod_phases[hash_int((index + 1) * (i + 1) * (j + 1)) % len(pod_phases)]
@@ -44,41 +43,53 @@ def generate_mock_pod(index: int, i: int, j: int):
usage_cpu = max(requests_cpu + random.randint(-30, 30), 1)
usage_memory = max(requests_memory + random.randint(-64, 128), 1)
container = {
'name': 'myapp',
'image': 'foo/bar/{}'.format(j),
'resources': {
'requests': {'cpu': f'{requests_cpu}m', 'memory': f'{requests_memory}Mi'},
'limits': {},
'usage': {'cpu': f'{usage_cpu}m', 'memory': f'{usage_memory}Mi'},
"name": "myapp",
"image": "foo/bar/{}".format(j),
"resources": {
"requests": {
"cpu": f"{requests_cpu}m",
"memory": f"{requests_memory}Mi",
},
"limits": {},
"usage": {"cpu": f"{usage_cpu}m", "memory": f"{usage_memory}Mi"},
},
'ready': True,
'state': {'running': {}}
"ready": True,
"state": {"running": {}},
}
if phase == 'Running':
if phase == "Running":
if j % 13 == 0:
container.update(**{'ready': False, 'state': {'waiting': {'reason': 'CrashLoopBackOff'}}})
container.update(
**{
"ready": False,
"state": {"waiting": {"reason": "CrashLoopBackOff"}},
}
)
elif j % 7 == 0:
container.update(**{'ready': False, 'state': {'running': {}}, 'restartCount': 3})
elif phase == 'Failed':
del container['state']
del container['ready']
container.update(
**{"ready": False, "state": {"running": {}}, "restartCount": 3}
)
elif phase == "Failed":
del container["state"]
del container["ready"]
containers.append(container)
pod = {
'name': '{}-{}-{}'.format(names[hash_int((i + 1) * (j + 1)) % len(names)], i, j),
'namespace': 'kube-system' if j < 3 else 'default',
'labels': pod_labels,
'phase': phase,
'containers': containers
"name": "{}-{}-{}".format(
names[hash_int((i + 1) * (j + 1)) % len(names)], i, j
),
"namespace": "kube-system" if j < 3 else "default",
"labels": pod_labels,
"phase": phase,
"containers": containers,
}
if phase == 'Running' and j % 17 == 0:
pod['deleted'] = 123
if phase == "Running" and j % 17 == 0:
pod["deleted"] = 123
return pod
def query_mock_cluster(cluster):
'''Generate deterministic (no randomness!) mock data'''
index = int(cluster.id.split('-')[-1])
"""Generate deterministic (no randomness!) mock data"""
index = int(cluster.id.split("-")[-1])
nodes = {}
for i in range(10):
# add/remove the second to last node every 13 seconds
@@ -88,11 +99,11 @@ def query_mock_cluster(cluster):
# only the first two clusters have master nodes
if i < 2 and index < 2:
if index == 0:
labels['kubernetes.io/role'] = 'master'
labels["kubernetes.io/role"] = "master"
elif index == 1:
labels['node-role.kubernetes.io/master'] = ''
labels["node-role.kubernetes.io/master"] = ""
else:
labels['master'] = 'true'
labels["master"] = "true"
pods = {}
for j in range(hash_int((index + 1) * (i + 1)) % 32):
# add/remove some pods every 7 seconds
@@ -100,7 +111,7 @@ def query_mock_cluster(cluster):
pass
else:
pod = generate_mock_pod(index, i, j)
pods['{}/{}'.format(pod['namespace'], pod['name'])] = pod
pods["{}/{}".format(pod["namespace"], pod["name"])] = pod
# use data from containers (usage)
usage_cpu = 0
@@ -111,27 +122,27 @@ def query_mock_cluster(cluster):
usage_memory += int(c["resources"]["usage"]["memory"].split("Mi")[0])
# generate longer name for a node
suffix = ''.join(
suffix = "".join(
[random.choice(string.ascii_letters) for n in range(random.randint(1, 20))]
)
node = {
'name': f'node-{i}-{suffix}',
'labels': labels,
'status': {
'capacity': {'cpu': '8', 'memory': '64Gi', 'pods': '110'},
'allocatable': {'cpu': '7800m', 'memory': '62Gi'}
"name": f"node-{i}-{suffix}",
"labels": labels,
"status": {
"capacity": {"cpu": "8", "memory": "64Gi", "pods": "110"},
"allocatable": {"cpu": "7800m", "memory": "62Gi"},
},
'pods': pods,
"pods": pods,
# get data from containers (usage)
'usage': {'cpu': f'{usage_cpu}m', 'memory': f'{usage_memory}Mi'}
"usage": {"cpu": f"{usage_cpu}m", "memory": f"{usage_memory}Mi"},
}
nodes[node['name']] = node
nodes[node["name"]] = node
pod = generate_mock_pod(index, 11, index)
unassigned_pods = {'{}/{}'.format(pod['namespace'], pod['name']): pod}
unassigned_pods = {"{}/{}".format(pod["namespace"], pod["name"]): pod}
return {
'id': 'mock-cluster-{}'.format(index),
'api_server_url': 'https://kube-{}.example.org'.format(index),
'nodes': nodes,
'unassigned_pods': unassigned_pods
"id": "mock-cluster-{}".format(index),
"api_server_url": "https://kube-{}.example.org".format(index),
"nodes": nodes,
"unassigned_pods": unassigned_pods,
}