# app.py

from flask import Flask, request, render_template, redirect, url_for
from sqlalchemy import create_engine, func
from sqlalchemy.orm import sessionmaker
from models.album import Base, Album

app = Flask(__name__)
engine = create_engine(config['DB_URL'])
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base.metadata.create_all(bind=engine)


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


@app.route('/')
def index():
    db = next(get_db())
    albums = db.query(Album).all()
    return render_template('index.html', albums=albums)


@app.route('/admin', methods=['GET', 'POST'])
def admin():
    if request.method == 'POST':
        action = request.form['action']
        album_id = int(request.form['album_id'])

        db = next(get_db())

        if action == 'edit':
            title = request.form['title']
            artist = request.form['artist']
            year = int(request.form['year'])

            album = db.query(Album).filter(Album.id == album_id).first()
            if album:
                album.title = title
                album.artist = artist
                album.year = year
                db.commit()
        elif action == 'delete':
            album = db.query(Album).filter(Album.id == album_id).first()
            if album:
                db.delete(album)
                db.commit()

        return redirect(url_for('admin'))

    db = next(get_db())
    albums = db.query(Album).all()
    return render_template('admin.html', albums=albums)


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