From 83081ad7e488c44757e43ff40e83458a2e1451ed Mon Sep 17 00:00:00 2001
From: Cédric Bonhomme <cedric@cedricbonhomme.org>
Date: Tue, 1 Mar 2016 22:47:53 +0100
Subject: begin integration of the new architecture

---
 src/web/models/right_mixin.py | 54 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 src/web/models/right_mixin.py

(limited to 'src/web/models/right_mixin.py')

diff --git a/src/web/models/right_mixin.py b/src/web/models/right_mixin.py
new file mode 100644
index 00000000..c4d92008
--- /dev/null
+++ b/src/web/models/right_mixin.py
@@ -0,0 +1,54 @@
+class RightMixin:
+
+    @staticmethod
+    def _fields_base_write():
+        return {}
+
+    @staticmethod
+    def _fields_base_read():
+        return {'id'}
+
+    @staticmethod
+    def _fields_api_write():
+        return {}
+
+    @staticmethod
+    def _fields_api_read():
+        return {'id'}
+
+    @classmethod
+    def fields_base_write(cls):
+        return cls._fields_base_write()
+
+    @classmethod
+    def fields_base_read(cls):
+        return cls._fields_base_write().union(cls._fields_base_read())
+
+    @classmethod
+    def fields_api_write(cls):
+        return cls.fields_base_write().union(cls._fields_api_write())
+
+    @classmethod
+    def fields_api_read(cls):
+        return cls.fields_base_read().union(cls._fields_api_read())
+
+    def __getitem__(self, key):
+        if not hasattr(self, '__dump__'):
+            self.__dump__ = {}
+        return self.__dump__.get(key)
+
+    def __setitem__(self, key, value):
+        if not hasattr(self, '__dump__'):
+            self.__dump__ = {}
+        self.__dump__[key] = value
+
+    def dump(self, role='admin'):
+        if role == 'admin':
+            dico = {k: getattr(self, k) for k in self.__table__.columns.keys()}
+        elif role == 'api':
+            dico = {k: getattr(self, k) for k in self.fields_api_read()}
+        else:
+            dico = {k: getattr(self, k) for k in self.fields_base_read()}
+        if hasattr(self, '__dump__'):
+            dico.update(self.__dump__)
+        return dico
-- 
cgit