Browse Source

Signed-off-by: Alan Woodman <alan@awoodman.net>

master
Alan Woodman 1 year ago
parent
commit
025013e9be
  1. 56
      main.py

56
main.py

@ -125,32 +125,32 @@ def address_search():
@main.route('/plan_search') @main.route('/plan_search')
def plan_search(): def plan_search():
plan_req: List[str] = request.args.get('plans', '').split(',') plan_req: Dict[str] = request.args.to_dict(flat=False)
if not plan_req or plan_req == ['']: if not plan_req or "plans" not in plan_req:
return jsonify({"plans": []}), HTTPStatus.OK return jsonify({"plans": []}), HTTPStatus.OK
else:
try: try:
plans = ( plans = (
db.session.query(Plans) db.session.query(Plans)
.filter(Plans.PlanIdentifier.in_(plan_req)) .filter(Plans.PlanIdentifier.in_(plan_req['plans']))
.all() .all()
) )
plan_list = [ plan_list = [
{ {
"plan": plan.PlanIdentifier, "plan": plan.PlanIdentifier,
"planName": plan.Name, "planName": plan.Name,
"cost": plan.Cost, "cost": plan.Cost,
"description": plan.Description, "description": plan.Description,
"uploadSpeed": plan.UploadSpeed, "uploadSpeed": plan.UploadSpeed,
"downloadSpeed": plan.DownloadSpeed, "downloadSpeed": plan.DownloadSpeed,
"wholesale": False "wholesale": False
} }
for plan in plans for plan in plans
] ]
return jsonify({"plans": plan_list}), HTTPStatus.OK return jsonify({"plans": plan_list}), HTTPStatus.OK
except Exception as e: except Exception as e:
db.session.rollback() db.session.rollback()
raise APIError(f"Error searching plans: {str(e)}", HTTPStatus.INTERNAL_SERVER_ERROR) raise APIError(f"Error searching plans: {str(e)}", HTTPStatus.INTERNAL_SERVER_ERROR)
Loading…
Cancel
Save