primitive 0.1.67__py3-none-any.whl → 0.1.69__py3-none-any.whl

primitive/lint/actions.py

@@ -1,76 +0,0 @@
-from pathlib import Path
-from primitive.utils.actions import BaseAction
-import subprocess
-from typing import Tuple
-from loguru import logger
-from ..utils.files import find_files_for_extension
-from ..utils.verible import install_verible
-from ..utils.cache import get_deps_cache
-
-
-class Lint(BaseAction):
-    def execute(self, source: Path = Path.cwd()) -> Tuple[bool, str]:
-        logger.debug(f"Starting linter for source: {source}")
-        files = find_files_for_extension(source, ".sv")
-        if not files:
-            message = "No files found to lint."
-            logger.warning(message)
-            return False, message
-
-        logger.debug("Checking if verible is installed")
-        verible_path = Path("verible-verilog-lint")
-        try:
-            subprocess.run([str(verible_path), "--version"], capture_output=True)
-        except FileNotFoundError:
-            logger.debug("Verible not found in $PATH. Looking in deps cache...")
-            cache_dir = get_deps_cache()
-
-            possible_dirs = cache_dir.glob("verible*")
-            verible_dir = next(possible_dirs, None)
-
-            if verible_dir is not None:
-                verible_path = verible_dir / "bin" / verible_path
-            else:
-                logger.debug("Did not find verible. Installing...")
-                try:
-                    system_info = self.primitive.hardware.get_system_info()
-                    verible_bin = install_verible(system_info=system_info)
-                    verible_path = verible_bin / verible_path
-                except Exception as exception:
-                    message = f"Failed to install verible. {exception}"
-                    logger.error(message)
-                    return False, message
-
-        try:
-            subprocess.run([str(verible_path), "--version"], capture_output=True)
-        except FileNotFoundError:
-            message = "Verible is not installed. Please install it to run lint."
-            logger.error(message)
-            return False, message
-
-        # TODO:
-        # run is great for now! we will need to switch to Popen if we want to stream the output
-        logger.debug("Running linter...")
-        result = subprocess.run(
-            [str(verible_path), *files],
-            capture_output=True,
-            text=True,
-        )
-
-        logger.debug("Linting complete.")
-
-        message = ""
-        if result.stderr:
-            logger.error("\n" + result.stderr)
-        if result.stdout:
-            logger.info("\n" + result.stdout)
-        message = "See above logs for linter output."
-
-        if result.returncode != 0:
-            if not self.primitive.DEBUG:
-                message = result.stdout + result.stderr
-            return (False, message)
-        else:
-            message = "Linting successful."
-
-        return True, message

primitive/lint/commands.py

@@ -1,17 +0,0 @@
-import click
-from pathlib import Path
-import typing
-from ..utils.printer import print_result
-
-if typing.TYPE_CHECKING:
-    from ..client import Primitive
-
-
-@click.command("lint")
-@click.pass_context
-@click.argument("source", type=click.Path(exists=True), default=".")
-def cli(context, source: str):
-    """Lint"""
-    primitive: Primitive = context.obj.get("PRIMITIVE")
-    result, message = primitive.lint.execute(source=Path(source))
-    print_result(message=message, context=context)

primitive/sim/actions.py

