import re
from services_page import ServicesPage
import page


class DataCentersPage(page.Page):
    page = None

    def create_data_center(self, name):

        self.Button('Create Windows Data Center').Click()

        self.EditBox('id_name').Set(name)
        xpath = "//input[@value='Create']"
        self.Button(xpath).Click()

        return self

    def delete_data_center(self, name):
        link = self.Link(name).Address()
        datacenter_id = re.search('windc/(\S+)', link).group(0)[6:-1]

        print 'Data Center ID: ' + datacenter_id

        xpath = ".//*[@id='windc__row__%s']/td[4]/div/a[2]" % datacenter_id
        self.Button(xpath).Click()

        button_id = "windc__row_%s__action_delete" % datacenter_id
        self.Button(button_id).Click()

        self.Button("Delete Data Center").Click()

        return self

    def select_data_center(self, name):
        self.Link(name).Click()
        page = ServicesPage(self.driver)
        return page