first commit

app/customer/__init__.py

@@ -0,0 +1,8 @@
+# coding=utf-8
+from flask import Blueprint
+
+
+customer = Blueprint('customer', __name__)
+
+
+from . import views

app/customer/views.py

@@ -0,0 +1,198 @@
+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())
+    print(get_customer)
+    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()
+    print("deleted")
+    return jsonify({
+        "ok": True,
+        'user': {
+            'user_name': get_customer.customer_last_name,
+            'user_email': get_customer.customer_email,
+        },
+    }), 200
+
+