from flask import Flask, jsonify
import uuid
import json
from webargs import fields, validate
from webargs.flaskparser import use_args
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy_utils import Timestamp, generic_repr
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker, relationship, backref
from sqlalchemy.types import Float, String, Boolean

app = Flask(__name__)
app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///jobs.db"
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False

db = SQLAlchemy(app)


@generic_repr
class Job(db.Model, Timestamp):
    id = db.Column(db.String, default=lambda: str(uuid.uuid4()), primary_key=True)
    jump_range = db.Column(db.Float, nullable=False)
    mode = db.Column(db.String, default="bfs")
    systems = db.Column(db.String)
    permute = db.Column(db.String, default=None, nullable=True)
    primary = db.Column(db.Boolean, default=False)
    factor = db.Column(db.Float, default=0.5)
    done = db.Column(db.DateTime, nullable=True, default=None)
    started = db.Column(db.DateTime, nullable=True, default=None)
    progress = db.Column(db.Float, default=0.0)

    # ============================================================

    @classmethod
    def new(cls, **kwargs):
        obj = cls(**kwargs)
        db.session.add(obj)
        db.session.commit()
        print(obj)
        return obj

    @property
    def dict(self):
        ret = {}
        for col in self.__table__.columns:
            ret[col.name] = getattr(self, col.name)
        ret["systems"] = json.loads(ret["systems"])
        return ret

    @dict.setter
    def set_dict(self, *args, **kwargs):
        raise NotImplementedError


db.create_all()
db.session.commit()


@app.errorhandler(422)
@app.errorhandler(400)
def handle_error(err):
    headers = err.data.get("headers", None)
    messages = err.data.get("messages", ["Invalid request."])
    if headers:
        return jsonify({"errors": messages}), err.code, headers
    else:
        return jsonify({"errors": messages}), err.code


@app.route("/route", methods=["GET", "POST"])
@use_args(
    {
        "jump_range": fields.Float(required=True),
        "mode": fields.String(
            missing="bfs", validate=validate.OneOf(["bfs", "greedy", "a-star"])
        ),
        "systems": fields.DelimitedList(fields.String, required=True),
        "permute": fields.String(
            missing=None,
            validate=validate.OneOf(["all", "keep_first", "keep_last", "keep_both"]),
        ),
        "primary": fields.Boolean(missing=False),
        "factor": fields.Float(missing=0.5),
    }
)
def route(args):
    args["systems"] = json.dumps(args["systems"])
    for k, v in args.items():
        print(k, v)
    return jsonify({"id": Job.new(**args).id})


@app.route("/status/<uuid:job_id>")
def status(job_id):
    job = db.session.query(Job).get_or_404(str(job_id))
    return jsonify(job.dict)


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=3777, debug=True)