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

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

@@ -788,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.16.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.16
+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.16.dist-info → math_spec_mapping-0.3.17.dist-info}/RECORD

@@ -1,4 +1,4 @@
-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
@@ -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=LCc9pbph0vcDvavCAWt1Z4uF2whnS8sVF87Aiyv4o_M,27592
+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.16.dist-info/LICENSE,sha256=ObyEzSw8kgCaFbEfpu1zP4TrcAKLA0xhqHMZZfyh7N0,1069
-math_spec_mapping-0.3.16.dist-info/METADATA,sha256=Bm-QWdPR6T4sJDJpc8omA2pKf8Zd_QdEOtwYSK9wHNc,6498
-math_spec_mapping-0.3.16.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-math_spec_mapping-0.3.16.dist-info/top_level.txt,sha256=AImhn9wgazkdV0a9vfiphtQR8uGe2nq-ZIOp-6yUk9o,18
-math_spec_mapping-0.3.16.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,,