app.py

import os
from flask import Flask, render_template, request, redirect
import random
import main
import pyqrcode
import string
import boto3

app = Flask(__name__)

@app.route('/')
def home():
    id = ''.join(random.choices(string.ascii_uppercase, k=6))
    return render_template('home.html', id=id)

@app.route('/admin/<id>', methods=['POST', 'GET'])
def admin(id):
    id = main.check_id(id)
    queue = ''
    if id is None:
        id = 'QUEUE ID DOESNT EXISTS'
    else:
        queue = main.get_queue(id)

    if request.method == 'POST':
        try:
            main.popped(id, queue[0])
            queue = main.remove(id, queue[0])
        except:
            print('Queue is Empty')
    
    try:
        if queue[0] == '':
            now_serving = 'No Customers In Line'
        else:
            now_serving = 'Now Serving: {}'. format(queue[0].replace('.', ' '))
    except:
        now_serving = 'No Customers in line'
    try:
        next_cust = 'Next Customer: {}'.format(queue[1].replace('.', ' '))
    except:
        next_cust = ''

    return render_template('admin.html', id=id, queue=queue, now_serving=now_serving, next_cust=next_cust)

@app.route('/create_queue/<id>', methods=['POST', 'GET'])
def create_queue(id):
    qrcode = pyqrcode.create('q.aldenwking.com/in_queue/id/{}'.format(id))
    qrcode.png('/tmp/code{}.png'.format(id), scale=7, module_color=[0x00, 0x00, 0x00], background=[0xFF,0xFF,0xFF, 0x94])
    client = boto3.client('s3')
    client.upload_file('/tmp/code{}.png'.format(id), 'queue-project', 'code{}.png'.format(id))
    image_src = 'https://{}.s3.amazonaws.com/{}'.format('queue-project', 'code{}.png'.format(id))

    if request.method == 'POST':
        main.create_queue(id)
        id = '{}'.format(main.check_id(id))
    
    if request.method == 'GET':
        id = main.check_id(id)
        if id is None:
            id = 'QUEUE ID DOESNT EXISTS'
        else:
            id = '{}'.format(id)

    return render_template('create_queue.html', id=id, image_src=image_src)

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

@app.route('/get_in_queue', methods=['POST'])
def get_id():
    id = request.form['id']
    id = main.check_id(id)
    if id is None:
        id = 'QUEUE ID DOESNT EXISTS'
    return id

@app.route('/in_queue', methods=['POST'])
def in_queue():
    id = get_id()
    return render_template('in_queue.html', id=id)

@app.route('/in_queue', methods=['POST'])
def get_name():
    name = request.form['name']
    return name

@app.route('/in_queue/id/<id>', methods=['POST', 'GET'])
def in_queue_id(id):
    position = 0
    if request.method == 'POST':
        name = get_name()
        main.get_in_queue(id, name)
        position = main.get_position(id, name)
        if position is None:
            main.get_in_queue(id, name)
            position = main.get_position(id, name)

        return redirect('/in_queue/id/{}/{}'.format(id, name.replace(' ', '.')))

    if request.method == 'GET':
        name = 'none'

    main.del_pics()

    return render_template('in_queue_id.html', position=position, id=id, name=name)

@app.route('/in_queue/id/<id>/<name>')
def in_queue_name(id, name):
    position = 0
    id = main.check_id(id)
    pop = main.get_pop(id)
    if id is None:
        id = 'QUEUE ID DOESNT EXISTS'
    else:
        queue = main.get_queue(id)
        if queue is not None:
            position = main.get_position(id, name)
        if pop is not None:
            if name in pop:
                position = 'out of line'
                queue = main.remove(id, name)
                queue = main.remove('pop{}'.format(id), name)
            else:
                position = main.get_position(id, name)
                if position is None:
                    main.get_in_queue(id, name)
                    position = main.get_position(id, name)

    return render_template('in_queue_id.html', position=position, id=id, name=name)

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

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

@app.route('/about')
def about():
    return render_template('about.html')
    
if __name__ == '__main__':
    app.run(host='127.0.0.1', debug=True, port=5001)