dialoghelper 0.0.31__tar.gz → 0.0.32__tar.gz

{dialoghelper-0.0.31/dialoghelper.egg-info → dialoghelper-0.0.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.31
+Version: 0.0.32
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -23,6 +23,8 @@ Requires-Dist: fastlite
 Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
 Requires-Dist: claudette
+Requires-Dist: ast-grep-cli
+Requires-Dist: ast-grep-py
 Provides-Extra: dev
 Requires-Dist: python-fasthtml; extra == "dev"
 Dynamic: author

dialoghelper-0.0.32/dialoghelper/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "0.0.32"
+from .core import *

{dialoghelper-0.0.31 → dialoghelper-0.0.32}/dialoghelper/_modidx.py

@@ -9,6 +9,8 @@ d = { 'settings': { 'branch': 'main',
                                    'dialoghelper.core._umsg': ('core.html#_umsg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.add_html': ('core.html#add_html', 'dialoghelper/core.py'),
                                    'dialoghelper.core.add_msg': ('core.html#add_msg', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.ast_grep': ('core.html#ast_grep', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.ast_py': ('core.html#ast_py', 'dialoghelper/core.py'),
                                    'dialoghelper.core.call_endp': ('core.html#call_endp', 'dialoghelper/core.py'),
                                    'dialoghelper.core.curr_dialog': ('core.html#curr_dialog', 'dialoghelper/core.py'),
                                    'dialoghelper.core.del_msg': ('core.html#del_msg', 'dialoghelper/core.py'),

{dialoghelper-0.0.31 → dialoghelper-0.0.32}/dialoghelper/core.py

@@ -2,8 +2,9 @@
 
 # %% auto 0
 __all__ = ['Placements', 'empty', 'find_var', 'call_endp', 'find_dname', 'find_msg_id', 'curr_dialog', 'find_msgs', 'msg_idx',
-           'read_msg', 'add_html', 'del_msg', 'run_msg', 'add_msg', 'update_msg', 'url2note', 'load_gist', 'gist_file',
-           'import_string', 'is_usable_tool', 'mk_toollist', 'import_gist', 'tool_info', 'asdict']
+           'read_msg', 'add_html', 'run_msg', 'add_msg', 'del_msg', 'update_msg', 'url2note', 'ast_py', 'ast_grep',
+           'load_gist', 'gist_file', 'import_string', 'is_usable_tool', 'mk_toollist', 'import_gist', 'tool_info',
+           'asdict']
 
 # %% ../nbs/00_core.ipynb
 import json, importlib, linecache
@@ -106,14 +107,6 @@ def add_html(
     "Send HTML to the browser to be swapped into the DOM"
     call_endp('add_html_', dname, content=to_xml(content))
 
-# %% ../nbs/00_core.ipynb
-def del_msg(
-    msgid:str=None, # id of message to delete
-    dname:str='' # Running dialog to get info for; defaults to current dialog
-):
-    "Delete a message from the dialog."
-    call_endp('rm_msg_', dname, raiseex=True, msid=msgid)
-
 # %% ../nbs/00_core.ipynb
 def run_msg(
     msgid:str=None, # id of message to execute
@@ -148,6 +141,14 @@ def add_msg(
         time_run=time_run, is_exported=is_exported, skipped=skipped, pinned=pinned,
         i_collapsed=i_collapsed, o_collapsed=o_collapsed, heading_collapsed=heading_collapsed)
 
+# %% ../nbs/00_core.ipynb
+def del_msg(
+    msgid:str=None, # id of message to delete
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Delete a message from the dialog."
+    call_endp('rm_msg_', dname, raiseex=True, msid=msgid)
+
 # %% ../nbs/00_core.ipynb
 @delegates(add_msg)
 def _add_msg_unsafe(
@@ -155,12 +156,12 @@ def _add_msg_unsafe(
     placement:str='add_after', # Can be 'add_after', 'add_before', 'at_start', 'at_end'
     msgid:str=None, # id of message that placement is relative to (if None, uses current message)
     run:bool=False, # For prompts, send it to the AI; for code, execute it (*DANGEROUS -- be careful of what you run!)
+    dname:str='', # Running dialog to get info for; defaults to current dialog
     **kwargs
 ):
     """Add/update a message to the queue to show after code execution completes, and optionally run it. Be sure to pass a `sid` (stable id) not a `mid` (which is used only for sorting, and can change).
     *WARNING*--This can execute arbitrary code, so check carefully what you run!--*WARNING"""
     if placement not in ('at_start','at_end') and not msgid: msgid = find_msg_id()
-    dname = kwargs.pop('dname')
     return call_endp(
         'add_relative_', dname, content=content, placement=placement, msgid=msgid, run=run, **kwargs)
 
@@ -201,6 +202,25 @@ def url2note(
     res = read_url(url, as_md=True, extract_section=extract_section, selector=selector)
     return add_msg(res)
 
+# %% ../nbs/00_core.ipynb
+def ast_py(code:str):
+    "Get an SgRoot root node for python `code`"
+    from ast_grep_py import SgRoot
+    return SgRoot(code, "python").root()
+
+# %% ../nbs/00_core.ipynb
+def ast_grep(
+    pattern:str, # ast-grep pattern to search
+    path=".", # path to recursively search for files
+    lang="python" # language to search/scan
+):
+    "Use the `ast-grep` command to find `pattern` in `path`"
+    import json, subprocess
+    cmd = f"ast-grep --pattern '{pattern}' --lang {lang} --json=compact"
+    if path != ".": cmd = f"cd {path} && {cmd}"
+    res = subprocess.run(cmd, shell=True, capture_output=True, text=True)
+    return json.loads(res.stdout) if res.stdout else res.stderr
+
 # %% ../nbs/00_core.ipynb
 def load_gist(gist_id:str):
     "Retrieve a gist"

{dialoghelper-0.0.31 → dialoghelper-0.0.32/dialoghelper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.31
+Version: 0.0.32
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -23,6 +23,8 @@ Requires-Dist: fastlite
 Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
 Requires-Dist: claudette
+Requires-Dist: ast-grep-cli
+Requires-Dist: ast-grep-py
 Provides-Extra: dev
 Requires-Dist: python-fasthtml; extra == "dev"
 Dynamic: author

{dialoghelper-0.0.31 → dialoghelper-0.0.32}/dialoghelper.egg-info/requires.txt

@@ -3,6 +3,8 @@ fastlite
 ghapi
 ipykernel-helper
 claudette
+ast-grep-cli
+ast-grep-py
 
 [dev]
 python-fasthtml

{dialoghelper-0.0.31 → dialoghelper-0.0.32}/settings.ini

@@ -1,11 +1,11 @@
 [DEFAULT]
 repo = dialoghelper
 lib_name = dialoghelper
-version = 0.0.31
+version = 0.0.32
 min_python = 3.9
 license = apache2
 black_formatting = False
-requirements = fastcore>=1.8.5 fastlite ghapi ipykernel-helper claudette
+requirements = fastcore>=1.8.5 fastlite ghapi ipykernel-helper claudette ast-grep-cli ast-grep-py
 dev_requirements = python-fasthtml
 doc_path = _docs
 lib_path = dialoghelper

dialoghelper-0.0.31/dialoghelper/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "0.0.31"
-from .core import *