visidata 2.11.1__py3-none-any.whl → 3.0.1__py3-none-any.whl

visidata/utils.py

@@ -1,10 +1,11 @@
+from contextlib import contextmanager
 import operator
 import string
 import re
 
 'Various helper classes and functions.'
 
-__all__ = ['AlwaysDict', 'AttrDict', 'moveListItem', 'namedlist', 'classproperty', 'cleanName', 'MissingAttrFormatter']
+__all__ = ['AlwaysDict', 'AttrDict', 'DefaultAttrDict', 'moveListItem', 'namedlist', 'classproperty', 'MissingAttrFormatter', 'getitem', 'setitem', 'getitemdef', 'getitemdeep', 'setitemdeep', 'getattrdeep', 'setattrdeep', 'ExplodingMock', 'ScopedSetattr']
 
 
 class AlwaysDict(dict):
@@ -36,6 +37,24 @@ class AttrDict(dict):
         return self.keys()
 
 
+class DefaultAttrDict(dict):
+    'Augment a dict with more convenient .attr syntax.  not-present keys store new DefaultAttrDict.  like a recursive defaultdict.'
+    def __getattr__(self, k):
+        if k not in self:
+            if k.startswith("__"):
+                raise AttributeError from e
+            self[k] = DefaultAttrDict()
+        return self[k]
+
+    def __setattr__(self, k, v):
+        self[k] = v
+
+    def __dir__(self):
+        return self.keys()
+
+
+
+
 class classproperty(property):
     def __get__(self, cls, obj):
         return classmethod(self.fget).__get__(None, obj or cls)()
@@ -50,34 +69,75 @@ def moveListItem(L, fromidx, toidx):
     return toidx
 
 
-def cleanName(s):
-    s = re.sub(r'[^\w\d_]', '_', s)  # replace non-alphanum chars with _
-    s = re.sub(r'_+', '_', s)  # replace runs of _ with a single _
-    s = s.strip('_')
-    return s
+def setitem(r, i, v):  # function needed for use in lambda
+    r[i] = v
+    return True
+
+def getitem(o, k, default=None):
+    return default if o is None else o[k]
+
+def getitemdef(o, k, default=None):
+    try:
+        return default if o is None else o[k]
+    except Exception:
+        return default
+
 
+def getattrdeep(obj, attr, *default, getter=getattr):
+    try:
+        'Return dotted attr (like "a.b.c") from obj, or default if any of the components are missing.'
+        if not isinstance(attr, str):
+            return getter(obj, attr, *default)
+
+        try:  # if attribute exists, return toplevel value, even if dotted
+            if attr in obj:
+                return getter(obj, attr)
+        except RecursionError:  #1696
+            raise
+        except Exception as e:
+            pass
+
+        attrs = attr.split('.')
+        for a in attrs[:-1]:
+            obj = getter(obj, a)
+
+        return getter(obj, attrs[-1])
+    except Exception as e:
+        if not default: raise
+        return default[0]
 
-class OnExit:
-    '"with OnExit(func, ...):" calls func(...) when the context is exited'
-    def __init__(self, func, *args, **kwargs):
-        self.func = func
-        self.args = args
-        self.kwargs = kwargs
 
-    def __enter__(self):
-        return self
+def setattrdeep(obj, attr, val, getter=getattr, setter=setattr):
+    'Set dotted attr (like "a.b.c") on obj to val.'
+    if not isinstance(attr, str):
+        return setter(obj, attr, val)
 
-    def __exit__(self, exc_type, exc_value, exc_traceback):
+    try:  # if attribute exists, overwrite toplevel value, even if dotted
+        getter(obj, attr)
+        return setter(obj, attr, val)
+    except Exception as e:
+        pass
+
+    attrs = attr.split('.')
+    for a in attrs[:-1]:
         try:
-            self.func(*self.args, **self.kwargs)
+            obj = getter(obj, a)
         except Exception as e:
-            vd.exceptionCaught(e)
+            obj = obj[a] = type(obj)()  # assume homogeneous nesting
+
+    setter(obj, attrs[-1], val)
 
 
-def itemsetter(i):
-    def g(obj, v):
-        obj[i] = v
-    return g
+def getitemdeep(obj, k, *default):
+    if not isinstance(k, str):
+        try:
+            return obj[k]
+        except IndexError:
+            pass
+    return getattrdeep(obj, k, *default, getter=getitem)
+
+def setitemdeep(obj, k, val):
+    return setattrdeep(obj, k, val, getter=getitemdef, setter=setitem)
 
 
 def namedlist(objname, fieldnames):
@@ -111,11 +171,24 @@ def namedlist(objname, fieldnames):
 
     return NamedListTemplate
 
+
+class ExplodingMock:
+    'A mock object that raises an exception for everything except conversion to True/False.'
+    def __init__(self, msg):
+        self.__msg = msg
+
+    def __getattr__(self, k):
+        raise Exception(self.__msg)
+
+    def __bool__(self):
+        return False
+
+
 class MissingAttrFormatter(string.Formatter):
     "formats {} fields with `''`, that would normally result in a raised KeyError or AttributeError; intended for user customisable format strings."
-    def get_field(self, field_name, *args, **kwargs):
+    def get_field(self, field_name, args, kwargs):
         try:
-            return super().get_field(field_name, *args, **kwargs)
+            return super().get_field(field_name, args, kwargs)
         except (KeyError, AttributeError):
             return (None, field_name)
 
@@ -126,3 +199,13 @@ class MissingAttrFormatter(string.Formatter):
         elif not value:
             return str(value)
         return super().format_field(value, format_spec)
