import falcon
from waitress import serve
from models import *
from playhouse.shortcuts import model_to_dict
import json


class UserIdResource():
     def on_get(self, req, resp, user_id):
         try:
             user = urozaev.get(urozaev.id == user_id)
             resp.body = json.dumps(model_to_dict(user))
         except urozaev.DoesNotExist:
             resp.status = falcon.HTTP_404


class UserResource():
     def on_get(self, req, resp):
         users = urozaev.select().order_by(urozaev.id)
         resp.body = json.dumps([model_to_dict(u) for u in users])


api = falcon.API()

users = UserResource()
users_id = UserIdResource()

api.add_route('/users/', users)
api.add_route('/users/{user_id}', users_id)

serve(api, host="127.0.0.1", port=8001)