def handle_query():
"""
Handle any and all web queries and return result as json.
POST Parameters
---------------
action:
"submit" if edits are to be carried out, or the name
of a special query type.
type:
E.g. "story", "theme", "storytheme", etc. indicating what kind of
object is concerned.
"""
act_type = get("action")
req_type = get("type")
obj_type = OBJECT_TYPES.get(req_type)
obj_name = get("name")
log.debug("responding to: %s, %s, %s", act_type, req_type, obj_name)
res = cached_special_query(act_type, req_type, obj_name)
if res is not None:
return res
## queries for each object type available
if obj_type:
if act_type == "submit":
handle_submit(obj_type)
return handle_response(obj_type, req_type)
else:
return json.dumps({
"error":
"unknown request: %s." % str((req_type, act_type, obj_type)),
})
def handle_query():
"""
Handle any and all web submits.
"""
action = get("action")
log.debug("responding to: %s", action)
## queries for each object type available
if action == "protostory":
data = (get("storyentry") or "").strip()
rows = data.split("\n")
sid = None
path = None
results = []
retval = {}
overwrite = False
for idx, row in enumerate(rows):
if row.startswith("===") and idx > 0:
sid = rows[idx - 1].strip()
break
if sid:
fn = safe_filename(sid) + ".st.txt"
basepath = os.path.join(tempfile.gettempdir(), "tstp",
"protostory")
if not os.path.exists(basepath):
os.makedirs(basepath)
os.chdir(basepath)
path = os.path.join(basepath, fn)
overwrite = os.path.isfile(path)
with open(path, "wb+") as fh:
fh.write(data.encode("utf-8"))
fh.write("\n")
else:
retval.update({
"error": "Commit failed.",
})
retval.update({
"stdout": "\n".join(results),
"overwrite": overwrite,
"filename": path,
})
return json.dumps(retval)
else:
return json.dumps({
"error": "unknown request: %s." % str(action),
})
def handle_query():
"""
Handle any and all web submits.
"""
action = get("action")
log.debug("responding to: %s", action)
## queries for each object type available
if action == "protostory":
data = (get("storyentry") or "").strip()
rows = data.split("\n")
sid = None
path = None
results = []
retval = {}
overwrite = False
for idx, row in enumerate(rows):
if row.startswith("===") and idx > 0:
sid = rows[idx-1].strip()
break
if sid:
fn = safe_filename(sid) + ".st.txt"
basepath = os.path.join(tempfile.gettempdir(), "tstp", "protostory")
if not os.path.exists(basepath):
os.makedirs(basepath)
os.chdir(basepath)
path = os.path.join(basepath, fn)
overwrite = os.path.isfile(path)
with open(path, "wb+") as fh:
fh.write(data.encode("utf-8"))
fh.write("\n")
else:
retval.update({
"error": "Commit failed.",
})
retval.update({
"stdout": "\n".join(results),
"overwrite": overwrite,
"filename": path,
})
return json.dumps(retval)
else:
return json.dumps({
"error" : "unknown request: %s." % str(action),
})
def json_story_from_url_webquery():
try:
from webphp import php_get as get
url = get("url")
res = json_story_from_url(url)
except BaseException as e:
res = json.dumps({
"error": traceback.format_exc(),
})
return res
def json_story_from_url_webquery():
try:
from webphp import php_get as get
url = get("url")
res = json_story_from_url(url)
except BaseException as e:
res = json.dumps({
"error": traceback.format_exc(),
})
return res
def handle_query():
"""
Handle any and all requests for visualization data and
return result as json.
"""
subject = get("subject")
data = None
if subject == "viz.themecube":
from viz.themecube import get_viz_data
data = get_data(subject, get_viz_data, None)
if data is not None:
return json.dumps(data)
return json.dumps({
"error" : "unknown request: %s." % str(subject),
})
def handle_submit(obj_type):
"""
Do the db-update part of the query.
"""
if issubclass(obj_type, webobject.TSTPConnection):
update = get("data")
attrs = [x[0] for x in update]
vals = [x[1] for x in update]
cat1 = get("type1")
name1 = get("filter1")
cat2 = get("type2")
name2 = get("filter2")
obj_type.edit_object(cat1, name1, cat2, name2, attrs, vals)
else:
update = get("data")
attrs = [x[0] for x in update]
vals = [x[1] for x in update]
cat = get("type")
name = get("filter")
obj_type.edit_object(cat, name, attrs, vals)
def handle_response(obj_type, variant=None):
"""
Do the JSON response part of the query.
"""
string_limit = int(get('slimit', 100))
row_limit = int(get('rlimit', 10000))
fields = (get('fields') or '').split(',')
fuzzysearch = get('fuzzysearch', None)
collapsecollections = get('collapsecollections', None)
regexpnamefilter = get('regexpnamefilter', None)
collectionfilter = get('collectionfilter', None)
version = get('version', "latest")
return get_response(obj_type,
fields=fields,
fuzzysearch=fuzzysearch,
variant=variant,
string_limit=string_limit,
row_limit=row_limit,
collapsecollections=collapsecollections,
regexpnamefilter=regexpnamefilter,
collectionfilter=collectionfilter,
version='latest')
def get_response(obj_type,
fields='',
fuzzysearch=None,
variant=None,
string_limit=100,
row_limit=10000,
collapsecollections=None,
regexpnamefilter=None,
collectionfilter=None,
version='latest'):
"""
"""
if variant == "proposedevents":
# a list of events session user has proposed for insertion,
# not yet stored in database
from webupload import get_pending_events
objs = get_pending_events()
elif issubclass(obj_type, webobject.TSTPEvent):
cat1s = get('c1')
name1s = get('f1')
cat2s = get('c2')
name2s = get('f2')
objs = obj_type.load(
cat1s=cat1s,
name1s=name1s,
cat2s=cat2s,
name2s=name2s,
limit=row_limit,
)
elif issubclass(obj_type, webobject.TSTPConnection):
name1s = get('f1')
name2s = get('f2')
objs = obj_type.load(
name1s=name1s,
name2s=name2s,
limit=row_limit,
)
else:
names = get('filter')
objs = obj_type.load(
names=names,
limit=row_limit,
)
## collections may list their component stories, in which case we add them on the stories here
if issubclass(obj_type, webobject.Story):
objmap = {obj.name: obj for obj in objs}
for obj in objs:
if obj.name.lower().startswith("collection:"):
for line in obj.components.split("\n"):
cname = line.strip()
if cname in objmap:
objmap[cname].collections += "\n%s" % obj.name
## order items according to a search string and "fuzzy" string matching heuristics
if isinstance(fuzzysearch, basestring) and fuzzysearch.strip():
core = obj_type.__name__.lower()
extra_fields = ("score", )
if version == "latest":
import lib.search
scores = {w: s for s, w in lib.search.find(core, fuzzysearch)}
nobjs = []
for obj in objs:
obj.score = scores.get(obj.name, 0.0)
obj.extra_fields = extra_fields
if obj.score > 0.0:
nobjs.append(obj)
objs = nobjs
objs.sort(key=lambda o: o.score, reverse=False)
## show one series entry instead of every episode in a series, etc.
miscstory = webobject.Story.create(
name="Collection: Miscellaneous",
title="Miscellaneous",
date="various",
description="Otherwise uncategorized stories in the database.")
everystory = webobject.Story.create(
name="Collection: All",
title="All Stories",
date="various",
description="All defined stories in the database.")
if collapsecollections == "on" and issubclass(obj_type, webobject.Story):
colls = set()
for obj in objs:
colls.update(x for x in obj.collections.split("\n") if x)
objs = [o for o in objs if o.name in colls]
objs.append(miscstory)
objs.append(everystory)
# filter by collection name or general regex
if collectionfilter == miscstory.name:
objs = [o for o in objs if not o.collections.strip()]
elif collectionfilter == everystory.name:
pass
elif collectionfilter is not None:
objs = [
o for o in objs if collectionfilter in o.collections.split("\n")
]
return obj_type.make_json(objs, fields=fields, limit=string_limit)
