initial site copied in

1 files changed, 79 insertions, 0 deletions

diff --git a/cli/app/server/helpers.py b/cli/app/server/helpers.py
new file mode 100644
index 0000000..d4ba84a
--- /dev/null
+++ b/cli/app/server/helpers.py
@@ -0,0 +1,79 @@
+from sqlalchemy import and_
+from app.settings import app_cfg
+from app.sql.models.media import Media
+import datetime
+import dateutil
+
+def parse_search_args(args):
+  try:
+    limit = min(int(args.get('limit', default=app_cfg.DEFAULT_LIMIT)), app_cfg.MAX_LIMIT)
+  except Exception as e:
+    limit = app_cfg.DEFAULT_LIMIT
+  try:
+    offset = int(args.get('offset', default=0))
+  except Exception as e:
+    offset = 0
+  return offset, limit
+
+def parse_sort_args(args, table=None):
+  try:
+    sort = args.get('sort', default="id")
+  except Exception as e:
+    sort = 'id'
+  try:
+    order = args.get('order', default="asc")
+  except:
+    order = 'asc'
+  if table is not None:
+    column = getattr(table, sort)
+    order_by = getattr(column, order)()
+    if sort != 'id':
+      order_by_id = getattr(table.id, order)()
+    else:
+      order_by_id = None
+  else:
+    order_by = None
+    order_by_id = None
+  return sort, order, order_by, order_by_id
+
+def parse_crop_arg(args):
+  try:
+    x = float(args.get('x'))
+    y = float(args.get('y'))
+    w = float(args.get('w'))
+    h = float(args.get('h'))
+    crop = (x, y, w, h,)
+  except Exception as e:
+    crop = None
+  return crop
+
+def parse_media_args(args):
+  criteria = []
+  criteria_kwargs = {}
+  try:
+    import_id = int(args.get('import_id'))
+    criteria.append(Media.import_id == import_id)
+    criteria_kwargs['import_id_1'] = import_id
+  except:
+    pass
+  try:
+    start_date = str(args.get('start_date'))
+    start_date = dateutil.parser.parse(start_date)
+    start_date = start_date.replace(tzinfo=datetime.timezone.utc)
+    criteria.append(Media.created_at >= start_date)
+    criteria_kwargs['created_at_1'] = start_date
+  except:
+    pass
+  try:
+    end_date = str(args.get('end_date'))
+    end_date = dateutil.parser.parse(end_date)
+    end_date = end_date.replace(tzinfo=datetime.timezone.utc)
+    criteria.append(Media.created_at <= end_date)
+    criteria_kwargs['created_at_2'] = end_date
+  except:
+    pass
+  if len(criteria) > 1:
+    return criteria, criteria_kwargs
+  elif len(criteria) == 1:
+    return criteria, criteria_kwargs
+  return None, {}