"""Test flask file handling in fargate."""

import os

from flask import Flask, request

FILE_FIELD_NAME = 'demo_file'

app = Flask(__name__)


@app.route('/', methods=['POST'])
def upload_endpoint():
    """Handle file upload."""
    if FILE_FIELD_NAME not in request.files:
        return 'no file'
    line_count = 0
    for line in request.files[FILE_FIELD_NAME]:
        line_count = line_count + 1
    return 'line_count: {}'.format(line_count)

if os.environ.get('ENVIRONMENT', 'dev') == 'dev':
    app.run(host='0.0.0.0', debug=True, port=8080)