master > master: src

src/thirdparty/misc.py

@@ -7,9 +7,62 @@
 
 from datetime import datetime;
 from datetime import timedelta;
+from functools import wraps;
 import lorem;
 import re;
-from textwrap import dedent;
+from textwrap import dedent as textwrap_dedent;
+from typing import Callable;
+from typing import TypeVar;
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+# MODIFICATIONS
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+def prestrip(first: bool = True, last: bool = True, all: bool = False):
+    '''
+    Returns a decorator that modifies string -> string methods
+    '''
+    T = TypeVar('T');
+    def dec(method: Callable[[str], T]) -> Callable[[str], T]:
+        '''
+        Performs method but first strips initial/final (empty) lines.
+        '''
+        @wraps(method)
+        def wrapped_method(text: str) -> T:
+            lines = re.split(pattern=r'\n', string=text);
+            if all:
+                if first:
+                    while len(lines) > 0 and lines[0].strip() == '':
+                        lines = lines[1:];
+                if last:
+                    while len(lines) > 0 and lines[-1].strip() == '':
+                        lines = lines[:-1];
+            else:
+                if first:
+                    lines = lines[1:];
+                if last:
+                    lines = lines[:-1];
+            text = '\n'.join(lines);
+            return method(text);
+        return wrapped_method;
+    return dec;
+
+@prestrip(all=False)
+def dedent(text: str) -> str:
+    '''
+    Remove any common leading whitespace from every line in `text`.
+
+    This can be used to make triple-quoted strings line up with the left
+    edge of the display, while still presenting them in the source code
+    in indented form.
+
+    Note that tabs and spaces are both treated as whitespace, but they
+    are not equal: the lines "  hello" and "\\thello" are
+    considered to have no common leading whitespace.
+
+    Entirely blank lines are normalized to a newline character.
+    '''
+    return textwrap_dedent(text);
 
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # EXPORTS

src/thirdparty/plots.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+# IMPORTS
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+from matplotlib import pyplot as mplot;
+from matplotlib import colors as mcolours;
+from matplotlib.figure import Figure;
+from matplotlib.axes import Axes;
+from matplotlib.patches import FancyArrowPatch;
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+# EXPORTS
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+__all__ = [
+    'mplot',
+    'mcolours',
+    'Figure',
+    'Axes',
+    'FancyArrowPatch',
+];