Source code for squabble.util

Odds and ends pieces that don't fit elsewhere, but aren't important
enough to have their own modules.

import re

_RST_DIRECTIVE = re.compile(r'^\.\. [\w-]+:: \w+$\n', flags=re.MULTILINE)

[docs]def strip_rst_directives(string): """ Strip reStructuredText directives out of a block of text. Lines containing a directive will be stripped out entirely >>> strip_rst_directives('hello\\n.. code-block:: foo\\nworld') 'hello\\nworld' """ return re.sub(_RST_DIRECTIVE, '', string)
[docs]def format_type_name(type_name): """ Return a simple stringified version of a ``pglast.node.TypeName`` node. Note that this won't be suitable for printing, and ignores type modifiers (e.g. ``NUMERIC(3,4)`` => ``NUMERIC``). >>> import pglast >>> sql = 'CREATE TABLE _ (y time with time zone);' >>> node = pglast.Node(pglast.parse_sql(sql)) >>> col_def = node[0]['stmt']['tableElts'][0] >>> format_type_name(col_def.typeName) 'pg_catalog.timetz' """ return '.'.join([p.string_value for p in type_name.names])