aboutsummaryrefslogtreecommitdiff
path: root/wqflask/db/call.py
blob: 1fe0772b615ded0c1bc9bf8eadf69527558cf0bd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# Module for calling the backend

from flask import g

import string
try:  # Python2 support
    import urllib.request
    import urllib.error
    import urllib.parse
except:
    import urllib2
import json
from utility.tools import USE_GN_SERVER, LOG_SQL, GN_SERVER_URL
from utility.benchmark import Bench

from utility.logger import getLogger
logger = getLogger(__name__)

# from inspect import stack


def fetch1(query, path=None, func=None):
    """Fetch one result as a Tuple using either a SQL query or the URI
path to GN_SERVER (when USE_GN_SERVER is True). Apply func to
GN_SERVER result when set (which should return a Tuple)

    """
    if USE_GN_SERVER and path:
        result = gn_server(path)
        if func != None:
            res2 = func(result)
        else:
            res2 = result,
        if LOG_SQL:
            logger.debug("Replaced SQL call", query)
        logger.debug(path, res2)
        return res2
    else:
        return fetchone(query)


def fetchone(query):
    """Return tuple containing one row by calling SQL directly (the
original fetchone, but with logging)

    """
    with Bench("SQL", LOG_SQL):
        def helper(query):
            res = g.db.execute(query)
            return res.fetchone()
        return logger.sql(query, helper)


def fetchall(query):
    """Return row iterator by calling SQL directly (the
original fetchall, but with logging)

    """
    with Bench("SQL", LOG_SQL):
        def helper(query):
            res = g.db.execute(query)
            return res.fetchall()
        return logger.sql(query, helper)


def gn_server(path):
    """Return JSON record by calling GN_SERVER

    """
    with Bench("GN_SERVER", LOG_SQL):
        res = ()
        try:
            res = urllib.request.urlopen(GN_SERVER_URL + path)
        except:
            res = urllib2.urlopen(GN_SERVER_URL + path)
        rest = res.read()
        res2 = json.loads(rest)
        logger.debug(res2)
        return res2