aboutsummaryrefslogtreecommitdiff
path: root/wqflask/utility/db_tools.py
blob: 98da33f247abaebcc8d2c7c856b38df4df16f3fa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from utility.tools import get_setting
from wqflask.database import database_connection


def escape_(string):
    with database_connection(get_setting("SQL_URI")) as conn:
        return conn.escape_string(str(string))


def create_in_clause(items):
    """Create an in clause for mysql"""
    in_clause = ', '.join("'{}'".format(x) for x in mescape(*items))
    in_clause = '( {} )'.format(in_clause)
    return in_clause


def mescape(*items):
    """Multiple escape"""
    return [escape_(str(item)).decode('utf8') for item in items]


def escape(string_):
    return escape_(string_).decode('utf8')