eamco_office_api/app/customer/views.py

from flask import request, jsonify
from flask_login import login_required
from app.customer import customer
from app import db
from datetime import datetime
from app.classes.cards import Card_Card
from app.classes.customer import \
    Customer_Customer, \
    Customer_Customer_schema
from app.classes.admin import Admin_Company
import string
import random

@customer.route("/all", methods=["GET"])
@login_required
def all_customers_around():
    customer_list = db.session \
        .query(Customer_Customer) \
        .all()
    customer_schema = Customer_Customer_schema(many=True)
    return jsonify(customer_schema.dump(customer_list))


@customer.route("/all/<int:page>", methods=["GET"])
@login_required
def all_customers(page):
    """
    pagination all customers
    """

    per_page_amount = 50
    if page is None:
        offset_limit = 0
    elif page == 1:
        offset_limit = 0
    else:
        offset_limit = (per_page_amount * page) - per_page_amount

    customer_list = db.session \
        .query(Customer_Customer) \
        .limit(per_page_amount).offset(offset_limit)

    customer_schema = Customer_Customer_schema(many=True)
    return jsonify(customer_schema.dump(customer_list))


@customer.route("/<int:customer_id>", methods=["GET"])
def get_a_customer(customer_id):
    """
    Checks auth token to ensure user is authenticated
    """
    get_customer = (db.session
        .query(Customer_Customer)
        .filter(Customer_Customer.id == customer_id)
        .first())
    customer_schema = Customer_Customer_schema(many=False)
    return jsonify(customer_schema.dump(get_customer))

def id_generator(size=6, chars=string.ascii_uppercase + string.digits):
    return ''.join(random.choice(chars) for _ in range(size))

@customer.route("/create", methods=["POST"])
@login_required
def create_customer():
    """
    """
    now = datetime.utcnow()
    get_company = db.session.query(Admin_Company).filter(Admin_Company.id == 1).first()

    starter_digits = str(get_company.account_prefix) + '-' + id_generator()

    made_account_number = starter_digits

    response_customer_last_name = request.json["customer_last_name"]
    response_customer_first_name = request.json["customer_first_name"]
    response_customer_town = request.json["customer_town"]
    response_customer_state = request.json["customer_state"]
    response_customer_zip = request.json["customer_zip"]
    response_customer_email = request.json["customer_email"]
    response_customer_automatic = request.json["customer_automatic"]
    response_customer_home_type = request.json["customer_home_type"]
    customer_phone_number = request.json["customer_phone_number"]
    customer_address = request.json["customer_address"]
    customer_apt = request.json["customer_apt"]
    if response_customer_automatic is True:
        auto_customer = 1
    else:
        auto_customer = 0

    int_customer_home_type = int(response_customer_home_type)
    response_customer_zip = int(response_customer_zip)
    response_customer_state = int(response_customer_state)

    new_customer = Customer_Customer(
        account_number=made_account_number,
        customer_last_name=response_customer_last_name,
        customer_first_name=response_customer_first_name,
        customer_town=response_customer_town,
        customer_state=response_customer_state,
        customer_zip=response_customer_zip,
        customer_first_call=now,
        customer_email=response_customer_email,
        customer_automatic=auto_customer,
        customer_home_type=int_customer_home_type,
        customer_phone_number=customer_phone_number,
        customer_address=customer_address,
        customer_apt=customer_apt

    )

    db.session.add(new_customer)
    db.session.commit()

    return jsonify({
        "ok": True,
        'user': {
            'user_id': new_customer.id,
            'user_name': new_customer.customer_last_name,
            'user_email': new_customer.customer_email,
        },
    }), 200


@customer.route("/edit/<int:customer_id>", methods=["PUT"])
@login_required
def edit_customer(customer_id):
    """
    """
    get_customer = (db.session
                    .query(Customer_Customer)
                    .filter(Customer_Customer.id == customer_id)
                    .first())
    response_customer_last_name = request.json["customer_last_name"]
    response_customer_first_name = request.json["customer_first_name"]
    response_customer_town = request.json["customer_town"]
    response_customer_state = request.json["customer_state"]
    response_customer_zip = request.json["customer_zip"]
    response_customer_phone_number = request.json["customer_phone_number"]
    response_customer_email = request.json["customer_email"]
    response_customer_automatic = request.json["customer_automatic"]
    response_customer_home_type = request.json["customer_home_type"]
    response_customer_address = request.json["customer_address"]

    get_customer.customer_address = response_customer_address
    get_customer.customer_home_type = response_customer_home_type
    get_customer.customer_automatic = response_customer_automatic
    get_customer.customer_phone_number = response_customer_phone_number
    get_customer.customer_last_name = response_customer_last_name
    get_customer.customer_first_name = response_customer_first_name
    get_customer.customer_town = response_customer_town
    get_customer.customer_state = response_customer_state
    get_customer.customer_zip = response_customer_zip
    get_customer.customer_email = response_customer_email

    db.session.add(get_customer)
    db.session.commit()

    return jsonify({
        "ok": True,
        'user': {
            'user_name': get_customer.customer_last_name,

            'user_email': get_customer.customer_email,
        },
    }), 200


@customer.route("/delete/<int:customer_id>", methods=["DELETE"])
@login_required
def delete_customer(customer_id):
    """
    """
    get_customer = (db.session
                    .query(Customer_Customer)
                    .filter(Customer_Customer.id == customer_id)
                    .first())

    get_cards = (db.session
                 .query(Card_Card)
                 .filter(Card_Card.user_id == get_customer.id)
                 .first())

    if get_cards is not None:
        db.session.delete(get_cards)

    db.session.delete(get_customer)
    db.session.commit()
    return jsonify({
        "ok": True,
        'user': {
            'user_name': get_customer.customer_last_name,
            'user_email': get_customer.customer_email,
        },
    }), 200