@@ -1,129 +0,0 @@
-import json
-import os
-import subprocess
-import xml.etree.ElementTree as ET
-from pathlib import Path, PurePath
-from typing import Tuple
-
-from loguru import logger
-from primitive_pal import process_vcd
-
-from primitive.utils.actions import BaseAction
-
-from ..utils.files import find_files_for_extension
-
-
-class Sim(BaseAction):
-    def execute(
-        self, source: Path = Path.cwd(), cmd: Tuple[str] = ["make"]
-    ) -> Tuple[bool, str]:
-        logger.debug(f"Starting simulation run for source: {source}")
-
-        os.chdir(source)
-        logger.debug(f"Changed to {source}, starting sim run")
-        try:
-            result = subprocess.run(cmd, capture_output=True, text=True, env=os.environ)
-        except FileNotFoundError:
-            message = f"Did not find {cmd}"
-            logger.error(message)
-            return False, message
-
-        logger.debug("Sim run complete.")
-
-        message = ""
-        if result.stderr:
-            logger.error("\n" + result.stderr)
-        if result.stdout:
-            logger.info("\n" + result.stdout)
-        message = "See above logs for sim output."
-
-        if result.returncode != 0:
-            if not self.primitive.DEBUG:
-                message = result.stderr
-            return False, message
-        else:
-            message = "Sim run successful."
-
-        return True, message
-
-    def collect_artifacts(self, source: Path, job_run_id: str) -> None:
-        # Parse VCD artifacts using rust binding
-        # TODO: eventually make this smarter, only parsing VCDs for failed tests
-        # For now we're uploading the last 10 MB of all VCD files
-        files = find_files_for_extension(source, ".vcd")
-        for file in files:
-            process_vcd(path=str(file.absolute()))
-
-        # Parse XML artifacts
-        files = find_files_for_extension(source, "results.xml")
-        for file in files:
-            self.parse_xml(path=file)
-
-        logger.debug("Uploading additional artifacts...")
-        # TODO: Figure out how to track ".log", ".history" files w/ analog stuff involved
-        file_ids = []
-        files = find_files_for_extension(
-            source,  # ("results.xml", ".vcd", ".vcd.json", ".xml.json")
-            ("results.xml", ".xml.json", ".vcd.json", ".vcd.parquet"),
-        )
-        for file_path in files:
-            try:
-                key_prefix = f"{job_run_id}/{str(PurePath(file_path).relative_to(Path(source)).parent)}"
-                file_upload_response = self.primitive.files.upload_file_via_api(
-                    file_path, key_prefix=key_prefix
-                )
-                file_id = file_upload_response.json()["data"]["fileUpload"]["id"]
-                file_ids.append(file_id)
-            except FileNotFoundError:
-                logger.warning(f"{file_path} not found...")
-
-        logger.debug("Updating job run...")
-        if len(file_ids) > 0:
-            job_run_update_result = self.primitive.jobs.job_run_update(
-                id=job_run_id, file_ids=file_ids
-            )
-            logger.success(job_run_update_result.data)
-
-    def parse_xml(self, path: Path) -> None:
-        results = ET.parse(path)
-        testsuites = results.getroot()
-
-        parsed_results = {}
-        testsuites_name = testsuites.attrib["name"]
-        parsed_results[testsuites_name] = {}
-
-        for testsuite in testsuites.findall("testsuite"):
-            testsuite_name = testsuite.attrib["name"]
-            parsed_results[testsuites_name][testsuite_name] = {
-                "properties": {},
-                "testcases": {},
-            }
-            props = parsed_results[testsuites_name][testsuite_name]["properties"]
-            testcases = parsed_results[testsuites_name][testsuite_name]["testcases"]
-
-            for prop in testsuite.findall("property"):
-                props[prop.attrib["name"]] = prop.attrib["value"]
-
-            for testcase in testsuite.findall("testcase"):
-                testcases[testcase.attrib["name"]] = {
-                    attr_key: attr_val for attr_key, attr_val in testcase.attrib.items()
-                }
-
-                failures = testcase.findall("failure")
-
-                if len(failures) > 0:
-                    for failure in failures:
-                        testcases[testcase.attrib["name"]]["status"] = {
-                            "conclusion": "failure",
-                            "message": failure.attrib["message"],
-                        }
-                else:
-                    testcases[testcase.attrib["name"]]["status"] = {
-                        "conclusion": "success",
-                        "message": "",
-                    }
-
-        # Write parsed file
-        data_path = path.parent / f"{path.name}.json"
-        with open(data_path, "w") as f:
-            f.write(json.dumps(parsed_results))

primitive/sim/commands.py

@@ -1,19 +0,0 @@
-import click
-from pathlib import Path
-import typing
-from typing import Tuple
-from ..utils.printer import print_result
-
-if typing.TYPE_CHECKING:
-    from ..client import Primitive
-
-
-@click.command("sim")
-@click.pass_context
-@click.argument("cmd", nargs=-1, required=True)
-@click.option("--source", type=click.Path(exists=True), default=".")
-def cli(context, source: str, cmd: Tuple[str]) -> None:
-    """Sim"""
-    primitive: Primitive = context.obj.get("PRIMITIVE")
-    result, message = primitive.sim.execute(source=Path(source), cmd=cmd)
-    print_result(message=message, context=context)