Merge pull request #1 from KaveTech/feat/manage-reference-entities

Manage reference entities

akeneo_api_client/client.py

@@ -1,7 +1,24 @@
-# -*- coding: utf-8 -*-
-
-from akeneo_api_client.auth import *
-from akeneo_api_client.resources import *
+import requests
+
+from akeneo_api_client.auth import Auth
+from akeneo_api_client.resources import (
+    AssociationTypesPool,
+    AttributesPool,
+    AttributeGroupsPool,
+    CategoriesPool,
+    ChannelsPool,
+    CurrenciesPool,
+    FamiliesPool,
+    LocalesPool,
+    MeasureFamiliesPool,
+    MediaFilesPool,
+    ProductsPool,
+    ProductModelsPool,
+    PublishedProductsPool,
+    AssetFamilyPool,
+    ReferenceEntityPool,
+)
+from akeneo_api_client.utils import urljoin
 
 
 class Client:
@@ -68,6 +85,10 @@ def _init(self, base_url, session, auth):
                 urljoin(self._base_url, self.BASIC_API_PATH, 'published-products/'), session),
             'asset_families': AssetFamilyPool(
                 urljoin(self._base_url, self.BASIC_API_PATH, 'asset-families/'), session),
+            "reference_entities": ReferenceEntityPool(
+                urljoin(self._base_url, self.BASIC_API_PATH, "reference-entities/"),
+                session,
+            ),
         }
 
     @property
@@ -130,3 +151,7 @@ def published_products(self):
     @property
     def asset_families(self):
         return self._resources["asset_families"]
+
+    @property
+    def reference_entities(self):
+        return self._resources["reference_entities"]

akeneo_api_client/resources.py

@@ -361,3 +361,27 @@ def assets(self, code):
             urljoin(self._endpoint, code, 'assets/'),
             self._session
         )
+
+
+class ReferenceEntityRecordPool(
+    ResourcePool,
+    CodeBasedResource,
+    GettableResource,
+    ListableResource,
+    UpdatableResource,
+    UpdatableListResource,
+):
+    pass
+
+
+class ReferenceEntityPool(
+    ResourcePool,
+    CodeBasedResource,
+    GettableResource,
+    ListableResource,
+    UpdatableResource,
+):
+    def records(self, entity_code):
+        return ReferenceEntityRecordPool(
+            urljoin(self._endpoint, entity_code, "records/"), self._session
+        )