math-spec-mapping 0.3.15__py3-none-any.whl → 0.3.17__py3-none-any.whl

math_spec_mapping/Classes/Block.py

@@ -69,7 +69,7 @@ class Block:
 
         self.find_all_spaces_used(data)
 
-    def render_mermaid(self, i):
+    def render_mermaid(self, i, go_deep=True):
         i += 1
         out = 'X{}["{}"]'.format(i, self.name)
         if self.block_type == "Mechanism":
@@ -87,7 +87,7 @@ class Block:
         elif self.block_type in ["Parallel Block", "Stack Block", "Split Block"]:
             updates = self.all_updates
         else:
-            return "\n", {}
+            return "\n", {}, 0
         updates = sorted(updates, key=lambda x: x[0].name + "-" + x[1].name)
 
         out = "\n"
@@ -115,18 +115,30 @@ class Block:
             out += "\n"
         out += "end\n"
         out += "\n"
-        return out, entity_variable_mapping
+        return out, entity_variable_mapping, len(entities)
 
-    def render_mermaid_root(self):
+    def render_mermaid_root(self, go_deep=True):
         out = """```mermaid\ngraph TB\n"""
-        add, entity_variable_mapping = self.render_ending_entities()
+        add, entity_variable_mapping, n_entities = self.render_ending_entities()
         out += add
-        out += self.render_mermaid(0)[0]
+        new, i = self.render_mermaid(0, go_deep=go_deep)
+        out += new
 
         for key in entity_variable_mapping:
             out = out.replace(key, entity_variable_mapping[key])
-        out += "\n```"
+        out += "\n"
+        for x in range(1, i):
+            if (
+                "X{}[Domain]".format(x) not in out
+                and "X{}[Codomain]".format(x) not in out
+                and "subgraph X{}".format(x) not in out
+            ):
+                out += "class X{} internal-link;\n".format(x)
 
+        for x in range(n_entities):
+            out += "class EE{} internal-link;\n".format(x)
+
+        out += "\n```"
         return out
 
     def find_all_spaces_used(self, data):
@@ -223,7 +235,19 @@ class ParallelBlock(Block):
         self.metadata = data["metadata"]
         self.find_all_spaces_used(data)
 
-    def render_mermaid(self, i):
+    def render_mermaid(self, i, go_deep=True):
+        if go_deep == "First":
+            go_deep = False
+        elif go_deep == False:
+            i += 1
+            out = 'X{}["{}"]'.format(i, self.name)
+            for u in self.all_updates:
+                out += "\n"
+                out += "X{} --> {}".format(
+                    i,
+                    (u[0].name + "-" + u[1].name).replace(" ", "-"),
+                )
+            return out, i
         multi = None
         if type(i) == list:
             multi = i
@@ -251,7 +275,7 @@ class ParallelBlock(Block):
                 if x.name not in ["Empty Space", "Terminating Space"]
             ]
 
-            component, i = component.render_mermaid(i)
+            component, i = component.render_mermaid(i, go_deep=go_deep)
             out += component
             out += "\n"
             domain_map[i] = domain
@@ -404,7 +428,19 @@ class StackBlock(Block):
                     )
         return channels
 
-    def render_mermaid(self, i):
+    def render_mermaid(self, i, go_deep=True):
+        if go_deep == "First":
+            go_deep = False
+        elif go_deep == False:
+            i += 1
+            out = 'X{}["{}"]'.format(i, self.name)
+            for u in self.all_updates:
+                out += "\n"
+                out += "X{} --> {}".format(
+                    i,
+                    (u[0].name + "-" + u[1].name).replace(" ", "-"),
+                )
+            return out, i
         multi = None
         if type(i) == list:
             multi = i
@@ -423,7 +459,7 @@ class StackBlock(Block):
                 for x in domain
                 if x.name not in ["Empty Space", "Terminating Space"]
             ]
