def default_select_path(self, path):
if not util.is_text_file(path):
return False
if os.path.getsize(path) > MAX_DEFAULT_SOURCECODE_FILE_SIZE:
self._warn_default_sourcecode_file_too_big(path)
return False
return True
def is_requirements(path):
if not util.is_text_file(path):
return False
try:
list(parse_requirements(path))
except Exception:
return False
else:
return True
def _init_file_template(path, run_dest=None, filters=None):
if not util.is_text_file(path):
return None
dirname, basename = os.path.split(path)
templates_home = _local_path("templates")
env = jinja2.Environment(
loader=jinja2.FileSystemLoader([dirname, templates_home]),
autoescape=jinja2.select_autoescape(['html', 'xml']))
RunFilters(run_dest).install(env)
if filters:
env.filters.update(filters)
try:
return env.get_template(basename)
except jinja2.TemplateError as e:
raise TemplateError(e)
def _base_file_type_info(path):
path_lower = path.lower()
if re.search(r"\.tfevents\.", path_lower):
return "Event log", "file-chart", "File", None
elif re.search(r"saved_model\.pb$", path_lower):
return "SavedModel protocol buffer", "file", "File", None
elif re.search(r"graph\.pb$", path_lower):
return "GraphDef protocol buffer", "file", "File", None
elif re.search(r"saved_model\.pbtxt$", path_lower):
return (
"SavedModel protocol buffer", "file-document", "Text file",
"text")
elif re.search(r"graph\.pbtxt$", path_lower):
return (
"GraphDef protocol buffer", "file-document", "Text file",
"text")
elif re.search(r"\.index$", path_lower):
return "Checkpoint index", "file", "File", None
elif re.search(r"\.meta$", path_lower):
return "Checkpoint meta graph", "file", "File", None
elif re.search(r"[/\\]checkpoint$", path_lower):
return "Latest checkpoint marker", "file", "File", None
elif re.search(r"data-\d+-of-\d+$", path_lower):
return "Checkpoint values", "file", "File", None
elif re.search(r"\.tfrecord$", path_lower):
return "Dataset file", "file", "File", None
elif re.search(r"\.(jpg|jpeg|gif|png|tiff)$", path_lower):
return "Image", "file-image", "Image", "image"
elif re.search(r"\.mid", path_lower):
return "Audio", "file-music", "Audio", "midi"
elif re.search(r"\.(wav|mp3)", path_lower):
return "Audio", "file-music", "Audio", None
elif re.search(r"\.(csv|tsv)", path_lower):
return "Table", "file-delimited", "Delimited", "table"
else:
if util.is_text_file(path):
return "Text file", "file-document", "Text file", "text"
else:
return "File", "file", "File", None
def _init_file_template(path, run_dest=None, filters=None):
"""Returns template for path or None if path is not a text file.
Raises TemplateError if path does not exist or cannot be parsed as
a template.
"""
if not os.path.exists(path):
raise TemplateError("%s does not exist" % path)
if not util.is_text_file(path):
return None
dirname, basename = os.path.split(path)
templates_home = _local_path("templates")
env = jinja2.Environment(
loader=jinja2.FileSystemLoader([dirname, templates_home]),
autoescape=jinja2.select_autoescape(['html', 'xml']),
)
RunFilters(run_dest).install(env)
if filters:
env.filters.update(filters)
try:
return env.get_template(basename)
except jinja2.TemplateError as e:
raise TemplateError(e)
def _default_file_type_info(path):
if not os.path.exists(path):
return "File", "file", "File", None
if util.is_text_file(path):
return "Text file", "file-document", "Text file", "text"
return "File", "file", "File", None
