Notes

from flask import Flask
from flask_restful import Api, Resource, reqparse
from pymongo import MongoClient
from bson import ObjectId
from model import Customer

# initialising the Flask app
app = Flask(__name__)
api = Api(app)

# connecting to the Mongo Client
client = MongoClient('mongodb://localhost:27017')
db = client.customer_service
collection = db.customers

# request parser
parser = reqparse.RequestParser()
parser.add_argument('customerId', type=int, required=True,
help='Customer ID cannot be empty!')
parser.add_argument('customerName', type=str, required=True,
help='Customer Name cannot be empty!')
parser.add_argument('customerMobile', type=str, required=True,
help='Customer Mobile cannot be empty!')
parser.add_argument('customerAddress', type=str, required=True,
help='Customer Address cannot be empty!')


class CustomerService:
def add_customer(self, customer):
# adding the customer to the collection
# before adding checking if any customer with same ID exists
cust = collection.find_one({"customerId": customer["customerId"]})

if cust:
# already the customer exists
# a conflict
return False

else:
# customer is not in the collection
insert = collection.insert_one(customer)

return insert

def get_all_customers(self):
# getting all the customers from the collection
customers = list(collection.find())

for customer in customers:
# converting all the ObjectId to string
customer["_id"] = str(customer["_id"])

return customers


class CustomerController(Resource):
def __init__(self):
super().__init__()

self.customer_service = CustomerService()

def get(self):
customers = self.customer_service.get_all_customers()

return customers, 200

def post(self):
args = parser.parse_args()

customer = {
"customerId": args["customerId"],
"customerName": args["customerName"],
"customerMobile": args["customerMobile"],
"customerAddress": args["customerAddress"]
}

# adding the customer
insert = self.customer_service.add_customer(customer)

if insert:
return {"message": "Customer added successfully!"}, 201

else:
return {"message": "Customer already exists!"}, 409


api.add_resource(CustomerController, '/', '/<int:customerId>')

if __name__ == "__main__":
app.run(debug=True, port=5000)