src/risotto/controller.py

@@ -39,9 +39,9 @@ class Controller:
                       **kwargs,
                       ):
         """ a wrapper to dispatcher's publish"""
+        version, message = uri.split('.', 1)
         if args:
             raise ValueError(_(f'the URI "{uri}" can only be published with keyword arguments'))
-        version, message = uri.split('.', 1)
         await dispatcher.publish(version,
                                  message,
                                  risotto_context,

src/risotto/dispatcher.py

@@ -4,7 +4,6 @@ try:
 except:
     from tiramisu import Config
     from tiramisu.error import ValueOptionError
-from asyncio import get_event_loop, ensure_future
 from traceback import print_exc
 from copy import copy
 from typing import Dict, Callable, List, Optional
@@ -16,6 +15,7 @@ from .logger import log
 from .config import get_config
 from .context import Context
 from . import register
+from .remote import Remote
 
 
 class CallDispatcher:
@@ -79,7 +79,7 @@ class CallDispatcher:
         """ execute the function associate with specified uri
         arguments are validate before
         """
-        risotto_context = self.build_new_context(old_risotto_context.__dict__,
+        risotto_context = self.build_new_context(old_risotto_context,
                                                  version,
                                                  message,
                                                  'rpc',
@@ -88,35 +88,20 @@ class CallDispatcher:
             raise CallError(_(f'cannot find version of message "{version}"'))
         if message not in self.messages[version]:
             raise CallError(_(f'cannot find message "{version}.{message}"'))
-        function_obj = self.messages[version][message]
+        function_objs = [self.messages[version][message]]
         # do not start a new database connection
         if hasattr(old_risotto_context, 'connection'):
             risotto_context.connection = old_risotto_context.connection
-            await self.check_message_type(risotto_context,
+            return await self.launch(version,
-                                          kwargs,
+                                     message,
-                                          )
+                                     risotto_context,
-            config_arguments = await self.load_kwargs_to_config(risotto_context,
-                                                                f'{version}.{message}',
-                                                                kwargs,
                                      check_role,
-                                                                internal,
-                                                                )
-            return await self.launch(risotto_context,
                                      kwargs,
-                                     config_arguments,
+                                     function_objs,
-                                     function_obj,
+                                     internal,
                                      )
         else:
             try:
-                await self.check_message_type(risotto_context,
-                                              kwargs,
-                                              )
-                config_arguments = await self.load_kwargs_to_config(risotto_context,
-                                                                    f'{version}.{message}',
-                                                                    kwargs,
-                                                                    check_role,
-                                                                    internal,
-                                                                    )
                 async with self.pool.acquire() as connection:
                     await connection.set_type_codec(
                         'json',
@@ -126,10 +111,13 @@ class CallDispatcher:
                     )
                     risotto_context.connection = connection
                     async with connection.transaction():
-                        return await self.launch(risotto_context,
+                        return await self.launch(version,
+                                                 message,
+                                                 risotto_context,
+                                                 check_role,
                                                  kwargs,
-                                                 config_arguments,
+                                                 function_objs,
-                                                 function_obj,
+                                                 internal,
                                                  )
             except CallError as err:
                 raise err
@@ -151,81 +139,43 @@ class CallDispatcher:
 
 
 class PublishDispatcher:
-    async def register_remote(self) -> None:
-        print()
-        print(_('======== Registered remote event ========'))
-        self.listened_connection = await self.pool.acquire()
-        for version, messages in self.messages.items():
-            for message, message_infos in messages.items():
-                # event not emit locally
-                if message_infos['pattern'] == 'event' and 'functions' in message_infos and message_infos['functions']:
-#                    module, submodule, submessage = message.split('.', 2)
-#                    if f'{module}.{submodule}' not in self.injected_self:
-                    uri = f'{version}.{message}'
-                    print(f'  - {uri}')
-                    await self.listened_connection.add_listener(uri,
-                                                                self.to_async_publish,
-                                                                )
-
     async def publish(self,
                       version: str,
                       message: str,
-                      risotto_context: Context,
+                      old_risotto_context: Context,
+                      check_role: bool=False,
+                      internal: bool=True,
                       **kwargs,
                       ) -> None:
-        if version not in self.messages or message not in self.messages[version]:
+        risotto_context = self.build_new_context(old_risotto_context,
-            raise ValueError(_(f'cannot find URI "{version}.{message}"'))
-
-        # publish to remote
-        remote_kw = dumps({'kwargs': kwargs,
-                           'context': {'username': risotto_context.username,
-                                       'paths': risotto_context.paths,
-                                       }
-                          })
-        # FIXME should be better :/
-        remote_kw = remote_kw.replace("'", "''")
-        await risotto_context.connection.execute(f'NOTIFY "{version}.{message}", \'{remote_kw}\'')
-
-    def to_async_publish(self,
-                         con: 'asyncpg.connection.Connection',
-                         pid: int,
-                         uri: str,
-                         payload: str,
-                         ) -> None:
-        version, message = uri.split('.', 1)
-        loop = get_event_loop()
-        remote_kw = loads(payload)
-        risotto_context = self.build_new_context(remote_kw['context'],
                                                  version,
                                                  message,
                                                  'event',
                                                  )
-        callback = lambda: ensure_future(self._publish(version,
+        try:
+            function_objs = self.messages[version][message].get('functions', [])
+        except KeyError:
+            raise ValueError(_(f'cannot find message {version}.{message}'))
+        # do not start a new database connection
+        if hasattr(old_risotto_context, 'connection'):
+            # publish to remove
+            remote_kw = dumps({'kwargs': kwargs,
+                               'context': risotto_context.__dict__,
+                              })
+            risotto_context.connection = old_risotto_context.connection
+            # FIXME should be better :/
+            remote_kw = remote_kw.replace("'", "''")
+            await risotto_context.connection.execute(f'NOTIFY "{version}.{message}", \'{remote_kw}\'')
+            return await self.launch(version,
                                      message,
                                      risotto_context,
-                                                       **remote_kw['kwargs'],
+                                     check_role,
-                                                       ))
-        loop.call_soon(callback)
-
-    async def _publish(self,
-                       version: str,
-                       message: str,
-                       risotto_context: Context,
-                       **kwargs,
-                       ) -> None:
-        config_arguments = await self.load_kwargs_to_config(risotto_context,
-                                                            f'{version}.{message}',
                                      kwargs,
-                                                            False,
+                                     function_objs,
-                                                            False,
+                                     internal,
                                      )
-        for function_obj in self.messages[version][message]['functions']:
-            print('======', function_obj['function'].__name__)
         async with self.pool.acquire() as connection:
             try:
-                    await self.check_message_type(risotto_context,
-                                                  kwargs,
-                                                  )
                 await connection.set_type_codec(
                     'json',
                     encoder=dumps,
@@ -234,10 +184,13 @@ class PublishDispatcher:
                 )
                 risotto_context.connection = connection
                 async with connection.transaction():
-                        await self.launch(risotto_context,
+                    return await self.launch(version,
+                                             message,
+                                             risotto_context,
+                                             check_role,
                                              kwargs,
-                                          config_arguments,
+                                             function_objs,
-                                          function_obj,
+                                             internal,
                                              )
             except CallError as err:
                 pass
@@ -258,13 +211,14 @@ class PublishDispatcher:
 
 
 class Dispatcher(register.RegisterDispatcher,
+                 Remote,
                  CallDispatcher,
                  PublishDispatcher):
     """ Manage message (call or publish)
     so launch a function when a message is called
     """
     def build_new_context(self,
-                          context: dict,
+                          old_risotto_context: Context,
                           version: str,
                           message: str,
                           type: str,
@@ -273,8 +227,8 @@ class Dispatcher(register.RegisterDispatcher,
         """
         uri = version + '.' + message
         risotto_context = Context()
-        risotto_context.username = context['username']
+        risotto_context.username = old_risotto_context.username
-        risotto_context.paths = copy(context['paths'])
+        risotto_context.paths = copy(old_risotto_context.paths)
         risotto_context.paths.append(uri)
         risotto_context.uri = uri
         risotto_context.type = type
@@ -388,12 +342,24 @@ class Dispatcher(register.RegisterDispatcher,
         raise NotAllowedError(_(f'You ({user_login}) don\'t have any authorisation to access to "{uri}"'))
 
     async def launch(self,
+                     version: str,
+                     message: str,
                      risotto_context: Context,
+                     check_role: bool,
                      kwargs: Dict,
-                     config_arguments: dict,
+                     function_objs: List,
-                     function_obj: Callable,
+                     internal: bool,
                      ) -> Optional[Dict]:
-        # so send the message
+        await self.check_message_type(risotto_context,
+                                      kwargs)
+        config_arguments = await self.load_kwargs_to_config(risotto_context,
+                                                            f'{version}.{message}',
+                                                            kwargs,
+                                                            check_role,
+                                                            internal,
+                                                            )
+        # config is ok, so send the message
+        for function_obj in function_objs:
             function = function_obj['function']
             submodule_name = function_obj['module']
             function_name = function.__name__
@@ -415,14 +381,12 @@ class Dispatcher(register.RegisterDispatcher,
                 await self.valid_call_returns(risotto_context,
                                               function,
                                               returns,
-                                          kwargs,
+                                              kwargs)
-                                          )
             # log the success
             await log.info_msg(risotto_context,
                                {'arguments': kwargs,
                                 'returns': returns},
-                           info_msg,
+                               info_msg)
-                           )
             # notification
             if function_obj.get('notification'):
                 notif_version, notif_message = function_obj['notification'].split('.', 1)
@@ -434,8 +398,7 @@ class Dispatcher(register.RegisterDispatcher,
                     await self.publish(notif_version,
                                        notif_message,
                                        risotto_context,
-                                   **ret,
+                                       **ret)
-                                   )
             if risotto_context.type == 'rpc':
                 return returns
 

src/risotto/register.py

@@ -300,7 +300,7 @@ class RegisterDispatcher:
             )
             if truncate:
                 async with connection.transaction():
-                    await connection.execute('TRUNCATE InfraServer, InfraSite, InfraZone, Log, ProviderDeployment, ProviderFactoryCluster, ProviderFactoryClusterNode, SettingApplicationservice, SettingApplicationServiceDependency, SettingRelease, SettingServer, SettingServermodel, SettingSource, UserRole, UserRoleURI, UserURI, UserUser, InfraServermodel, ProviderZone, ProviderServer, ProviderSource, ProviderApplicationservice, ProviderServermodel')
+                    await connection.execute('TRUNCATE InfraServer, InfraSite, InfraZone, Log, ProviderDeployment, ProviderFactoryCluster, ProviderFactoryClusterNode, SettingApplicationservice, SettingApplicationServiceDependency, SettingRelease, SettingServer, SettingServermodel, SettingSource, UserRole, UserRoleURI, UserURI, UserUser, InfraServermodel, ProviderZone, ProviderServer, ProviderSource, ProviderApplicationservice ProviderServermodel')
             async with connection.transaction():
                 for submodule_name, module in self.injected_self.items():
                     risotto_context = Context()

src/risotto/remote.py

@@ -0,0 +1,42 @@
+from asyncio import get_event_loop, ensure_future
+from json import loads
+
+
+from .context import Context
+from .config import get_config
+from .utils import _
+
+
+class Remote:
+    async def register_remote(self) -> None:
+        print()
+        print(_('======== Registered remote event ========'))
+        self.listened_connection = await self.pool.acquire()
+        for version, messages in self.messages.items():
+            for message, message_infos in messages.items():
+                # event not emit locally
+                if message_infos['pattern'] == 'event':
+                    module, submodule, submessage = message.split('.', 2)
+                    if f'{module}.{submodule}' not in self.injected_self:
+                        uri = f'{version}.{message}'
+                        print(f'  - {uri}')
+                        await self.listened_connection.add_listener(uri, self.to_async_publish)
+
+    def to_async_publish(self,
+                         con: 'asyncpg.connection.Connection',
+                         pid: int,
+                         uri: str,
+                         payload: str,
+                         ) -> None:
+        version, message = uri.split('.', 1)
+        loop = get_event_loop()
+        remote_kw = loads(payload)
+        context = Context()
+        for key, value in remote_kw['context'].items():
+            setattr(context, key, value)
+        callback = lambda: ensure_future(self.publish(version,
+                                                      message,
+                                                      context,
+                                                      **remote_kw['kwargs'],
+                                                      ))
+        loop.call_soon(callback)