from flask import Flask, render_template, send_from_directory
import os
import socket
import cPickle
app = Flask(__name__)
def dir_check(addr):
    if not os.path.exists('/home/pronto/scripts/flask/static/IPs/'+addr):
        os.makedirs('/home/pronto/scripts/flask/static/IPs/'+addr)
def scapy_svg(addr):
    if os.path.isfile('/home/pronto/scripts/flask/static/IPs/'+addr+'/trace.svg'):
        return (addr, "http://vps2.pronto185.com/static/IPs/"+addr+"/trace.svg")



@app.route('/newfail')
def newfail2():
    ip_list=cPickle.load(open("/home/pronto/logs/ssh_fails.dict.p", "rb"))
    svg_list=[]
    for a in ip_list:
        dir_check(a['IP'])
        svgcheck=scapy_svg(a['IP'])
        if svgcheck:
            svg_list.append(svgcheck)
    return render_template('new2.html', ip_list=ip_list, svg_list=svg_list)