-            component, i = component.render_mermaid(i)
+            component, i = component.render_mermaid(i, go_deep=go_deep)
             domain_map[i] = domain
             out += component
             out += "\n"
@@ -535,7 +571,7 @@ class SplitBlock(Block):
         self.metadata = data["metadata"]
         self.find_all_spaces_used(data)
 
-    def render_mermaid(self, i):
+    def render_mermaid(self, i, go_deep=True):
         multi = None
         if type(i) == list:
             multi = i

math_spec_mapping/Classes/MathSpec.py

@@ -8,7 +8,8 @@ import os
 from copy import deepcopy
 import shutil
 import pandas as pd
-from inspect import signature, getsource
+from inspect import signature, getsource, getfile
+from IPython.display import display, Markdown
 
 
 class MathSpec:
@@ -1116,13 +1117,34 @@ class MathSpecImplementation:
 
     def load_source_files(self):
         self.source_files = {}
-        for key in self.components:
+        self.file_names = {}
+        for key in self.blocks:
             self.source_files[key] = getsource(self.components[key])
+            self.file_names[key] = getfile(self.components[key])
 
-    def print_source_code_files(self, keys=None):
+    def print_source_code_files(self, keys=None, markdown=True):
         if not keys:
             keys = list(self.source_files.keys())
         for key in keys:
+            # Skip wirings
+            if key not in self.ms.components:
+                continue
             print("-" * 20 + key + "-" * 20)
-            print(self.source_files[key])
+            if markdown:
+                display(
+                    Markdown(
+                        """```python
+{}
+```""".format(
+                            self.source_files[key]
+                        )
+                    )
+                )
+            else:
+                print(self.source_files[key])
+            print("\n")
+            full_path = self.file_names[key]
+            print("File path: {}".format(full_path))
+            relative_path = "./" + os.path.relpath(full_path, os.getcwd())
+            print("Relative file path: {}".format(relative_path))
             print("\n\n\n")

math_spec_mapping/Convenience/__init__.py

@@ -1,3 +1,8 @@
 # from .documentation import write_top_level_json_description
 from .starter import remove_dummy_repo_components
-from .github import write_scaffold_to_github_issues
+from .github import (
+    write_scaffold_to_github_issues,
+    find_open_issues,
+    create_priority_label_matrix,
+    create_milestone_label_matrix,
+)

math_spec_mapping/Convenience/github.py

@@ -1,5 +1,8 @@
 import os
 import requests
+from github import Github, Auth
+import pandas as pd
+from dotenv import load_dotenv
 
 
 def write_scaffold_to_github_issues(
@@ -32,3 +35,74 @@ def write_scaffold_to_github_issues(
             headers=headers,
             json={"title": issue_name, "body": checklist},
         )
+
+
+def find_open_issues():
+    load_dotenv()
+    g = Github(auth=Auth.Token(os.getenv("GITHUB_TOKEN")))
+    repo = g.get_repo("BlockScience/MSML")
+    open_issues = repo.get_issues(state="open")
+    open_issues = list(open_issues)
+
+    df = pd.DataFrame(
+        [[x.title, x.labels, x.html_url, x.milestone] for x in open_issues],
+        columns=["Name", "Labels", "URL", "Milestone"],
+    )
+    df["Milestone"] = df["Milestone"].apply(lambda x: x.title if x else None)
+    df = df.join(
+        df["Labels"]
+        .apply(lambda x: {y.name: True for y in x})
+        .apply(pd.Series)
+        .fillna(False)
+    )
+
+    empty = df[df["Labels"].apply(lambda x: len(x)) == 0]
+    empty = empty[~empty["URL"].str.contains("MSML/pull")]
+    assert len(empty) == 0, "Some issues do not have labels"
+
+    return df, open_issues
+
+
+def create_priority_label_matrix(df, exclude_milestones=None):
+    if exclude_milestones:
+        df = df[~df["Milestone"].isin(exclude_milestones)]
+    priority_labels = ["High Priority", "Medium Priority", "Low Priority"]
+    labels = list(df.columns[4:])
+    labels = sorted([x for x in labels if x not in priority_labels])
+
+    table = []
+    for label in labels:
+        row = []
+        for priority in priority_labels:
+            row.append(df[(df[priority]) & (df[label])])
+        table.append(row)
+    table = pd.DataFrame(table, index=labels, columns=priority_labels)
+    table = table.applymap(
+        lambda y: "\n".join(
+            y.apply(lambda x: "[{}]({})".format(x["Name"], x["URL"]), axis=1).values
+        )
+    )
+    return table
+
+
+def create_milestone_label_matrix(df, exclude_milestones=None):
+    if exclude_milestones:
+        df = df[~df["Milestone"].isin(exclude_milestones)]
+    priority_labels = ["High Priority", "Medium Priority", "Low Priority"]
+    labels = list(df.columns[4:])
+    labels = sorted([x for x in labels if x not in priority_labels])
+    milestones = sorted(list(df[~pd.isnull(df["Milestone"])]["Milestone"].unique()))
+
+    table = []
+    for label in labels:
+        row = []
+        for milestone in milestones:
+            row.append(df[(df["Milestone"] == milestone) & (df[label])])
+        table.append(row)
+    table = pd.DataFrame(table, index=labels, columns=milestones)
+    table = table.applymap(
+        lambda y: "\n".join(
+            y.apply(lambda x: "[{}]({})".format(x["Name"], x["URL"]), axis=1).values
+        )
+    )
+    return table

math_spec_mapping/Reports/markdown.py

@@ -1,5 +1,19 @@
 import os
 from .state import write_state_section
+from inspect import signature, getsource, getfile
+
+
+def get_source_code(ms, component_type, implementation_name):
+    if implementation_name in ms.implementations["python"][component_type]:
+        code = ms.implementations["python"][component_type][implementation_name]
+    else:
+        return None
+    source_code = """```python
+{}```""".format(
+        getsource(code)
+    )
+    file_path = getfile(code)
+    return source_code, file_path
 
 
 def write_entity_markdown_report(ms, path, entity, add_metadata=True):
@@ -179,7 +193,23 @@ def write_boundary_action_markdown_report(ms, path, boundary_action, add_metadat
 
             out += "#### Logic\n"
             out += x.logic
-            out += "\n\n"
+            out += "\n"
+
+            temp = get_source_code(ms, "boundary_action_options", x.name)
+            if temp:
+                source_code, file_path = temp
+                file_path = os.path.relpath(
+                    file_path, "{}/Boundary Actions".format(path)
+                )
+                out += "#### Python Implementation\n"
+                out += source_code
+                out += "\n"
+                out += "Implementation Path (only works if vault is opened at level including the src folder): [{}]({})".format(
+                    file_path, file_path
+                )
+                out += "\n"
+
+            out += "\n"
 
     with open(
         "{}/Boundary Actions/{}.md".format(path, boundary_action.label), "w"
@@ -256,7 +286,21 @@ def write_policy_markdown_report(ms, path, policy, add_metadata=True):
 
             out += "#### Logic\n"
             out += x.logic
-            out += "\n\n"
+            out += "\n"
+
+            temp = get_source_code(ms, "policies", x.name)
+            if temp:
+                source_code, file_path = temp
+                file_path = os.path.relpath(file_path, "{}/Policies".format(path))
+                out += "#### Python Implementation\n"
+                out += source_code
+                out += "\n"
+                out += "Implementation Path (only works if vault is opened at level including the src folder): [{}]({})".format(
+                    file_path, file_path
+                )
+                out += "\n"
+
+            out += "\n"
 
     with open("{}/Policies/{}.md".format(path, policy.label), "w") as f:
         f.write(out)
@@ -323,6 +367,20 @@ def write_mechanism_markdown_report(ms, path, mechanism, add_metadata=True):
         )
         out += "\n"
 
+    temp = get_source_code(ms, "mechanisms", mechanism.name)
+    if temp:
+        source_code, file_path = temp
+        file_path = os.path.relpath(file_path, "{}/Mechanisms".format(path))
+        out += "## Python Implementation\n"
+        out += source_code
+        out += "\n"
+        out += "Implementation Path (only works if vault is opened at level including the src folder): [{}]({})".format(
+            file_path, file_path
+        )
+        out += "\n"
+
+    out += "\n"
+
     with open("{}/Mechanisms/{}.md".format(path, mechanism.label), "w") as f:
         f.write(out)
 
@@ -434,7 +492,23 @@ def write_control_action_markdown_report(ms, path, control_action, add_metadata=
 
             out += "#### Logic\n"
             out += x.logic
-            out += "\n\n"
+            out += "\n"
+
+            temp = get_source_code(ms, "control_action_options", x.name)
+            if temp:
+                source_code, file_path = temp
+                file_path = os.path.relpath(
+                    file_path, "{}/Control Actions".format(path)
+                )
+                out += "#### Python Implementation\n"
+                out += source_code
+                out += "\n"
+                out += "Implementation Path (only works if vault is opened at level including the src folder): [{}]({})".format(
+                    file_path, file_path
+                )
+                out += "\n"
+
+            out += "\n"
 
     with open("{}/Control Actions/{}.md".format(path, control_action.label), "w") as f:
         f.write(out)
@@ -454,6 +528,13 @@ def write_wiring_markdown_report(ms, path, wiring, add_metadata=True):
 """.format(
                 "\n".join(["{}: {}".format(x, metadata[x]) for x in metadata])
             )
+    out += "## Wiring Diagram (Zoomed Out)"
+    out += "\n"
+    out += "\n"
+    out += "- For display of only depth of 1 in the components/nested wirings\n"
+    out += wiring.render_mermaid_root(go_deep="First")
+    out += "\n"
+    out += "\n"
 
     out += "## Wiring Diagram"
     out += "\n"
@@ -707,6 +788,12 @@ def write_state_variables_markdown_reports(ms, path, state, add_metadata=True):
             out += variable.domain
         out += "\n\n"
 
+        out += "Updated By:\n"
+        for i, x in enumerate(sorted(variable.updated_by, key=lambda x: x.name)):
+            out += "{}. [[{}]]".format(i + 1, x.name)
+            out += "\n"
+        out += "\n"
+
         with open(
             "{}/State Variables/{}.md".format(
                 path, "{}-{}".format(state.name, variable.name)

math_spec_mapping/__init__.py

@@ -30,6 +30,12 @@ from .Reports import (
     write_parameter_table_markdown,
 )
 from .schema import schema
-from .Convenience import remove_dummy_repo_components, write_scaffold_to_github_issues
+from .Convenience import (
+    remove_dummy_repo_components,
+    write_scaffold_to_github_issues,
+    find_open_issues,
+    create_priority_label_matrix,
+    create_milestone_label_matrix,
+)
 
 # from .Convenience import write_top_level_json_description

{math_spec_mapping-0.3.15.dist-info → math_spec_mapping-0.3.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: math-spec-mapping
-Version: 0.3.15
+Version: 0.3.17
 Summary: A library for easy mapping of mathematical specifications.
 Author-email: Sean McOwen <Sean@Block.Science>
 Classifier: Programming Language :: Python :: 3

{math_spec_mapping-0.3.15.dist-info → math_spec_mapping-0.3.17.dist-info}/RECORD

@@ -1,12 +1,12 @@
-math_spec_mapping/__init__.py,sha256=LvuIbfILQf3UGyHympLJyncfnnpZpEyvN5UDOpi8bWw,1140
+math_spec_mapping/__init__.py,sha256=uuz8qXGzsumq0W4-iAIw2w-xxyI10PhbNlGGQ9MxR60,1244
 math_spec_mapping/schema.py,sha256=6mrRqzEnTTSXjb19xJ63MBp0KjKH0s7i6TfT4MkAY9k,233
 math_spec_mapping/schema.schema.json,sha256=hJP2TcV5WPFPmx4u_A5U1xtnpkE1LeYaTeYOXadTot0,30916
 math_spec_mapping/Classes/ActionTransmissionChannel.py,sha256=zWMo5QsgPh5WGIWXl-xOrZNMXYJXmK6Vejw1dQvi0og,246
-math_spec_mapping/Classes/Block.py,sha256=geaH3cNPJ-BOSj1pqccexFl6FqmolAFbrb7wpQQqaUE,18215
+math_spec_mapping/Classes/Block.py,sha256=x5eYzuWp6C_n1tUVMxECpdYi4w_Tv0NLyxg0m8pHjMo,19602
 math_spec_mapping/Classes/BoundaryAction.py,sha256=_rFvEZ4LNxmlM59js4SuQ9n5CgVmITw4YWKs0-q0r-4,560
 math_spec_mapping/Classes/ControlAction.py,sha256=4AzMSA8fbnUf-fGlvMJXHJFbz32G1h1QVWf2yzrXrLA,493
 math_spec_mapping/Classes/Entity.py,sha256=fA0-b128_OHHxfCg4pzqyQV083EYev1HlVpy86S5igg,1226
-math_spec_mapping/Classes/MathSpec.py,sha256=94cJsKpdcBj_jzlQhNYnNBJQyBVOI-RrNHuWHdbieUs,42637
+math_spec_mapping/Classes/MathSpec.py,sha256=jOWfO5mx_7RjAy4SF9Faiw-tGEhL2HeyzV9lKk0pVDQ,43432
 math_spec_mapping/Classes/Mechanism.py,sha256=2sLm3wYBIeTQaMBcsJ9btqIWsbS895Ra8NY6Y9_G_Dg,379
 math_spec_mapping/Classes/Metric.py,sha256=iQhH4g8Z60QlM22nPX9ytGmidOsZSiSs0KjqwmsScwU,636
 math_spec_mapping/Classes/Parameter.py,sha256=ZuJ_w0sChvRElJ4sOnXZ2EV4Ell2xXFulKLjVOpgz2E,1237
@@ -17,9 +17,9 @@ math_spec_mapping/Classes/StateUpdateTransmissionChannel.py,sha256=3hBLvD1lE64Pk
 math_spec_mapping/Classes/StatefulMetric.py,sha256=plMFMAFEk1y2t4DR5lA2SRC9UrYArsx_W33l3mZSdgE,804
 math_spec_mapping/Classes/Type.py,sha256=2KFY8d3cv1PzJJ7SSMHJf1zcfQ3ZbqxotK2KgTaLZdM,289
 math_spec_mapping/Classes/__init__.py,sha256=0zxgOqns_9JybD74HKMVh6aw8ij8WVbfQ4Q_1uWzof0,761
-math_spec_mapping/Convenience/__init__.py,sha256=P8WgrbpJkH08GeyOdhF-ITfT4Yd4ny8QJEvZK0fSljc,164
+math_spec_mapping/Convenience/__init__.py,sha256=BEZr1rXohKAknOa3wjNVx3EvSVxDcfftJsvtSw5mvmQ,264
 math_spec_mapping/Convenience/documentation.py,sha256=1ziWVJznbCUxeAAt03nAdEYtMlXNo5TeedHfgs0vSBU,1625
-math_spec_mapping/Convenience/github.py,sha256=mMOhIH3Nvh_91fNTOPjz1oQq1iDq2m5hNLMhtcrr5FU,1025
+math_spec_mapping/Convenience/github.py,sha256=Wddr1FCxcIxaFeiKyvx3mvVDbzjcgTxs4rU8ulc4qXk,3539
 math_spec_mapping/Convenience/starter.py,sha256=9dBKU3EHscut2wA6Nx1XUmehrFIdr4dCjP9V1BbTF6s,12567
 math_spec_mapping/Load/__init__.py,sha256=_ga5nHi7U5rY5lCF36_XI9Qmybq4P8R4m5I5mmjLBk8,33
 math_spec_mapping/Load/action_transmission_channel.py,sha256=9Wer7g2s5SSOcUYuZ0PqSKUVVnW3EvGQJZNXJTwW__0,2561
@@ -45,7 +45,7 @@ math_spec_mapping/Reports/boundary_actions.py,sha256=45BPp4QjWdD-3E9ZWwqgj_nI2-Y
 math_spec_mapping/Reports/control_actions.py,sha256=NksekZKIPFSIkubttFstKFthc5AU9B9PWRLSl9j1wWs,1216
 math_spec_mapping/Reports/general.py,sha256=WOOn6Wlb8M4fsdN49FlKLwOka6vJPQ9aCUy88TL2ki0,1610
 math_spec_mapping/Reports/html.py,sha256=RpyQQON8pDnMmfcyxOsm8UAD5Ad8VKnnW-OyOarTmzA,9711
-math_spec_mapping/Reports/markdown.py,sha256=oLammMJMkWhZXmODKzCuZDdJpiJIs5P-84gVsYhs0Rw,24605
+math_spec_mapping/Reports/markdown.py,sha256=DG0DmzE6Pas5dLLjbSnGa0xmWSFz0IGLa0sT_1MgYvI,27804
 math_spec_mapping/Reports/mechanisms.py,sha256=d2Rxt3JBYvqAOAYUynl0buYVoXEHrO8EGq7GK6hK8NA,1322
 math_spec_mapping/Reports/node_map.py,sha256=FdSMDQG16NX6n9sZcH-T5xwsvgjrV9OqBHc9J_VlNK0,3129
 math_spec_mapping/Reports/parameters.py,sha256=-ucL71lolqU0xvV7yb0sXl4pFMRl5tXNWdoBfUjLOaQ,1944
@@ -54,8 +54,8 @@ math_spec_mapping/Reports/spaces.py,sha256=-76hR5wQBv4lsG000ypBJ-OprjsNjI-rNRMYd
 math_spec_mapping/Reports/state.py,sha256=QYeCvX5cHeZBrbvMeDsTqJcUDTuDFJSLvPbasjLspk8,3643
 math_spec_mapping/Reports/tables.py,sha256=O0CNuqh3LMECq5uLjBOoxMUk5hUvkUK660FNnwWUxDY,1505
 math_spec_mapping/Reports/wiring.py,sha256=u9SvKWy6T-WJUEgFI6-zgZanoOaTTs_2YwmEceDLsV8,1618
-math_spec_mapping-0.3.15.dist-info/LICENSE,sha256=ObyEzSw8kgCaFbEfpu1zP4TrcAKLA0xhqHMZZfyh7N0,1069
-math_spec_mapping-0.3.15.dist-info/METADATA,sha256=H3FF0wDPZ2BUneG-JTOGfaGh8Z24tr2tgLlfOqe67Y4,6498
-math_spec_mapping-0.3.15.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-math_spec_mapping-0.3.15.dist-info/top_level.txt,sha256=AImhn9wgazkdV0a9vfiphtQR8uGe2nq-ZIOp-6yUk9o,18
-math_spec_mapping-0.3.15.dist-info/RECORD,,
+math_spec_mapping-0.3.17.dist-info/LICENSE,sha256=ObyEzSw8kgCaFbEfpu1zP4TrcAKLA0xhqHMZZfyh7N0,1069
+math_spec_mapping-0.3.17.dist-info/METADATA,sha256=niURiLlRa8ejIvpDjiOcvg1jFeanlxz-76-Mb1joMsU,6498
+math_spec_mapping-0.3.17.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+math_spec_mapping-0.3.17.dist-info/top_level.txt,sha256=AImhn9wgazkdV0a9vfiphtQR8uGe2nq-ZIOp-6yUk9o,18
+math_spec_mapping-0.3.17.dist-info/RECORD,,