fix: Update time-entries model #122

Author: fabidick22

commons/data_access_layer/cosmos_db.py

@@ -222,17 +222,8 @@ def find(
         function_mapper = self.get_mapper_or_dict(mapper)
         return function_mapper(self.check_visibility(found_item, visible_only))
 
-    def find_all(
-        self,
-        event_context: EventContext,
-        conditions: dict = {},
-        custom_sql_conditions: List[str] = [],
-        custom_params: dict = {},
-        max_count=None,
-        offset=0,
-        visible_only=True,
-        mapper: Callable = None,
-    ):
+    def find_all(self, event_context: EventContext, conditions: dict = {}, custom_sql_conditions: List[str] = [],
+                 custom_params: dict = {}, max_count=None, offset=0, visible_only=True, mapper: Callable = None):
         partition_key_value = self.find_partition_key_value(event_context)
         max_count = self.get_page_size_or(max_count)
         params = [
@@ -242,8 +233,8 @@ def find_all(
         ]
         params.extend(self.generate_params(conditions))
         params.extend(custom_params)
-        result = self.container.query_items(
-            query="""
+        print("before query")
+        query_str = """
             SELECT * FROM c
             WHERE c.{partition_key_attribute}=@partition_key_value
             {conditions_clause}
@@ -261,7 +252,10 @@ def find_all(
                     custom_sql_conditions
                 ),
                 order_clause=self.create_sql_order_clause(),
-            ),
+            )
+
+        result = self.container.query_items(
+            query=query_str,
             parameters=params,
             partition_key=partition_key_value,
             max_item_count=max_count,

time_tracker_api/projects/projects_model.py

@@ -73,7 +73,11 @@ def get_all(self, conditions: dict = None, **kwargs) -> list:
         customers_id = [customer.id for customer in customers]
         conditions = conditions if conditions else {}
         custom_condition = "c.customer_id IN {}".format(str(tuple(customers_id)))
-        return self.repository.find_all(event_ctx, conditions,  custom_sql_conditions=[custom_condition], **kwargs)
+        if "custom_sql_conditions" in kwargs:
+            kwargs["custom_sql_conditions"].append(custom_condition)
+        else:
+            kwargs["custom_sql_conditions"] = [custom_condition]
+        return self.repository.find_all(event_ctx, conditions, **kwargs)
 
 
 def create_dao() -> ProjectDao:

time_tracker_api/time_entries/time_entries_model.py

@@ -21,6 +21,7 @@
 from time_tracker_api.time_entries.custom_modules.utils import (
     add_project_name_to_time_entries,
 )
+from time_tracker_api.projects.projects_model import ProjectCosmosDBModel, create_dao as project_create_dao
 from time_tracker_api.projects import projects_model
 from time_tracker_api.database import CRUDDao, APICosmosDBDao
 from time_tracker_api.security import current_user_id
@@ -74,6 +75,14 @@ def __init__(self, data):  # pragma: no cover
     def running(self):
         return self.end_date is None
 
+    def __add__(self, other):
+        if type(other) is ProjectCosmosDBModel:
+            time_entry = self.__class__
+            time_entry.project_id = other.__dict__
+            return time_entry
+        else:
+            raise NotImplementedError
+
     def __repr__(self):
         return '<Time Entry %r>' % self.start_date  # pragma: no cover
 
@@ -102,6 +111,7 @@ def create_sql_ignore_id_condition(id: str):
 
     @staticmethod
     def create_sql_date_range_filter(date_range: dict) -> str:
+        print("data: {}".format(date_range))
         if 'start_date' and 'end_date' in date_range:
             return """
             ((c.start_date BETWEEN @start_date AND @end_date) OR
@@ -134,13 +144,12 @@ def find_all(
         conditions: dict = {},
         date_range: dict = {},
     ):
-        custom_sql_conditions = []
-        custom_sql_conditions.append(
-            self.create_sql_date_range_filter(date_range)
-        )
+        custom_sql_conditions = [self.create_sql_date_range_filter(date_range)]
 
-        custom_params = self.generate_params(date_range)
+        if event_context.is_admin:
+            conditions.pop("owner_id")
 
+        custom_params = self.generate_params(date_range)
         time_entries = CosmosDBRepository.find_all(
             self,
             event_context=event_context,
@@ -149,9 +158,14 @@ def find_all(
             custom_params=custom_params,
         )
 
-        project_dao = projects_model.create_dao()
-        projects = project_dao.get_all()
-        add_project_name_to_time_entries(time_entries, projects)
+        if time_entries:
+            projects_id = [project.project_id for project in time_entries]
+            p_ids = str(tuple(projects_id)).replace(",", "") if len(projects_id) == 1 else str(tuple(projects_id))
+            custom_conditions = "c.id IN {}".format(p_ids)
+
+            project_dao = projects_model.create_dao()
+            projects = project_dao.get_all(custom_sql_conditions=[custom_conditions])
+            add_project_name_to_time_entries(time_entries, projects)
         return time_entries
 
     def on_create(self, new_item_data: dict, event_context: EventContext):
@@ -305,14 +319,18 @@ def checks_owner_and_is_not_started(cls, data: dict):
                 "The specified time entry is already running",
             )
 
-    def get_all(self, conditions: dict = {}) -> list:
+    def get_all(self, conditions: dict = None, **kwargs) -> list:
         event_ctx = self.create_event_context("read-many")
         conditions.update({"owner_id": event_ctx.user_id})
 
-        date_range = self.handle_date_filter_args(args=conditions)
-        return self.repository.find_all(
-            event_ctx, conditions=conditions, date_range=date_range
-        )
+        if "start_date" and "end_date" in conditions:
+            date_range = conditions.copy()
+            date_range.pop("owner_id")
+            conditions.pop("start_date")
+            conditions.pop("end_date")
+        else:
+            date_range = self.handle_date_filter_args(args=conditions)
+        return self.repository.find_all(event_ctx, conditions=conditions, date_range=date_range)
 
     def get(self, id):
         event_ctx = self.create_event_context("read")

time_tracker_api/time_entries/time_entries_namespace.py

@@ -151,6 +151,21 @@
     location='args',
 )
 
+attributes_filter.add_argument(
+    'start_date',
+    required=False,
+    store_missing=False,
+    help="(Filter) Start to filter by",
+    location='args',
+)
+attributes_filter.add_argument(
+    'end_date',
+    required=False,
+    store_missing=False,
+    help="(Filter) End time to filter by",
+    location='args',
+)
+
 
 @ns.route('')
 class TimeEntries(Resource):