def get_category(id):
response = category_pb2.Category(result="failed",
status=status.STATUS_404_NOT_FOUND,
message="Category not found!")
try:
with session_scope() as session:
try:
category = session.query(Category).filter(
Category.id == id, Category.is_deleted == False).first()
except DataError:
session.commit()
response.message = "Category does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
if category:
response = category_pb2.Category(
id=str(category.id),
name=category.name,
description=category.description,
parent_id=str(category.parent_id))
if category.parent:
response.parent_name = category.parent.name
else:
response.parent_name = "None"
session.commit()
except Exception as e:
print(e)
pass
return response
def create_category(self, user_id, category_data):
category = pb2.Category(
name=category_data['name'].strip(),
user_id=user_id,
description=category_data['description'].strip(),
parent_id=category_data['parentId'])
category = self.stub.Create(category)
response = MessageToDict(category)
return response
def delete_category(id, user_id):
response = category_pb2.Category(result="failed")
try:
with session_scope() as session:
try:
child = session.query(Category).filter(
Category.parent_id == id,
Category.is_deleted == False).first()
except DataError:
session.commit()
response.message = "Category does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
if child:
response.message = "Category have sub categories cannot be deleted!"
response.status = status.STATUS_400_BAD_REQUEST
else:
try:
category = session.query(Category).filter(
Category.id == id,
Category.is_deleted == False).first()
except DataError:
session.commit()
response.message = "Category does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
if category:
category.updated_at = None
category.deleted_at = datetime.now()
category.is_deleted = True
category.deleted_by = user_id
response.message = "Category deleted successfully!"
response.result = "success"
response.status = status.STATUS_204_NO_CONTENT
session.commit()
except Exception as e:
print(e)
response.message = "Unexpected error occurred!"
response.status = status.STATUS_500_INTERNAL_SERVER_ERROR
pass
return response
def get_category_parents():
categories_list = []
"""Reads the route guide database.
Returns:
The full contents of the route guide database as a sequence of
route_guide_pb2.Features.
"""
with session_scope() as session:
categories = session.query(Category).filter(
Category.parent_id == None, Category.is_deleted == False).all()
for item in categories:
category = category_pb2.Category(id=str(item.id),
name=item.name,
description=item.description,
parent_id=str(item.parent_id))
if item.parent:
category.parent_name = item.parent.name
else:
category.parent_name = "None"
categories_list.append(category)
return categories_list
def Destroy(self, request, context):
response = category_pb2.Category()
response = crud.delete_category(request.id, request.user_id)
return response
def Update(self, request, context):
response = category_pb2.Category()
response = crud.update_category(request.id, request.name,
request.description, request.parent_id,
request.user_id)
return response
def Retrieve(self, request, context):
response = category_pb2.Category()
response = crud.get_category(request.id)
return response
def create_category(parent_id, name, description, user_id):
name = name.title()
category = Category(name=name, description=description, created_by=user_id)
response = category_pb2.Category(result="failed")
try:
with session_scope() as session:
if parent_id:
try:
parent = session.query(Category).filter(
Category.is_deleted == False,
Category.id == parent_id).first()
if parent:
category.parent_id = parent.id
else:
session.commit()
response.message = "Category parent does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
except DataError:
session.commit()
response.message = "Category parent does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
session.add(category)
response = category_pb2.Category(
result="success",
status=status.STATUS_201_CREATED,
message="Category created successfully!")
session.commit()
except IntegrityError as e:
with session_scope() as session:
try:
category = session.query(Category).filter(
Category.is_deleted == True,
Category.name.ilike(name)).first()
except DataError:
category = None
if category:
category.name = name
category.description = description
category.created_at = datetime.now()
category.created_by = user_id
category.updated_at = None
category.updated_by = None
category.is_deleted = False
category.deleted_at = None
category.deleted_by = None
if parent_id:
category.parent_id = parent_id
else:
category.parent_id = None
# result = session.query(Category).filter(Category.is_deleted==True, Category.name.ilike(name)) \
# .update({"name": name, "description": description, "created_at": datetime.now(), "created_by": user_id,
# "updated_at": None, "updated_by": None, "parent_id": None,
# "is_deleted": False, "deleted_at": None, "deleted_by": None})
# if result:
session.commit()
response = category_pb2.Category(
result="success",
status=status.STATUS_201_CREATED,
message="Category created successfully!")
return response
# else:
# session.commit()
# response.message = "Unable to create category!"
# response.status = status.STATUS_500_INTERNAL_SERVER_ERROR
# return response
else:
session.commit()
response.message = "Category already exists with same name!"
response.status = status.STATUS_403_FORBIDDEN
return response
except Exception as e:
print(e)
response.message = "Unexpected error occurred!"
response.status = status.STATUS_500_INTERNAL_SERVER_ERROR
pass
return response
def update_category(id, name, description, parent_id, user_id):
name = name.title()
response = category_pb2.Category(result="failed",
status=status.STATUS_404_NOT_FOUND,
message="Category not found!")
try:
with session_scope() as session:
try:
category = session.query(Category).filter(
Category.id == id, Category.is_deleted == False).first()
except DataError:
session.commit()
response.message = "Category does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
if not category:
return response
elif parent_id:
try:
parent = session.query(Category).filter(
Category.is_deleted == False,
Category.id == parent_id).first()
except DataError:
session.commit()
response.message = "Category parent does not exist!"
response.status = status.STATUS_404_NOT_FOUND
return response
if parent:
if parent.id == category.id:
response.message = "Category cannot be sub category of itself!"
response.status = status.STATUS_403_FORBIDDEN
return response
result = session.query(Category).filter(Category.id == id, Category.is_deleted==False) \
.update({"name": name, "description": description, "parent_id": parent.id, "updated_at": datetime.now(), "updated_by": user_id})
if result:
response = category_pb2.Category(
result="success",
status=status.STATUS_200_OK,
message="Category updated successfully!")
else:
session.commit()
response.message = "Category parent does not exist!"
return response
else:
result = session.query(Category).filter(Category.id == id, Category.is_deleted==False) \
.update({"name": name, "description": description, "updated_at": datetime.now(), "updated_by": user_id})
if result:
response = category_pb2.Category(
result="success",
status=status.STATUS_200_OK,
message="Category updated successfully!")
session.commit()
except IntegrityError:
response.message = "Category already exists with same name!"
response.status = status.STATUS_403_FORBIDDEN
return response
except Exception as e:
print(e)
response.message = "Unexpected error occurred!"
response.status = status.STATUS_500_INTERNAL_SERVER_ERROR
pass
return response
import grpc
import sys
sys.path.append("/home/ali/gatz")
# import the generated classes
from services.protos.category import category_pb2, category_pb2_grpc
# open a gRPC channel
channel = grpc.insecure_channel('localhost:50052')
# create a stub (client)
stub = category_pb2_grpc.CategoryControllerStub(channel)
# create a valid request message
category = category_pb2.Category(name="My Category", description="My Description", parent_id=None)
# make the call
response = stub.Create(category)
# et voilà
print(response)