pyDiffTools 0.1.8__py3-none-any.whl

pydifftools/flowchart/watch_graph.py

@@ -0,0 +1,168 @@
+import subprocess
+import time
+import shutil
+from pathlib import Path
+from watchdog.events import FileSystemEventHandler
+from watchdog.observers import Observer
+from pydifftools.command_registry import register_command
+from .graph import write_dot_from_yaml
+
+
+def _reload_svg(driver, svg_file: Path) -> None:
+    """Refresh the embedded SVG while preserving zoom and scroll."""
+    zoom = driver.execute_script("return window.visualViewport.scale")
+    scroll_x = driver.execute_script("return window.scrollX")
+    scroll_y = driver.execute_script("return window.scrollY")
+    svg_uri = svg_file.resolve().as_uri() + f"?t={time.time()}"
+    driver.execute_async_script(
+        "const [src,z,x,y,done]=arguments;const"
+        " s=document.getElementById('svg-view');s.onload=function()"
+        "{document.body.style.zoom=z;"
+        " window.scrollTo(x,y); done();};s.setAttribute('src', src);",
+        svg_uri,
+        zoom,
+        scroll_x,
+        scroll_y,
+    )
+
+
+def build_graph(
+    yaml_file,
+    dot_file,
+    svg_file,
+    wrap_width,
+    order_by_date=False,
+    prev_data=None,
+):
+    # Graphviz is required for dot -> svg rendering.
+    if shutil.which("dot") is None:
+        raise RuntimeError(
+            "Graphviz is required to render flowcharts. Install it so the"
+            " 'dot' executable is available on your PATH."
+        )
+    data = write_dot_from_yaml(
+        str(yaml_file),
+        str(dot_file),
+        wrap_width=wrap_width,
+        order_by_date=order_by_date,
+        old_data=prev_data,
+        validate_due_dates=True,
+    )
+    subprocess.run(
+        ["dot", "-Tsvg", str(dot_file), "-o", str(svg_file)],
+        check=True,
+    )
+    return data
+
+
+class GraphEventHandler(FileSystemEventHandler):
+    def __init__(
+        self,
+        yaml_file,
+        dot_file,
+        svg_file,
+        driver,
+        wrap_width,
+        data,
+        order_by_date=False,
+        debounce=0.25,
+    ):
+        self.yaml_file = Path(yaml_file)
+        self.dot_file = Path(dot_file)
+        self.svg_file = Path(svg_file)
+        self.driver = driver
+        self.wrap_width = wrap_width
+        self.data = data
+        self.order_by_date = order_by_date
+        self.debounce = debounce
+        self._last_handled = 0.0
+        self._last_mtime = None
+
+    def on_modified(self, event):
+        if Path(event.src_path) == self.yaml_file:
+            mtime = self.yaml_file.stat().st_mtime
+            if self._last_mtime is not None and mtime == self._last_mtime:
+                return
+            now = time.time()
+            if now - self._last_handled < self.debounce:
+                return
+            self._last_handled = now
+            self.data = build_graph(
+                self.yaml_file,
+                self.dot_file,
+                self.svg_file,
+                self.wrap_width,
+                self.order_by_date,
+                self.data,
+            )
+            _reload_svg(self.driver, self.svg_file)
+            self._last_mtime = self.yaml_file.stat().st_mtime
+
+
+@register_command(
+    "Watch a flowchart YAML file, rebuild DOT/SVG output, and open the"
+    " preview",
+    help={
+        "yaml": "Path to the flowchart YAML file",
+        "wrap_width": "Line wrap width used when generating node labels",
+        "d": "Render nodes by date without showing connections",
+    },
+)
+def wgrph(yaml, wrap_width=55, d=False):
+    # Selenium is only required when actually launching the watcher, so it is
+    # imported here to avoid breaking the command-line tools when the optional
+    # dependency is not installed.
+    try:
+        from selenium import webdriver
+        from selenium.webdriver.chrome.options import Options
+        from selenium.common.exceptions import (
+            WebDriverException,
+            NoSuchWindowException,
+        )
+    except ImportError as exc:
+        raise ImportError(
+            "The 'watch_graph' command requires the 'selenium' package to be"
+            " installed."
+        ) from exc
+
+    yaml_file = Path(yaml)
+    if not yaml_file.exists():
+        raise FileNotFoundError(f"YAML file not found: {yaml_file}")
+
+    dot_file = yaml_file.with_suffix(".dot")
+    svg_file = yaml_file.with_suffix(".svg")
+    html_file = yaml_file.with_suffix(".html")
+
+    # Use date ordering when requested so boxes appear in calendar order.
+    data = build_graph(yaml_file, dot_file, svg_file, wrap_width, d)
+    html_file.write_text(
+        "<html><body style='margin:0'><embed id='svg-view'"
+        " type='image/svg+xml'"
+        f" src='{svg_file.name}?t={time.time()}'/></body></html>"
+    )
+    options = Options()
+    driver = webdriver.Chrome(options=options)
+    driver.get(html_file.resolve().as_uri())
+    event_handler = GraphEventHandler(
+        yaml_file, dot_file, svg_file, driver, wrap_width, data, d
+    )
+    observer = Observer()
+    observer.schedule(event_handler, yaml_file.parent, recursive=False)
+    observer.start()
+    try:
+        while True:
+            try:
+                _ = driver.window_handles
+                driver.execute_script("return 1")
+            except (NoSuchWindowException, WebDriverException):
+                break
+            time.sleep(1)
+    except KeyboardInterrupt:
+        pass
+    finally:
+        observer.stop()
+        observer.join()
+        try:
+            driver.quit()
+        except Exception:
+            pass

pydifftools/html_comments.py

@@ -0,0 +1,33 @@
+# again rerun
+from lxml import html, etree
+import os
+from pyspecdata.fornotebook import *
+from pyspecdata import *
+import re
+
+fp = open(sys.argv[1], "r")
+content = fp.read()
+fp.close()
+doc = html.fromstring(content)
+commentlabel_re = re.compile(r"\[([A-Z]+)([0-9])\]")
+comment_dict = {}
+# for j in doc.xpath('descendant::*[@style="mso-element:comment"]'):
+newlist = []
+thisbody = doc.find("body")
+print("I found the body", lsafen(thisbody))
+commentlist = etree.Element("div", style="mso-element:comment-list")
+for j in doc.xpath('//span[@style="mso-element:comment"]'):
+    # for j in doc.xpath('//span[@style="mso-element:comment"]'):
+    # print 'found span with style:\n\n',lsafen(html.tostring(j),wrap = 60)
+    # if j.attrib['style'] == 'mso-element:comment':
+    print("found span with style:\n\n", lsafen(j.attrib, wrap=60))
+    newlist.append(j)
+    j.drop_tree()
+    commentlist.append(j)
+thisbody.append(commentlist)
+# print lsafen(map(html.tostring,newlist),wrap = 60)
+newfile = re.sub(r"(.*)(\.htm.*)", r"\1_htmlcomm\2", sys.argv[1])
+fp = open(newfile, "w")
+content = html.tostring(doc)
+fp.write(content)
+fp.close()