Cutieguwu/balloon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Cutieguwu
2025-07-21 22:09:42 -04:00
parent 62ed4969c6
commit 0ea69a5c89
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.py
View File

@@ -13,7 +13,7 @@ from typing import TYPE_CHECKING
if TYPE_CHECKING: if TYPE_CHECKING:
from _typeshed import StrPath from _typeshed import StrPath
WORK_DIR: StrPath = os.getcwd() WORK_DIR: StrPath = os.getcwd() + '/'
@dataclass @dataclass
class Tag: class Tag:
@@ -104,16 +104,24 @@ def main() -> None:
# Incorrect number of arguments # Incorrect number of arguments
# Long help flag # Long help flag
# Short help flag # Short help flag
if len(sys.argv) != 1 or ( if len(sys.argv) != 2 or (
sys.argv[0] == '--help' sys.argv[0] == '--help'
or sys.argv[0] == '-h' or sys.argv[0] == '-h'
): ):
help() help()
return return
with open(str(WORK_DIR) + 'src' + sys.argv[0], 'rt') as f: file_name = sys.argv[1].removeprefix('src/')
with open(str(WORK_DIR) + 'src/' + file_name, 'rt') as f:
html_src = HTML(f.read()) html_src = HTML(f.read())
with open(str(WORK_DIR) + 'target' + sys.argv[0], 'w') as f:
try:
os.makedirs(str(WORK_DIR) + 'target/' + os.path.dirname(file_name))
except FileExistsError:
pass
with open(str(WORK_DIR) + 'target/' + file_name, 'w') as f:
f.write(html_src.write()) f.write(html_src.write())
def help() -> None: def help() -> None: