Graphql groups resolvers (canada-ca#88)

* Started work on groups resolvers

* Started work on groups resolvers testing

* Added docstrings to group resolvers

* Group resolver test has now been updated to match changes with new db system

* Updated groups resovlers to include description. Added enums to group selections

* Initial testing started

* cloudbuild.yaml update

* Fixed tests for sectors and groups to remove information after testing to remove conflicts between tests

* Cleaning up groups and sectors resolvers test files

* Added invalid request resolver tests

Author: Nicholas

api/Pipfile.lock

@@ -6,35 +6,39 @@
 import pyotp
 
 from model_enums.sectors import SectorEnums, ZoneEnums
-
-from schemas.user import (
-	UserConnection,
-	CreateUser,
-	SignInUser,
-	UpdateUserPassword
-)
+from model_enums.groups import GroupEnums
 
 from schemas.user import *
 
 from schemas.sectors import Sectors
+from schemas.groups import Groups
+
 
 from resolvers.sectors import (
 	resolve_get_sector_by_id,
 	resolve_get_sectors_by_sector,
 	resolve_get_sector_by_zone
 )
 
+from resolvers.groups import (
+	resolve_get_group_by_id,
+	resolve_get_group_by_group,
+	resolve_get_group_by_sector
+)
+
 
 class Query(graphene.ObjectType):
 	"""The central gathering point for all of the GraphQL queries."""
 	node = relay.Node.Field()
 	all_users = SQLAlchemyConnectionField(UserConnection, sort=None)
 	# all_users = graphene.List(UserObject, failedAttempts=graphene.Int(), resolver=resolve_all_users)
+	# all_sectors = SQLAlchemyConnectionField(SectorsConnection, sort=None)
+	# all_groups = SQLAlchemyConnectionField(GroupsConnection, sort=None)
 	get_sector_by_id = graphene.List(
 		of_type=Sectors,
 		id=graphene.Argument(graphene.Int, required=True),
 		resolver=resolve_get_sector_by_id,
-		description="Allows selection of all sectors from a given sector ID"
+		description="Allows selection of a sector from a given sector ID"
 	)
 	get_sectors_by_sector = graphene.List(
 		of_type=Sectors,
@@ -48,6 +52,24 @@ class Query(graphene.ObjectType):
 		resolver=resolve_get_sector_by_zone,
 		description="Allows selection of all sectors from a given zone enum"
 	)
+	get_group_by_id = graphene.List(
+		of_type=Groups,
+		id=graphene.Argument(graphene.Int, required=False),
+		resolver=resolve_get_group_by_id,
+		description="Allows selection of a group from a given group ID"
+	)
+	get_group_by_group = graphene.List(
+		of_type=Groups,
+		group=graphene.Argument(GroupEnums, required=True),
+		resolver=resolve_get_group_by_group,
+		description="Allows the selection of group information from a given group enum"
+	)
+	get_group_by_sector = graphene.List(
+		of_type=Groups,
+		sector=graphene.Argument(SectorEnums, required=True),
+		resolver=resolve_get_group_by_sector,
+		description="Allows selection of groups from a given sector enum"
+	)
 
 	generate_otp_url = String(email=String(required=True))
 

api/queries.py

@@ -0,0 +1,55 @@
+from graphql import GraphQLError
+from sqlalchemy.orm import load_only
+
+from schemas.groups import (
+	Groups,
+	GroupsModel
+)
+
+from schemas.sectors import (
+	Sectors,
+	SectorsModel
+)
+
+
+# Resolvers
+def resolve_get_group_by_id(self, info, **kwargs):
+	"""Return a group by its row ID"""
+	group_id = kwargs.get('id', 1)
+	query = Groups.get_query(info).filter(
+		GroupsModel.id == group_id
+	)
+	if not len(query.all()):
+		raise GraphQLError("Error, Invalid ID")
+	return query.all()
+
+
+def resolve_get_group_by_group(self, info, **kwargs):
+	"""Return a list of groups by its group"""
+	group = kwargs.get('group')
+	query = Groups.get_query(info).filter(
+		GroupsModel.s_group == group
+	)
+	if not len(query.all()):
+		raise GraphQLError("Error, group does not exist ")
+	return query.all()
+
+
+def resolve_get_group_by_sector(self, info, **kwargs):
+	"""Return a list of groups by by their associated sector"""
+	sector = kwargs.get('sector')
+
+	sector_id = Sectors.get_query(info).filter(
+		SectorsModel.sector == sector
+	).options(load_only('id'))
+
+	if not len(sector_id.all()):
+		raise GraphQLError("Error, no sector associated with that enum")
+
+	query = Groups.get_query(info).filter(
+		GroupsModel.sector_id == sector_id
+	)
+
+	if not len(query.all()):
+		raise GraphQLError("Error, no groups associated with that sector")
+	return query.all()

api/resolvers/groups.py

@@ -1,6 +1,6 @@
 import graphene
 from graphene import relay
-from graphene_sqlalchemy import SQLAlchemyObjectType, SQLAlchemyConnectionField
+from graphene_sqlalchemy import SQLAlchemyObjectType
 
 from models import Groups as GroupsModel