From ae42ffeb94bc351e56d2b276b7f273fefeac22d6 Mon Sep 17 00:00:00 2001
From: Heinrichs <Heinrichs@itc.rwth-aachen.de>
Date: Fri, 26 Jun 2020 13:23:08 +0200
Subject: [PATCH] Add a method to create a returnobject for projectquota

---
 src/Database/Models/ProjectQuotasModel.cs | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/Database/Models/ProjectQuotasModel.cs b/src/Database/Models/ProjectQuotasModel.cs
index f8d21b8..5452a3c 100644
--- a/src/Database/Models/ProjectQuotasModel.cs
+++ b/src/Database/Models/ProjectQuotasModel.cs
@@ -1,12 +1,34 @@
 using Coscine.Database.DataModel;
+using Coscine.Database.ReturnObjects;
 using LinqToDB;
 using System;
+using System.Collections.Generic;
 using System.Linq.Expressions;
 
 namespace Coscine.Database.Models
 {
     public class ProjectQuotaModel : DatabaseModel<ProjectQuota>
     {
+        public ProjectQuotaObject CreateReturnObjectFromDatabaseObject(ProjectQuota projectQuota)
+        {
+            if (projectQuota.ResourceType == null)
+            {
+                ResourceTypeModel resourceTypeModel = new ResourceTypeModel();
+                projectQuota.ResourceType = resourceTypeModel.GetById(projectQuota.ResourceTypeId);
+            }
+
+            return new ProjectQuotaObject(
+                projectQuota.RelationId,
+                projectQuota.ProjectId,
+                new ResourceTypeObject(
+                    projectQuota.ResourceType.Id,
+                    projectQuota.ResourceType.DisplayName,
+                    projectQuota.ResourceType.Enabled
+                ),
+                projectQuota.Quota
+            );
+        }
+
         public override Expression<Func<ProjectQuota, Guid>> GetIdFromObject()
         {
             return (projectQuota) => projectQuota.RelationId;
-- 
GitLab