def copy_button_clicked(self):
self.copy_button['text'] = 'Loading...'
self.copy_button['state'] = tk.DISABLED
parser = Parser()
url = self.url.get()
directory = self.directory.get()
css_directory = directory + self.css_path
js_directory = directory + self.js_path
if FileHelper.directory_exists(directory):
parser.set_url(url)
try:
content = parser.get_content()
self.write_log("Received HTML page from " + url)
FileHelper.create_file('index.html', directory, content.decode("UTF-8"))
self.write_log("Saved HTML page to " + directory + '/index.html')
FileHelper.delete_folder_if_exists(css_directory)
FileHelper.create_path_recursively(css_directory)
self.write_log("Created directory for saving CSS stylesheets")
link_nodes = parser.get_nodes('link')
css_regexp = re.compile("\.css$", re.IGNORECASE)
for node in link_nodes:
link = Parser.form_script_url(url, node.get('href'))
link = str(link)
if css_regexp.search(link):
script_name = re.findall('\w+\.css', link, re.IGNORECASE)
if len(script_name) > 0:
script_name = script_name[0]
else:
script_name = uuid.uuid4().hex
css_content = Parser.get_content_from_url(link)
if css_content:
FileHelper.create_file(script_name, directory + self.css_path, css_content.decode("UTF-8"))
self.write_log("Saved CSS - " + self.css_path + script_name)
else:
self.write_log("Can't get content from - " + link)
FileHelper.delete_folder_if_exists(js_directory)
FileHelper.create_path_recursively(js_directory)
self.write_log("Created directory for saving JS scripts")
script_nodes = parser.get_nodes('script')
js_regexp = re.compile("\.js", re.IGNORECASE)
for node in script_nodes:
src = node.get('src')
link = None
if src:
link = Parser.form_script_url(url, src)
link = str(link)
if js_regexp.search(link):
script_name = re.findall('\w+\.js', str(link), re.IGNORECASE);
if len(script_name) > 0:
script_name = script_name[0]
else:
script_name = uuid.uuid4().hex
js_content = Parser.get_content_from_url(link)
if js_content:
FileHelper.create_file(script_name, directory + self.js_path, js_content.decode("UTF-8"))
self.write_log("Saved JS - " + self.js_path + script_name)
else:
self.write_log("Can't get content from - " + link)
except Exception as e:
self.write_log("ERROR")
self.write_log(e)
self.show_error_message(e)
else:
self.write_log("ERROR")
self.write_log("Directory '" + directory + "' not exists")
self.show_error_message("Directory '" + directory + "' not exists")
self.copy_button['text'] = 'Start'
self.copy_button['state'] = tk.NORMAL
