make a test and main

Author: dragoncoder047

pyproject.toml

@@ -25,5 +25,5 @@ requires-python = ">=3.10"
 [project.urls]
 Homepage = "https://github.com/dragoncoder047/schemascii"
 
-# [project.scripts]
-# schemascii = "schemascii.__init__:cli_main"
+[project.scripts]
+schemascii = "schemascii.__main__:cli_main"

schemascii/__init__.py

@@ -43,6 +43,6 @@ def render(filename: str, text: str = None, **options) -> str:
 
 if __name__ == '__main__':
     print(render(
-        "../test_data/test_resistors.txt",
+        "test_data/test_resistors.txt",
         scale=20, padding=20, stroke_width=2,
         stroke="black"))

schemascii/__main__.py

@@ -0,0 +1,29 @@
+import argparse
+import sys
+from . import render, __version__
+
+
+def cli_main():
+    ap = argparse.ArgumentParser(
+        prog="schemascii",
+        description="Render ASCII-art schematics into SVG.")
+    ap.add_argument("in_file",
+                    help="File to process.")
+    ap.add_argument("-o", "--out",
+                    default=None,
+                    dest="out_file",
+                    help="Output SVG file. (default input file plus .svg)")
+    args = ap.parse_args()
+    if args.out_file is None:
+        args.out_file = args.in_file + ".svg"
+    try:
+        result_svg = render(args.in_file)
+    except Exception as e:
+        print("error:", e, file=sys.stderr)
+        sys.exit(1)
+    with open(args.out_file, "w", encoding="utf-8") as out:
+        out.write(result_svg)
+
+
+if __name__ == '__main__':
+    cli_main()

schemascii/components.py

@@ -95,7 +95,7 @@ def find_all(grid: Grid) -> tuple[list[Cbox], list[BOMData]]:
 
 
 if __name__ == '__main__':
-    grid = Grid("../test_data/test_resistors.txt")
+    grid = Grid("test_data/test_resistors.txt")
     bbb, _ = find_all(grid)
     all_pts = []
     for box in bbb:

schemascii/components_render.py

@@ -262,7 +262,7 @@ def render_component(
     "Render the component into an SVG string."
     if box.type not in RENDERERS:
         raise NameError(
-            f"No renderer defined for {box.type} component")
+            f"Unsupported component type: {box.type}")
     return XML.g(
         RENDERERS[box.type](box, terminals, bom_data, **kwargs),
         class_=f"component {box.type}"

schemascii/wires.py

@@ -137,6 +137,6 @@ def get_wires(grid: Grid, **options) -> str:
 
 
 if __name__ == '__main__':
-    xg = Grid('../test_data/test_resistors.txt')
+    xg = Grid('test_data/test_resistors.txt')
     print(get_wires(xg, scale=20))
     print(xg)