# -*- coding: utf-8 -*-
"""
Created on Sun Sep  2 21:08:57 2018

@author: toni
"""
import novaclient
import novaclient.client as nvclient
import neutronclient.neutron.client as neutron
from saharaclient.api import client
import keystoneclient.v2_0.client as ksclient
from credentials import get_creds
from credentials import get_nova_creds
from saharaclient.api import images
import glanceclient.v2.client as glclient
keystone = ksclient.Client(**get_creds())
from keystoneauth1.identity import v2
from keystoneauth1 import session as keystone_session

sahara_endpoint = keystone.service_catalog.url_for(service_type='data-processing')
neutron_endpoint = keystone.service_catalog.url_for(service_type='network')

auth = v2.Password(auth_url=keystone.auth_url,
                   username=keystone.username,
                   password=keystone.password,
                   tenant_name=keystone.tenant_name)
ses = keystone_session.Session(auth=auth)
nova = nvclient.Client('2', session=ses)
sahara_client = client.Client('1.1', session=ses)
glance = glclient.Client(session=ses)
sahara_image = sahara_client.images.update_image('a00b33a2-0eaf-4e23-9ec4-149f3eadbde1', 'ubuntu')
sahara_client.images.update_tags('a00b33a2-0eaf-4e23-9ec4-149f3eadbde1', ['vanilla', '2.7.1'])
ng_master = sahara_client.node_group_templates.create(
            name='toni-master-node2',
            plugin_name='vanilla',
            hadoop_version='2.7.1',
            flavor_id='2',
            node_processes=["namenode", "secondarynamenode" ,"resourcemanager", "historyserver", "oozie"],
            floating_ip_pool='admin_floating_net',
            #security_groups='default',
            image_id='a00b33a2-0eaf-4e23-9ec4-149f3eadbde1'
            )
ng_worker = sahara_client.node_group_templates.create(
            name='toni-worker-node2',
            plugin_name='vanilla',
            hadoop_version='2.7.1',
            flavor_id='2',
            node_processes=["datanode", "nodemanager"],
            floating_ip_pool='admin_floating_net',
            #security_groups='default',
            image_id='a00b33a2-0eaf-4e23-9ec4-149f3eadbde1'
            )

cl_template = sahara_client.cluster_templates.create(
            name='toni-hadoop-cluster-template2',
            plugin_name='vanilla',
            hadoop_version='2.7.1',
            )
sahara_client.cluster_templates.update(
            cluster_template_id=cl_template.id,
            node_groups=[
                    dict(
                    name='toni-master-node2',
                    node_group_template_id=ng_master.id,
                    count=1),
                    dict(
                    name='toni-worker-node2',
                    node_group_template_id=ng_worker.id,
                    count=2)
                    ]
                    )