from flask import Flask,render_template, request, redirect, flash, url_for
from flask_sqlalchemy import SQLAlchemy



app = Flask(__name__)

'''
#CONEXION CON LA BASE DE DATOS
'''
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///vuls.db'
app.config['SECRET_KEY'] = "random string"

db = SQLAlchemy(app)
class Vulnerabilidad(db.Model):
    cod = db.Column('cod', db.Integer, primary_key = True)
    id_vuln = db.Column(db.String(10))
    origen_vul = db.Column(db.String(50))
    cve = db.Column(db.String(12))
    titulo = db.Column(db.String(30))
    descripcion  = db.Column(db.String(255))
    plan_acc = db.Column(db.String(255))
    exploit = db.Column(db.String(255))
    categoria = db.Column(db.String(255))
    impacto = db.Column(db.String(255))
    habilidad = db.Column(db.String(255))
    riesgo = db.Column(db.String(255))
    fecha_val = db.Column(db.String(255))
    nie = db.Column(db.String(255))


    def __init__(self,id_vuln, origen_vuln, cve, titulo, descripcion, plan, exploit, cat, impacto, habilidad, riesgo, fecha, nie):
        self.id_vuln = id_vuln
        self.origen_vuln = origen_vuln
        self.cve = cve
        self.titulo = titulo
        self.descripcion = descripcion
        self.plan_acc = plan_acc
        self.exploit = exploit
        self.categoria = categoria
        self.impacto = impacto
        self.habilidad = habilidad
        self.riesgo = riesgo
        self.fecha_val = fecha_val
        self.nie = nie




db.create_all()

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

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


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

'''
Esqueleto funcion subir datos con sqlite MODIFICAR CON LA BD ELEGIDA
'''
@app.route('/addrec',methods = ['POST', 'GET'])
def addrec():

    if request.method == 'POST':
      try:

        id_v = str(request.form['id_vuln'])
        ori_vul =str(request.form['origen_vuln'])
        cve = str(request.form['cve'])
      	titulo = str(request.form['titulo'])
      	descripcion = str(request.form['descripcion'])
      	plan = str(request.form['plan_acc'])
      	exploit = str(request.form['exploit'])
      	cat = str(request.form['categoria'])
      	impact = str(request.form['impacto'])
      	hab = str(request.form['habilidad'])
      	riesgo = str(request.form['riesgo'])
      	fecha = str(request.form['fecha_val'])
      	nie = str(request.form['nie'])



        test=Vulnerabilidad(id_v, ori_vul,cve, titulo, descripcion, plan, exploit,cat, impact, hab, riesgo, fecha, nie)


        msg = "creado objeto"


      except:
      	msg = "error"
      finally:
    	return render_template('result.html', msg = msg)

    if request.method == 'GET':
    	return 'Ok'

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