web.py

#!/usr/bin/python3
from flask import Flask, request, render_template, abort, redirect, jsonify
from database import Database, Anime, Episode
import requests
import logging
from to_dict import anime_to_dict, ep_to_dict


log = logging.getLogger('werkzeug')
log.setLevel(logging.ERROR)
app = Flask(__name__)
app.config["database"] = "instance/database.db"
database = Database(app.config.get("database"))



@app.route("/")
def index():
    return render_template("anime.html")


@app.route("/get_anime_list")
def get_anime_list():
    anime = database.anime_all() # List of all anime
    anime = anime_to_dict(anime)
    return jsonify(anime)


# get specific anime with its episode list
@app.route("/get_anime_ep/<int:ID>")
def get_anime_ep(ID):
    anime = database.get_anime(ID)
    if anime:
        ep_list = database.anime_ep_list(ID)
        ep_list = ep_to_dict(ep_list)
        return jsonify({"episodes": ep_list, "index": anime.index})
    return abort(404)


# update/set index of anime
@app.route("/set_index/<int:ID>/<int:index>")
def set_anime_index(ID, index):
    anime = database.get_anime(ID)
    result = False
    if anime:
        ep_num = database.episodes_count(ID)
        # Number of episodes should be greater than index being set because index starts with 0
        if ep_num >= 0 and ep_num > index:
            anime.index = index
            database.update_anime(anime)
            result = True
    return jsonify({"result": result})


@app.route("/anime_data/<int:ID>")
def anime_data(ID):
    anime = database.get_anime(ID)
    if not anime:
        return abort(404)

    count = database.episodes_count(ID)
    return jsonify({"title": anime.title, "url": anime.anime_url, "count": count})


# save new anime, or update list of episode if anime already exist
@app.route("/save_anime", methods=["POST"])
def save_anime():
    anime_dict = request.get_json()
    anime_id = anime_dict.get("anime_id")
    anime_url = anime_dict.get("anime_url")
    if not anime_id:
        title = anime_dict.get("title").replace("-", " ")
        anime = Anime((None, title, anime_url, None, 0))
        anime = database.insert_anime(anime)
    else:
        anime = database.get_anime(anime_id)
        if not anime:
            return jsonify({"result": "No Anime Found to update"})
        anime.anime_url = anime_url

    ep_list = anime_dict.get("ep_list")
    for i in ep_list:
        ep_title = i[0]
        ep_url = i[1]
        episode = Episode((None, anime.id, ep_title, ep_url))
        database.insert_episode(episode)
    database.commit()
    return jsonify({"result": True})


# Delete anime with its episodes
@app.route("/del_anime/<int:ID>")
def del_anime(ID):
    anime = database.get_anime(ID)
    result = False
    if anime:
        database.delete_anime(ID)
        result = ID
    return jsonify({"result": result})


# Update title of anime
@app.route("/edit_title/<int:ID>", methods=["POST"])
def edit_title(ID):
    title = request.get_json().get("title")
    anime = database.get_anime(ID)
    if anime and title:
        anime.title = title
        database.update_anime(anime)
        return jsonify({"result": anime.title})
    return jsonify({"result": False})


@app.route("/fetch", methods=["POST"])
def fetch():
    result = False
    try:
        url = request.get_json().get("url")
        r = requests.get(url)
        if r.status_code != 200:
            return abort(r.status_code)
        result = r.content.decode()
    except Exception as err:
        print(err)
    return jsonify({"result": result})


if __name__ == "__main__":
    HOST = "0.0.0.0"
    PORT = 5050
    print("+-------------------------------+")
    print(f"URL: http://localhost:{PORT}")
    print("+-------------------------------+")
    app.run(host=HOST, port=PORT, debug=True)