def save_flight(self, file_name):
     '''Chea chea chae'''
     page = util.create_page_from_file(file_name)
     print page
     sleep(3)
     page = util.create_page(self.driver)
     util.save_page(page, file_name)
 def get_mileage_balance(self):
     '''Find the amount of miles on the page'''
     sleep(5)
     page = util.create_page(self.driver)
     try:
         if page.xpath('//span[@class="points"]'):
             mileage_balance = page.xpath('//span[@class="points"]'
                               )[0].text_content().strip('\n pts.')
         elif page.xpath('//th'):
             mileage_balance = page.xpath('//th')[0].text.strip('\n ')
         elif page.xpath('.//div[@id="FormUserControl__'
                         'myOverview__mileagePlanInfo"]'):
             header = page.xpath('.//div[@id="FormUserControl__'
                                 'myOverview__mileagePlanInfo"]'
                                 )[0].text_content()
             mileage_balance = re.search(r'Miles:(.+$)',
                               header).group(1).replace(u'\xa0', u'')
         print mileage_balance
     except:
         pass