from flask import Flask, request, jsonify
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
db = SQLAlchemy(app)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root@localhost/alchemy'


class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(20))
    password = db.Column(db.String(100))

    def __init__(self, username, password):
        self.username = username
        self.password = password


@app.route('/login/', methods=['POST'])
def login():
    if request.method == 'POST':
        data = request.form.to_dict()
        username = data['username']
        password = data['password']
        registered_user = User.query.filter_by(username=username, password=password).first()
        if registered_user is None:
            return jsonify({'status': 0})
        return jsonify({'status': 1})


@app.route('/users/', methods=['POST'])
def save_users():
    if request.method == 'POST':
        data = request.form.to_dict()
        new_user = User(data['username'], data['password'])
        db.session.add(new_user)
        db.session.commit()
        return jsonify({'status': 1})


@app.route('/users/', methods=['GET'])
def users():
    if request.method == 'GET':
        results = User.query.limit(10).offset(0).all()

        json_results = []
        for result in results:
            d = {
                'id': result.id,
                'username': result.username,
                'password': result.password
            }
            json_results.append(d)
        return jsonify(json_results)


@app.route('/users/<int:user_id>', methods=['GET'])
def user(user_id):
    if request.method == 'GET':
        result = User.query.filter_by(id=user_id).first()

        json_result = {
            'id': result.id,
            'username': result.username,
            'password': result.password
        }
        return jsonify(json_result)


if __name__ == '__main__':
    app.run(debug=True)