Skip to content

Commit 2aba1bc

Browse files
committed
🎨 figure out textwrap, fix warnings
- separate common imports - remove import and specify encoding
1 parent ec43047 commit 2aba1bc

File tree

1 file changed

+22
-7
lines changed

1 file changed

+22
-7
lines changed

src/vuegen/streamlit_reportview.py

Lines changed: 22 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
2-
import re
32
import subprocess
43
import sys
4+
import textwrap
55
from pathlib import Path
66
from typing import List
77

@@ -75,16 +75,29 @@ def generate_report(
7575
self.report.logger.debug("Processing app navigation code.")
7676
# Define the Streamlit imports and report manager content
7777
report_manag_content = []
78+
report_manag_content.append(
79+
textwrap.dedent(
80+
"""\
81+
import streamlit as st
82+
"""
83+
)
84+
)
7885
if self.report.logo:
7986
report_manag_content.append(
80-
f"""import streamlit as st\n
81-
st.set_page_config(layout="wide", page_title="{self.report.title}", page_icon="{self.report.logo}")
82-
st.logo("{self.report.logo}")"""
87+
textwrap.dedent(
88+
f"""\
89+
st.set_page_config(layout="wide", page_title="{self.report.title}", page_icon="{self.report.logo}")
90+
st.logo("{self.report.logo}")
91+
"""
92+
)
8393
)
8494
else:
8595
report_manag_content.append(
86-
f"""import streamlit as st\n
87-
st.set_page_config(layout="wide", page_title="{self.report.title}")"""
96+
textwrap.dedent(
97+
f"""\
98+
st.set_page_config(layout="wide", page_title="{self.report.title}")
99+
"""
100+
)
88101
)
89102
report_manag_content.append(
90103
self._format_text(
@@ -147,7 +160,9 @@ def generate_report(
147160
)
148161

149162
# Write the navigation and general content to a Python file
150-
with open(Path(output_dir) / self.REPORT_MANAG_SCRIPT, "w") as nav_manager:
163+
with open(
164+
Path(output_dir) / self.REPORT_MANAG_SCRIPT, "w", encoding="utf8"
165+
) as nav_manager:
151166
nav_manager.write("\n".join(report_manag_content))
152167
self.report.logger.info(
153168
f"Created app navigation script: {self.REPORT_MANAG_SCRIPT}"

0 commit comments

Comments
 (0)