main.py

from flask import Flask
from flask import request
from flask import render_template
from flask import send_file
from api import report_api
from modules.reportgen import create_csv
import threading
import validators
import io
from ast import literal_eval as make_tuple
from datetime import datetime, timedelta, timezone
import functools
import webbrowser
from config import *

if (not SB_API) or (not VT_API) or (not US_API) or (not MD_API):
    raise ValueError("Toolkit is not configured! Please check config.py")

app = Flask(__name__)

def timed_cache(**timedelta_kwargs):                                              
                                                                                  
    def _wrapper(f):                                                              
        update_delta = timedelta(**timedelta_kwargs)                              
        next_update = datetime.now(timezone.utc) + update_delta
        f = functools.lru_cache(None)(f)
                                                                 
        @functools.wraps(f)                                                       
        def _wrapped(*args, **kwargs):                                            
            nonlocal next_update
            now = datetime.now(timezone.utc)
            if now >= next_update:
                f.cache_clear()
                next_update = now + update_delta
            return f(*args, **kwargs)
        return _wrapped
    return _wrapper

@timed_cache(seconds=300)
def create_report(url: str, is_domain=False) -> tuple:
    report = report_api(url, is_domain)
    todo = [report.whois, report.urlscan, report.safe_browsing, report.virustotal, report.metadefender, report.yandex_status, report.wayback_machine, report.threatminer, report.netcraft, report.metamask, report.openphish]
    threads = []

    for i in todo:
        t = threading.Thread(target=i)
        threads.append(t)
        t.start()

    for t in threads:
        t.join()

    return report.domain, report.information, report.report_content

@app.route("/", methods=["GET"])
def index():
    return render_template("index.html")

@app.route("/search", methods=["POST"])
def search():
    try:
        url = request.form.get("url").strip()
    except TypeError:
        return "URL argument not found"

    if (validators.url(url) or validators.domain(url)):
        if not validators.url(url):
            results = create_report(f"https://{url}/", is_domain=True)
        else:
            results = create_report(url)

        return render_template("search.html", results=results, domain=results[0], information=results[1], records=results[2])
    else:
        return "Invalid search query"
        
@app.route("/export", methods=["POST"])
def exportcsv():
    try:
        results = make_tuple(request.form.get("results"))
    except TypeError:
        return "Results argument not found"

    report = create_csv(results)
    mem = io.BytesIO()
    mem.write(report.getvalue().encode())
    mem.seek(0)
    return send_file(mem, download_name="report.csv", mimetype="text/csv", as_attachment=True)

if __name__ == "__main__":
    webbrowser.open("http://localhost:9000/", new=2)
    app.run(host="localhost", port=9000)