You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.6 KiB
Python
102 lines
2.6 KiB
Python
import logging
|
|
import sys
|
|
import textwrap
|
|
import warnings
|
|
|
|
from sqlalchemy.engine import url
|
|
|
|
from .compat import binary_type
|
|
from .compat import collections_abc
|
|
from .compat import py27
|
|
from .compat import string_types
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
if py27:
|
|
# disable "no handler found" errors
|
|
logging.getLogger("alembic").addHandler(logging.NullHandler())
|
|
|
|
|
|
try:
|
|
import fcntl
|
|
import termios
|
|
import struct
|
|
|
|
ioctl = fcntl.ioctl(0, termios.TIOCGWINSZ, struct.pack("HHHH", 0, 0, 0, 0))
|
|
_h, TERMWIDTH, _hp, _wp = struct.unpack("HHHH", ioctl)
|
|
if TERMWIDTH <= 0: # can occur if running in emacs pseudo-tty
|
|
TERMWIDTH = None
|
|
except (ImportError, IOError):
|
|
TERMWIDTH = None
|
|
|
|
|
|
def write_outstream(stream, *text):
|
|
encoding = getattr(stream, "encoding", "ascii") or "ascii"
|
|
for t in text:
|
|
if not isinstance(t, binary_type):
|
|
t = t.encode(encoding, "replace")
|
|
t = t.decode(encoding)
|
|
try:
|
|
stream.write(t)
|
|
except IOError:
|
|
# suppress "broken pipe" errors.
|
|
# no known way to handle this on Python 3 however
|
|
# as the exception is "ignored" (noisily) in TextIOWrapper.
|
|
break
|
|
|
|
|
|
def status(_statmsg, fn, *arg, **kw):
|
|
newline = kw.pop("newline", False)
|
|
msg(_statmsg + " ...", newline, True)
|
|
try:
|
|
ret = fn(*arg, **kw)
|
|
write_outstream(sys.stdout, " done\n")
|
|
return ret
|
|
except:
|
|
write_outstream(sys.stdout, " FAILED\n")
|
|
raise
|
|
|
|
|
|
def err(message):
|
|
log.error(message)
|
|
msg("FAILED: %s" % message)
|
|
sys.exit(-1)
|
|
|
|
|
|
def obfuscate_url_pw(u):
|
|
u = url.make_url(u)
|
|
if u.password:
|
|
u.password = "XXXXX"
|
|
return str(u)
|
|
|
|
|
|
def warn(msg, stacklevel=2):
|
|
warnings.warn(msg, UserWarning, stacklevel=stacklevel)
|
|
|
|
|
|
def msg(msg, newline=True, flush=False):
|
|
if TERMWIDTH is None:
|
|
write_outstream(sys.stdout, msg)
|
|
if newline:
|
|
write_outstream(sys.stdout, "\n")
|
|
else:
|
|
# left indent output lines
|
|
lines = textwrap.wrap(msg, TERMWIDTH)
|
|
if len(lines) > 1:
|
|
for line in lines[0:-1]:
|
|
write_outstream(sys.stdout, " ", line, "\n")
|
|
write_outstream(sys.stdout, " ", lines[-1], ("\n" if newline else ""))
|
|
if flush:
|
|
sys.stdout.flush()
|
|
|
|
|
|
def format_as_comma(value):
|
|
if value is None:
|
|
return ""
|
|
elif isinstance(value, string_types):
|
|
return value
|
|
elif isinstance(value, collections_abc.Iterable):
|
|
return ", ".join(value)
|
|
else:
|
|
raise ValueError("Don't know how to comma-format %r" % value)
|