filters.py
5.31 KB
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import datetime
from collections import defaultdict
from django.http import JsonResponse
from django.shortcuts import render
from connections.models import Entry, Subentry, ArgumentConnection
from syntax.models import Schema
from semantics.models import Frame
from .forms import FiltersForm
from .polish_strings import SCHEMA_OPINION, FRAME_OPINION
def filters_test(request):
return render(request, 'filters_test.html', { 'filters_form' : FiltersForm() })
def filter_objects(objects, queries):
print('===================================================================')
for query in queries:
print('***', query)
objects = objects.filter(query).distinct()
print('---------------------------------------------------------------')
print(objects)
print('\n')
print('===================================================================')
return objects.distinct()
def get_entries(request):
t1 = datetime.datetime.now()
if request.method == 'GET':
form = FiltersForm(request.GET)
if not form.is_valid():
return JsonResponse({ 'result' : [], 'errors' : form.errors })
if form.is_valid():
queries = form.get_queries()
t2 = datetime.datetime.now()
entries = filter_objects(Entry.objects.all(), queries)
t3 = datetime.datetime.now()
#entries_list = [(e.id, e.name, e.status.key) for e in entries]#[:500]]
entries_list = list(entries.values_list('id', 'name', 'status__key'))
t4 = datetime.datetime.now()
debug = '[{} s] [{} results] '.format(t2, entries.count())
debug += '[filtering: {:.1} s] '.format((t2 - t1).total_seconds())
debug += '[processing: {:.1} s] '.format((t4 - t3).total_seconds())
debug += '[total time: {:.1} s] '.format((t4 - t1).total_seconds())
return JsonResponse({ 'result' : entries_list, 'debug' : debug })
return JsonResponse({})
def subentry2str(subentry):
ret = subentry.entry.name
if subentry.inherent_sie.name == 'true':
ret += ' się'
elems = []
if subentry.aspect:
elems.append(subentry.aspect.name)
if subentry.negativity:
elems.append(subentry.negativity.name)
if elems:
ret += ' ({})'.format(', '.join(elems))
if subentry.predicativity.name == 'true':
ret += ' pred.'
return ret
def schema2dict(schema, phr2arg):
return {
'opinion' : SCHEMA_OPINION[schema.opinion.key],
'id' : str(schema.id),
'positions' : [
{
'func' : p.function.name if p.function else '',
'phrases' : [
{
'str' : str(pt),
'id' : '-'.join(map(str, (schema.id, p.id, pt.id))),
'arguments' : list(phr2arg['-'.join(map(str, (schema.id, p.id, pt.id)))]),
} for pt in p.phrase_types.all()
],
} for p in schema.positions.all()
],
}
def frame2dict(frame, arg2phr):
return {
'opinion' : FRAME_OPINION[frame.opinion.key],
'arguments' : [
{
'str' : str(a),
'id' : '-'.join(map(str, (frame.id, a.id))),
'phrases' : list(arg2phr['-'.join(map(str, (frame.id, a.id)))]),
} for a in frame.argument_set.all()
],
}
def get_entry_schemata_frames(request):
if request.method == 'GET':
form = FiltersForm(request.GET)
eid = request.GET['entry']
if eid.isdigit() and form.is_valid():
eid = int(eid)
entry = Entry.objects.get(id=eid)
#===================================================================
phr2arg, arg2phr = defaultdict(set), defaultdict(set)
connections = ArgumentConnection.objects.filter(schema_connections__subentry__entry=entry)
for conn in connections:
arg = '-'.join(map(str, (conn.argument.frame.id, conn.argument.id)))
for hook in conn.schema_connections.all():
phr = '-'.join(map(str, (hook.schema.id, hook.position.id, hook.phrase_type.id)))
phr2arg[phr].add(arg)
arg2phr[arg].add(phr)
#===================================================================
schema_queries = []
if form.cleaned_data['filter_schema_']:
schema_queries = form.get_queries('schema')
frame_queries = []
if form.cleaned_data['filter_frame_']:
frame_queries = form.get_queries('frame')
subentries = []
for subentry in entry.subentries.all():
schemata = []
for schema in filter_objects(subentry.schemata.all(), schema_queries):
schemata.append(schema2dict(schema, phr2arg))
if schemata:
subentries.append({ 'str' : subentry2str(subentry), 'schemata' : schemata })
frame_objs = Frame.objects.filter(argument__argument_connections__schema_connections__subentry__entry=entry)
frames = [frame2dict(frame, arg2phr) for frame in filter_objects(frame_objs, frame_queries)]
return JsonResponse({ 'subentries' : subentries, 'frames' : frames })
return JsonResponse({})