+
+
+@contextmanager
+def ScopedSetattr(obj, attrname, val):
+    oldval = getattr(obj, attrname)
+    try:
+        setattr(obj, attrname, val)
+        yield
+    finally:
+        setattr(obj, attrname, oldval)

visidata/vdobj.py

@@ -1,5 +1,4 @@
 from functools import wraps
-from unittest import mock
 import curses
 
 import visidata
@@ -25,7 +24,7 @@ def asyncthread(func):
 
 
 class VisiData(visidata.Extensible):
-    allPrefixes = ['g', 'z', ESC]  # embig'g'en, 'z'mallify, ESC=Alt/Meta
+    allPrefixes = ['g', 'z', 'Alt+']  # embig'g'en, 'z'mallify, Alt/Esc=User
 
     @classmethod
     def global_api(cls, func):
@@ -42,9 +41,16 @@ class VisiData(visidata.Extensible):
         self.lastErrors = []
         self.pendingKeys = []
         self.keystrokes = ''
-        self.scrFull = mock.MagicMock(__bool__=mock.Mock(return_value=False))  # disable curses in batch mode
+        self.scrFull = None
         self._cmdlog = None
+        self.currentReplay = None
         self.contexts = [self]  # objects whose attributes are in the fallback context for eval/exec.
+        self.importingModule = None
+        self.importedModules = []
+
+    @property
+    def cursesEnabled(self):
+        return bool(self.scrFull)
 
     def sheetstack(self, pane=0):
         'Return list of sheets in given *pane*. pane=0 is the active pane.  pane=-1 is the inactive pane.'
@@ -74,10 +80,6 @@ class VisiData(visidata.Extensible):
     def activeStack(self):
         return self.sheetstack() or self.sheetstack(-1)
 
-    @visidata.drawcache_property
-    def mousereg(self):
-        return []
-
     def __copy__(self):
         'Dummy method for Extensible.init()'
         pass
@@ -100,12 +102,29 @@ class VisiData(visidata.Extensible):
         'Invalidate internal caches between command inputs.'
         visidata.Extensible.clear_all_caches()
 
+    def resetVisiData(self):
+        self.clearCaches()  # we want vd to return a new VisiData object for each command
+        vd = visidata.vd  # get the new vd
+        vd.cmdlog.rows = []
+        vd.sheets = []
+        vd.allSheets = []
+        return vd
+
+    def get_wch(self, scr):
+        try:
+            return scr.get_wch()
+        except AttributeError:  #192 some packages don't have wide chars
+            k = scr.getch()
+            if k == -1:  # mimic get_wch behavior
+                raise curses.error('no char ready')
+            return k
+
     def drainPendingKeys(self, scr):
         '''Call scr.get_wch() until no more keypresses are available.  Return True if any keypresses are pending.'''
         scr.timeout(0)
         try:
             while True:
-                k = scr.get_wch()
+                k = self.get_wch(scr)
                 if k:
                     self.pendingKeys.append(k)
                 else:
@@ -127,7 +146,7 @@ class VisiData(visidata.Extensible):
             curses.reset_prog_mode()  #1347
             try:
                 scr.refresh()
-                k = scr.get_wch()
+                k = self.get_wch(scr)
                 vs = vs or self.activeSheet
                 if vs:
                     self.drawRightStatus(vs._scr, vs) # continue to display progress %
@@ -140,14 +159,6 @@ class VisiData(visidata.Extensible):
             k = ord(k)
         return curses.keyname(k).decode('utf-8')
 
-    def onMouse(self, scr, y, x, h, w, **kwargs):
-        self.mousereg.append((scr, y, x, h, w, kwargs))
-
-    def getMouse(self, _scr, _x, _y, button):
-        for scr, y, x, h, w, kwargs in self.mousereg[::-1]:
-            if scr is _scr and x <= _x < x+w and y <= _y < y+h and button in kwargs:
-                return kwargs[button]
-
     @property
     def screenHeight(self):
         return self.scrFull.getmaxyx()[0] if self.scrFull else 25

visidata/windows.py

@@ -0,0 +1,10 @@
+'''
+Windows-specific code
+'''
+
+import string
+from visidata import vd
+
+
+for c in string.ascii_uppercase:
+    vd.bindkey('ALT_'+c , 'Alt+'+c.lower())  #1630

visidata/wrappers.py

@@ -5,9 +5,6 @@ import functools
 
 from visidata import options, stacktrace, BaseSheet, vd
 
-__all__ = ['forward', 'wrmap', 'wrapply', 'TypedWrapper', 'TypedExceptionWrapper']
-
-vd.option('null_value', None, 'a value to be counted as null', replay=True)
 
 @BaseSheet.api
 def isNullFunc(sheet):
@@ -109,3 +106,12 @@ def wrapply(func, *args, **kwargs):
     except Exception as e:
         e.stacktrace = stacktrace()
         return TypedExceptionWrapper(func, *args, exception=e)
+
+
+vd.addGlobals(
+    forward=forward,
+    wrmap=wrmap,
+    wrapply=wrapply,
+    TypedWrapper=TypedWrapper,
+    TypedExceptionWrapper=TypedExceptionWrapper
+)

visidata-3.0.1.data/data/share/applications/visidata.desktop

@@ -0,0 +1,7 @@
+[Desktop Entry]
+Type=Application
+Terminal=true
+Name=VisiData
+Icon=utilities-terminal
+Exec=bash -i -c 'vd %f'
+Categories=Application;