def import_website(user=None, path_to_site=None): """import a website * create templates * add content to database * copy static files """ print 'import_website' if not (user and path_to_site): path_to_site = request.args['path_to_site'] user = request.args['user'] sitename = os.path.basename(os.path.normpath(path_to_site)) static_path = os.path.join(app.config['STATIC_MEDIA'], user, sitename) if not os.path.exists(static_path): os.makedirs(static_path) # copy static files # path = os.path.dirname(__file__) # new_static_path = os.path.join(path, '..', 'static', user, sitename) # if os.path.exists(new_static_path): # shutil.rmtree(new_static_path) # if os.path.exists(os.path.join(path_to_site, 'static')): # shutil.copytree(os.path.join(path_to_site, 'static'), new_static_path) # else: # os.makedirs(os.path.join(new_static_path, 'static')) # if os.path.exists(os.path.join(path_to_site, '__icon.png')): # shutil.copyfile(os.path.join(path_to_site, '__icon.png'), os.path.join(new_static_path, '__icon.png')) # create templates and add parsed stuff to db new_templates_path = os.path.join(app.config['TEMPLATE_PATH'], user, sitename) if not os.path.exists(new_templates_path): os.makedirs(new_templates_path) parser = Parser(user, sitename) for filename in os.listdir(path_to_site): if any([filename.endswith(ext) for ext in app.config['HTML_EXT']]): template = parser.parse_html(os.path.join(path_to_site, filename)) save_path = os.path.join(new_templates_path, os.path.basename(filename)) with open(save_path, 'w') as f: f.write(str(template)) for entry in parser.fields: Entry.get_or_create(user=user, site=sitename, **entry) for resource in set(parser.resources): if any([resource.endswith(ext) for ext in app.config["STYLESHEET_EXT"]]): parser.parse_css(path_to_site, resource) for resource in set(parser.resources): source = os.path.join(path_to_site, resource) destination = os.path.join(static_path, resource) if not os.path.exists(source): print "WARNING", source, "does not exist." else: if not os.path.exists(os.path.dirname(destination)): os.makedirs(os.path.dirname(destination)) shutil.copyfile(source, destination) print "Copying", resource #shutil.copyfile(os.path.join(path_to_site, '__icon.png'), os.path.join(new_static_path, '__icon.png')) return ''