infrared/utils.py at main · jblowe/infrared · GitHub

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
80
81
82
83
84
85
86
87
88
89
import os, time, sys, re
import lxml.etree as ET
from urllib.parse import urlencode

from bottle import template, redirect

from config import *
import solr_query
import subprocess

# nb: we are trying to get the directory above the directory this file is in
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))


def get_version():
    try:
        tag = subprocess.check_output(
            ['git', 'describe', '--tags', '--abbrev=0'],
            stderr=subprocess.DEVNULL
        ).decode().strip()
        return tag
    except subprocess.CalledProcessError:
        return "no-tag"
    except FileNotFoundError:
        return "git-not-found"


def spaceless(rendered_html):
    return re.sub(r'>\s+<', '><', rendered_html.strip())


def add_time_and_version():
    return 'version: %s, system last restarted: %s' % (
        get_version(), time.strftime("%Y-%m-%d %H:%M:%S UTC", time.gmtime()))


def check_template(tpl, data, controls, request):
    if 'back' not in data:
        data['back'] = '/'
    # redirect('/facet/' + data['query_string'])

    return template(spaceless(tpl), controls=controls, data=data)


def query(parameters):
    query_terms = {}
    ROW_LIMIT = 80
    try:
        start_row = ROW_LIMIT * (int(parameters['controls']['page']) - 1)
    except:
        start_row = 0
    result_fields = parameters['result_fields'] + [parmz.IMAGE_FIELD] + [parmz.TITLE_FIELD] + ['id']
    facet_fields = parameters['facet_fields']
    results = solr_query.solr_main_query(parameters['terms'], result_fields, facet_fields, ROW_LIMIT, start_row,
                                         parmz.FACET_LIMIT,
                                         parmz.FACET_MINCOUNT)
    full_facets = {}
    for f in results['facets']:
        if results['facets'][f] != {}:
            full_facets[f] = results['facets'][f]
    results['facets'] = full_facets
    results['start_row'] = start_row
    # for r in results['results']:
    #     for f in r:
    #         if type(r[f]) == type([]):
    #             r[f] = ', '.join(r[f])
    return results


def do_solr_query(parameters, result_fields, terms, controls):
    return {
        'results': query(parameters),
        'selected_field': '',
        'result_fields': result_fields,
        'terms': terms,
        'controls': controls,
        'query_string': '&'.join([urlencode(terms), urlencode(controls)]),
        'base_string': urlencode(terms),
        'image_prefix': parmz.IMAGE_PREFIX,
        'image_field': parmz.IMAGE_FIELD}


def set_controls(controls):
    if not 'page' in controls: controls['page'] = 1
    if not 'per_page' in controls: controls['per_page'] = 80
    if not 'view' in controls: controls['view'] = 'LIST'
    return controls

VERSION = get_version()