main.py

import os

from flask import (Flask, escape, render_template, request, send_file,
                   send_from_directory, url_for)

import BS_scrapper as BS
import CJ_scrapper as CJ
import IB_scrapper as IB
import RG_scrapper as RG

app = Flask(__name__)


@app.route('/')
def homepage():
    return render_template('homepage.html')


@app.route('/IB', methods=['GET', 'POST'])
def Idlebrain():
    error = None
    try:
        if request.method == 'GET':
            url = request.args.get("url")
            if url is not None:
                print(f'\nURL: {url}\n')
                IB_obj = IB.IB(url)
                invalid_url = IB_obj.start()
                zip_file_name = os.path.basename(IB_obj.imgs_dir) + '.zip'
                if not invalid_url:
                    print(zip_file_name)
                    return {'status': 'False', 'filename': zip_file_name}
                return {'status': 'True'}
        else:
            error = 'Invalid URL'
        return render_template('IB.html', error=error)
    except Exception as e:
        print(f'EXCEPTION!\n{e}')
        return {'status': 'Exception'}


@app.route('/RG', methods=['GET', 'POST'])
def ragalahari():
    error = None
    try:
        if request.method == 'GET':
            url = request.args.get("url")
            if url is not None:
                print(f'\nURL: {url}\n')
                (invalid_url, dir_name) = RG.start(url)
                zip_file_name = os.path.basename(dir_name) + '.zip'
                if not invalid_url:
                    print(zip_file_name)
                    return {'status': 'False', 'filename': zip_file_name}
                return {'status': 'True'}
        else:
            error = 'Invalid URL'
        return render_template('RG.html', error=error)
    except Exception as e:
        print(f'EXCEPTION!\n{e}')
        return {'status': 'Exception'}


@app.route('/CJ', methods=['GET', 'POST'])
def cinejosh():
    error = None
    try:
        if request.method == 'GET':
            url = request.args.get("url")
            if url is not None:
                print(f'\nURL: {url}\n')
                (invalid_url, dir_name) = CJ.start(url)
                zip_file_name = os.path.basename(dir_name) + '.zip'
                if not invalid_url:
                    print(zip_file_name)
                    return {'status': 'False', 'filename': zip_file_name}
                return {'status': 'True'}

        else:
            error = 'Invalid URL'
        return render_template('CJ.html', error=error)
    except Exception as e:
        print(f'EXCEPTION!\n{e}')
        return {'status': 'Exception'}


@app.route('/BS', methods=['GET', 'POST'])
def bst():
    error = None
    try:
        if request.method == 'GET':
            url = request.args.get("url")
            if url is not None:
                print(f'\nURL: {url}\n')
                (invalid_url, dir_name) = BS.start(url)
                zip_file_name = os.path.basename(dir_name) + '.zip'
                if not invalid_url:
                    print(zip_file_name)
                    return {'status': 'False', 'filename': zip_file_name}
                return {'status': 'True'}
        else:
            error = 'Invalid URL'
        return render_template('BS.html', error=error)
    except Exception as e:
        print(f'EXCEPTION!\n{e}')
        return {'status': 'Exception'}


@app.route('/favicon.ico')
def favicon():
    return send_from_directory(os.path.join(app.root_path, 'static'),
                               'favicon.ico', mimetype='image/vnd.microsoft.icon')


@app.route('/download', methods=['GET', 'POST'])
def download():
    if request.method == 'GET':
        file_path = request.args.get("filepath")
        if file_path is not None:
            #print(f'\n\nFile_path: {file_path}\n\n')
            path = os.path.join(os.getcwd(), file_path)
            #print(f'\n\nFilepath: {path}\n\n')
            return send_file(path, as_attachment=True)
        else:
            return "<p>Error</p>"
    else:
        return "<h3>Internal Error!</h3>"


@app.route('/about')
def about():
    return render_template('about.html')


@app.route('/privacy')
def privacy():
    return render_template('privacy.html')


@app.route('/contact')
def contact():
    return render_template('contact.html')

@app.route('/robots.txt')
@app.route('/sitemap.xml')
def static_from_root():
    return send_from_directory(app.static_folder, request.path[1:])


@app.route('/privacy_policy')
def privacy_policy():
    return render_template('privacy_policy.html')

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