utils.py

import datetime
from constants import *

def read_date(date):
  if date[0:4] == '2021':
    date = '2020' + date[4:]
  return datetime.datetime.strptime(date[0:10], '%Y-%m-%d')

def write_date(date):
  return date.strftime("%B %d")

def unlist(var):
  if type(var) == list:
    return var[0]
  else:
    return var

def dict_stringify(entities):
  ## Converts dictionary of lists/strings to a string separated by ,
  ## TODO: Simplify using json.dumps(entities)
  str_ = ""
  for key in sorted(entities.keys()) :
    if len(entities[key]) > 0:
      str_ += key + ":["
      if type(entities[key]) == str:
        str_ += entities[key]
      elif type(entities[key]) == list:
        str_ += ','.join(sorted(entities[key]))
      str_ += "],"
  return str_

def get_context_from_request(req, context_name):
  context_id = req["session"] + "/contexts/"+context_name

  context = {}
  for item in req["queryResult"]["outputContexts"]:
    if item['name'] == context_id:
      context = item["parameters"] if "parameters" in item else {}
  
  return context_id, context

def get_updated_context(params, context):

  # TODO: Construct a list of history instead of replacing with new? Set expiry?
  for param in params:
    if params[param]:
      context['ctx_' + param] = params[param]

  return context

def clear_from_context(context, fields):
  for field in fields:
    if context_index[field] in context:
      context[context_index[field]] = ''
  return context

def read_entry(entities, context, field):
  entry = None
  if entities_index[field] in entities and len(entities[entities_index[field]]) > 0:
    entry = entities[entities_index[field]]
  elif context_index[field] in context  and len(context[context_index[field]]) > 0:
    entry = context[context_index[field]]
  else:
    entry = default_values[field]
  return unlist(entry)

def read_entry_arr(entities, context, fields):
  return [read_entry(entities, context, x) for x in fields]