mirror of
https://github.com/Ragora/TSScraper.git
synced 2026-01-19 20:24:57 +00:00
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
import re
|
|
import os
|
|
import sys
|
|
import importlib
|
|
import os.path
|
|
import shutil
|
|
|
|
class Exporter(object):
|
|
data = None
|
|
|
|
def __init__(self, data, target_directory):
|
|
self.data = data
|
|
self.target_directory = target_directory
|
|
|
|
def _path_visitor(self, arg, dirname, names):
|
|
for name in names:
|
|
mirrored_path = os.path.join(dirname, name)
|
|
relative_path = os.path.join(arg, mirrored_path.replace(self.target_directory + "/", ""))
|
|
|
|
try:
|
|
if (os.path.isdir(mirrored_path)):
|
|
os.mkdir(relative_path)
|
|
except OSError:
|
|
pass
|
|
|
|
def write(self, directory):
|
|
import jinja2
|
|
|
|
# Read the template files first
|
|
file_template = None
|
|
with open("data/filetempl.html", "r") as handle:
|
|
file_template = handle.read()
|
|
|
|
index_template = None
|
|
with open("data/indextempl.html", "r") as handle:
|
|
index_template = handle.read()
|
|
|
|
html_filenames = [ ]
|
|
|
|
# Recurse the target directory and recreate its structure
|
|
os.path.walk(self.target_directory, self._path_visitor, directory)
|
|
|
|
# For each file entry...
|
|
script_relative_paths = [ ]
|
|
for file in self.data["files"]:
|
|
if (len(file.global_functions) == 0 and len(file.bound_functions.keys()) == 0 and len(file.datablocks) == 0):
|
|
continue
|
|
|
|
# First, we collapse to a file path relative to our output dir
|
|
# FIXME: Dirty hack to make sure the os.path.join works
|
|
html_filename = file.path.replace(self.target_directory + "/", "")
|
|
script_relative = html_filename
|
|
file.mod_path = html_filename
|
|
script_relative_paths.append(script_relative)
|
|
|
|
# Next, we ensure that the subdirectories exist
|
|
#html_filename = html_filename.lstrip("./").replace("/", "-")
|
|
html_filename, oldextension = os.path.splitext(html_filename)
|
|
html_filename = "%s.html" % html_filename
|
|
html_filenames.append(html_filename)
|
|
file.web_path = html_filename
|
|
|
|
with open(os.path.join(directory, html_filename), "w") as handle:
|
|
template = jinja2.Template(file_template)
|
|
handle.write(template.render(file=file))
|
|
|
|
# Dump the index file
|
|
with open(os.path.join(directory, "index.html"), "w") as handle:
|
|
template = jinja2.Template(index_template)
|
|
|
|
handle.write(template.render(files=self.data["files"]))
|
|
|
|
# Puke bootstrap into the directory
|
|
try:
|
|
shutil.copytree("data/bootstrap", os.path.join(directory, "bootstrap"))
|
|
except OSError:
|
|
pass
|
|
|
|
print("Done processing.")
|