import time

from flask import request, jsonify
from __init__ import app
import auth, database, reloader
from models import FlagStatus
from spam import is_spam_flag


@app.route('/api/get_config')
@auth.api_auth_required
def get_config():
    config = reloader.get_config()
    return jsonify({key: value for key, value in config.items()
                    if 'PASSWORD' not in key and 'TOKEN' not in key})


@app.route('/api/post_flags', methods=['POST'])
@auth.api_auth_required
def post_flags():
    flags = request.get_json()
    flags = [item for item in flags if not is_spam_flag(item['flag'])]

    cur_time = round(time.time())
    rows = [(item['flag'], item['sploit'], item['team'], cur_time, FlagStatus.QUEUED.name)
            for item in flags]

    db = database.get()
    cursor = db.cursor()
    cursor.executemany("INSERT INTO flags (flag, sploit, team, time, status) "
                   "VALUES (%s, %s, %s, %s, %s)", rows)
    db.commit()

    return ''