CassOnKube/main.py at master · CodeMaxx/CassOnKube · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#! /usr/bin/env python3

#####
### Author: Akash Trehan
### Email: akash.trehan123@gmail.com
### Web: https://www.akashtrehan.com/
#####

### Tested on Ubuntu 18.04

import argparse
import subprocess
import sys
from termcolor import colored

VERSION = '1.0.0'

def buildArgParser():
    parser = argparse.ArgumentParser(description='Cassandra On Kube ver ' + VERSION)
    parser.add_argument('-v', '--visor', help='Specify which hypervisor you want to use for virtualisation', default='virtualbox', choices=['virtualbox', 'kvm'])
    parser.add_argument('-s', '--server', help='Number of server', default=3, type=int)
    return parser

def run_command(command):
    p = subprocess.run(command, shell=True)
    if p.returncode != 0:
        print(colored("Aborting...", "red"))
        sys.exit()

def setup_cassandra(num):
    vols = ""

    for i in range(num):
        vols += "---\n\
        apiVersion: v1\n\
        kind: PersistentVolume\n\
        metadata:\n\
          name: cassandra-data-%d\n\
          labels:\n\
            type: local\n\
            app: cassandra\n\
        spec:\n\
          capacity:\n\
            storage: 1Gi\n\
          accessModes:\n\
            - ReadWriteOnce\n\
          hostPath:\n\
            path: /tmp/data/cassandra-data-%d\n\
          persistentVolumeReclaimPolicy: Recycle\n" % (i, i)

    with open('cassandra/local-volumes.yaml', 'w') as f:
        f.write(vols)

    run_command("scripts/setup_cassandra.sh %d" % num)

def main():
    # Reading arguments
    parser = buildArgParser()
    args = parser.parse_args()

    # Install Docker
    run_command("scripts/install_docker.sh")

    # Install Kubernetes
    run_command("scripts/install_kubernetes.sh")

    # Install minikube
    run_command("scripts/install_minikube.sh")

    # Install a hypervisor
    if args.visor == 'virtualbox':
        run_command("scripts/install_virtualbox.sh")
    elif args.visor == 'kvm':
        run_command("scripts/install_kvm.sh")

    # Run minikube
    run_command("scripts/base_setup.sh")

    setup_cassandra(args.server)


if __name__ == '__main__':
    main()