Skip to content

feat: show directory tree after generating #28

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 26, 2021
Merged

feat: show directory tree after generating #28

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 30 additions & 16 deletions app/streamlit_app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
import os
import shutil
from pathlib import Path
from subprocess import check_output

import streamlit as st
from codegen import CodeGenerator
Expand Down Expand Up @@ -46,8 +48,16 @@ def sidebar(self, template_list=None, config=None):

def render_code(self, fname="", code=""):
"""Main content with the code."""
with st.beta_expander(f"View generated {fname}"):
st.code(code)
with st.beta_expander(f"View rendered {fname}"):
if fname.endswith(".md"):
st.markdown(code)
else:
st.code(code)

def render_directory(self, dir):
output = check_output(["tree", dir], encoding="utf-8")
st.markdown("Generated files and directory structure")
st.code(output)

def add_sidebar(self):
def config(template_name):
Expand All @@ -58,25 +68,29 @@ def config(template_name):
def add_content(self):
"""Get generated/rendered code from the codegen."""
content = [*self.codegen.render_templates(self.template_name, self.config)]
if st.checkbox("View generated code ?"):
if st.checkbox("View rendered code ?"):
for fname, code in content:
self.render_code(fname, code)

def add_download(self):
st.markdown("")
archive_format = st.radio("Archive formats", self.codegen.available_archive_formats)
# temporary hack until streamlit has official download option
# https://github.com/streamlit/streamlit/issues/400
# https://github.com/streamlit/streamlit/issues/400#issuecomment-648580840
if st.button("Generate an archive"):
archive_fname = self.codegen.make_archive(self.template_name, archive_format)
# this is where streamlit serves static files
# ~/site-packages/streamlit/static/static/
dist_path = Path(st.__path__[0]) / "static/static/dist"
if not dist_path.is_dir():
dist_path.mkdir()
shutil.copy(archive_fname, dist_path)
st.success(f"Download link : [{archive_fname}](./static/{archive_fname})")
col1, col2 = st.beta_columns(2)
with col1:
archive_format = st.radio("Archive formats", self.codegen.available_archive_formats)
# temporary hack until streamlit has official download option
# https://github.com/streamlit/streamlit/issues/400
# https://github.com/streamlit/streamlit/issues/400#issuecomment-648580840
if st.button("Generate an archive"):
archive_fname = self.codegen.make_archive(self.template_name, archive_format)
# this is where streamlit serves static files
# ~/site-packages/streamlit/static/static/
dist_path = Path(st.__path__[0]) / "static/static/dist"
if not dist_path.is_dir():
dist_path.mkdir()
shutil.copy(archive_fname, dist_path)
st.success(f"Download link : [{archive_fname}](./static/{archive_fname})")
with col2:
self.render_directory(os.path.join(self.codegen.dist_dir, self.template_name))

def run(self):
self.add_sidebar()
Expand Down