from flask import Blueprint, Response, jsonify

from app.services.image_service import ImageService
from app.services.torrent_service import TorrentService

bp = Blueprint("image", __name__)


@bp.route("/export")
def image_export():
    image_object = ImageService()

    def generate():
        yield from image_object.export()

    return Response(generate(), mimetype='text/event-stream')


@bp.route("/hasTorrent")
def image_has_torrent():
    torrent_obj = TorrentService(["C:\\Users\\Localhost\\Desktop\\Images"])
    return jsonify({"code": 200, "message": torrent_obj.has_torrent()})