Adding project to tz function

scripts/constants/__init__.py

@@ -21,8 +21,10 @@ class Endpoints:
 
     # OEE SERVICES
     oee_services = "/oee"
-    oee_services = "/oee_post"
-
+    oee_services_mapping = "/oee_mapping"
+    oee_services_getting = "/oee_getting"
+    delete_oee_service = "/oee_service_update"
+    update_oee_services = "/oee_services_delete"
     # tag_lists
     api_tags_lists = "/tag_lists"
     api_tags = "/tags"

scripts/constants/db_connections.py

 from scripts.config import DBConf
-from scripts.utils.mongo_util import MongoConnect
+from scripts.utils.mongo_util import MongoConnect, MongoCollectionBaseClass
 
 mongo_client = MongoConnect(uri=DBConf.MONGO_URI)()
+mongo_connection = MongoCollectionBaseClass(mongo_client=mongo_client, database="ilens_configuration",
+                                            collection="customer_projects")

scripts/core/handlers/oee_handlers.py

@@ -4,12 +4,14 @@ from scripts.constants.db_connections import mongo_client
 from scripts.db.mongo.ilens_configuration.collections import collection_constants
 from scripts.logging import logger
 from scripts.schemas.oee_config_schema import Oee_Tag_Mapping_List, Get_Oee_Tag, Update_Oee_Tags, Get_Project_Id
+from scripts.constants.db_connections import mongo_connection
 
 
 class Oee_Services:
     def __init__(self, project_id=Get_Project_Id.project_id):
         try:
             self.connect = collection_constants(mongo_client=mongo_client)
+            self.mongo_connect = mongo_connection
         except Exception as e:
             tb = traceback.format_exc()
             logger.exception(e)
@@ -81,3 +83,12 @@ class Oee_Services:
             logger.exception(tb)
             logger.exception(f"Exception occurred while fetching data as {e}")
             raise
+
+    def get_project_tz(self):
+        connection = self.mongo_connect
+        project_tz = connection.find()
+        return_json = {}
+        for each_project in project_tz:
+            for each_tz in each_project:
+                return_json.update({str(each_tz.get("customer_project_name")): str(each_tz.get("timezone"))})
+        return return_json

scripts/services/oee_services.py

@@ -7,7 +7,7 @@ from scripts.core.handlers import oee_handlers
 oee_services = APIRouter(prefix=Endpoints.oee_services, tags=["OEE Calculator"])
 
 
-@oee_services.post(Endpoints.oee_services)
+@oee_services.post(Endpoints.oee_services_mapping)
 async def oee_tag_mapping(oee_tag_mapping_list: GetOeeServices):
     try:
         tag_mapping = oee_tag_mapping_list.dict()
@@ -18,7 +18,7 @@ async def oee_tag_mapping(oee_tag_mapping_list: GetOeeServices):
         raise logger.exception("Exception occurred while inserting data")
 
 
-@oee_services.post(Endpoints.oee_services)
+@oee_services.post(Endpoints.oee_services_getting)
 async def get_oee_tag_mapping(get_oee_tags: GetOeeServices):
     try:
         get_oee_tags = get_oee_tags.dict()
@@ -29,7 +29,7 @@ async def get_oee_tag_mapping(get_oee_tags: GetOeeServices):
         raise logger.exception(f"Error while getting information as {e}")
 
 
-@oee_services.post(Endpoints.oee_services)
+@oee_services.post(Endpoints.delete_oee_service)
 async def delete_oee_tagging(delete_oee_tags: GetOeeServices):
     try:
         delete_oee_tags = delete_oee_tags.dict()
@@ -40,7 +40,7 @@ async def delete_oee_tagging(delete_oee_tags: GetOeeServices):
         raise logger.exception(f"Error while getting information as {e}")
 
 
-@oee_services.post(Endpoints.oee_services)
+@oee_services.post(Endpoints.update_oee_services)
 async def update_oee_tagging(update_oee_tags: GetOeeServices):
     try:
         update_oee_tags = update_oee_tags.dict()