def download(self, share, directory=None): if not directory: dl = Download.get_download(share, Configuration.save_dir + "/" + share.name, date.today(), 1) else: dl = Download.get_download(share, directory + "/" + share.name, date.today(), 1) TabDownloads.instance.add_download(dl) dl.start_download() self.tabs.setCurrentWidget(self.tab_downloads)
def __init__(self, **kwargs): self.url = kwargs.get('url', None) self.browser = Firefox(options=opts) self.reporttype = kwargs.get('reporttype', None) if self.reporttype: self.download = Download(sub_dir=self.reporttype.name) else: self.download = Download()
class EIAGasPrice: def __init__(self, **kwargs): self.url = kwargs.get('url', None) self.browser = Firefox(options=opts) self.reporttype = kwargs.get('reporttype', None) if self.reporttype: self.download = Download(sub_dir=self.reporttype.name) else: self.download = Download() def download_file(self): url = self.find_download_url() if bool(url): file, header = self.download.run(url) return file return None def create_csv(self, xlsfile): csv = CSVGenerator(xlsfile, sheet_name='Data 1', header=2, names=['Date', 'Price'], sub_dir=self.reporttype.name, ext='csv', rt=self.reporttype) csv.generate() def find_download_url(self): element = self.browser.find_elements_by_link_text( 'Download Data (XLS File)') if len(element) > 0: return element[0].get_attribute('href') else: return '' def open(self): self.browser.get(self.url) assert "Henry Hub Natural Gas Spot Price (Dollars per Million Btu)" in self.browser.title def close(self): self.browser.close()
def download(self, share): dl = Download.get_download(share, share.name, date.today()) TabDownloads.instance.add_download(dl) dl.start_download()