Walle/walle/service/utils.py

# -*- coding: utf-8 -*-
"""Helper utilities and decorators."""


import fnmatch
import sys
import time
from datetime import datetime

import os
import re
from flask import flash
from invoke import Responder


def flash_errors(form, category='warning'):
    """Flash all errors for a form."""
    for field, errors in list(form.errors.items()):
        for error in errors:
            flash('{0} - {1}'.format(getattr(form, field).label.text, error), category)


def date_str_to_obj(ymd):
    return time.strptime(ymd, '%Y-%m-%d')


def datetime_str_to_obj(ymd_his):
    return datetime.strptime(ymd_his, "%Y-%m-%d %H:%i:%s")


PY2 = int(sys.version[0]) == 2

if PY2:
    text_type = unicode  # noqa
    binary_type = str
    string_types = (str, unicode)  # noqa
    unicode = unicode  # noqa
    basestring = basestring  # noqa
    reload(sys)
    sys.setdefaultencoding('utf8')
else:
    text_type = str
    binary_type = bytes
    string_types = (str,)
    unicode = str
    basestring = (str, bytes)


def detailtrace():
    from flask import current_app
    retStr = ""
    f = sys._getframe()
    f = f.f_back
    while hasattr(f, "f_code"):
        co = f.f_code
        retStr = "->%s(%s:%s)\n" % (os.path.basename(co.co_filename),
                                    co.co_name,
                                    f.f_lineno) + retStr
        f = f.f_back
    current_app.logger.info(retStr)
    print(retStr)


def color_clean(text_with_color):
    '''
    e.g \x1b[?1h\x1b=
    e.g \x1b[?1l\x1b>
    @param text_with_color:
    @return:
    '''
    pure_text = text_with_color.strip()
    pure_text = re.sub('\x1B\[[0-9;]*[mGK]', '', pure_text, flags=re.I)
    pure_text = re.sub('\x1B\[\?[0-9;]*[a-z]\x1B[=><]', '', pure_text, flags=re.I)
    return pure_text.strip()


def say_yes():
    return Responder(
        pattern=r'yes/no',
        response='yes\n',
    )


def excludes_format(path, excludes_string=None):
    '''
    排除文件，支持正则匹配，支持多选字符串
    @param path:
    @param excludes_string:
    @return:
    '''
    path = os.path.basename(path) + '/'
    if not excludes_string:
        return path

    prefix = '--exclude='
    excludes = [prefix + i for i in excludes_string.split('\n') if i.strip()]

    return ' {excludes} {path} '.format(excludes=' '.join(excludes), path=path)


def includes_format(path, includes_string=None):
    '''
    指定发布文件，支持正则匹配，如：*.war。支持多行字符串。

    @param path: release目录，非路径
    @param includes_string:
    @return:
    '''
    path = os.path.basename(path) + '/'
    if not includes_string:
        return path

    prefix = path
    includes = [prefix + i for i in includes_string.split('\n') if i.strip()]

    if not includes:
        return path

    return ' '.join(includes)