GH zatosource#282 - Moving GenericModel SQL utils to a different module.

code/zato-common/src/zato/common/util/sql.py

@@ -254,3 +254,18 @@ def get_security_by_id(session, security_id):
            one()
 
 # ################################################################################################################################
+
+def get_instance_by_id(session, model_class, id):
+    return session.query(model_class).\
+           filter(model_class.id==id).\
+           one()
+
+# ################################################################################################################################
+
+def get_instance_by_name(session, model_class, type_, name):
+    return session.query(model_class).\
+           filter(model_class.type_==type_).\
+           filter(model_class.name==name).\
+           one()
+
+# ################################################################################################################################

code/zato-server/src/zato/server/connection/sftp.py

@@ -637,14 +637,3 @@ def write(self, data, remote_path, mode='w+b', overwrite=False, log_level=0):
 
 # ################################################################################################################################
 # ################################################################################################################################
-
-class MyService(Service):
-    def handle(self):
-        conn = self.out.sftp['local']
-        config = conn.config
-
-        api = SFTPIPCFacade(self.cid, self.server, config)
-        api.ping()
-
-# ################################################################################################################################
-# ################################################################################################################################

code/zato-server/src/zato/server/service/internal/generic/__init__.py

@@ -7,6 +7,7 @@
 """
 
 # Zato
+from zato.common.util.sql import get_instance_by_id, get_instance_by_name
 from zato.server.service.internal import AdminService
 
 # ################################################################################################################################
@@ -16,16 +17,11 @@ class _BaseService(AdminService):
 # ################################################################################################################################
 
     def _get_instance_by_id(self, session, model_class, id):
-        return session.query(model_class).\
-               filter(model_class.id==id).\
-               one()
+        return get_instance_by_id(session, model_class, id)
 
 # ################################################################################################################################
 
     def _get_instance_by_name(self, session, model_class, type_, name):
-        return session.query(model_class).\
-               filter(model_class.type_==type_).\
-               filter(model_class.name==name).\
-               one()
+        return get_instance_by_name(session, model_class, type_, name)
 
 # ################################################################################################################################

code/zato-server/src/zato/server/service/internal/outgoing/sftp.py

@@ -8,9 +8,13 @@
 
 from __future__ import absolute_import, division, print_function, unicode_literals
 
+# stdlib
+from contextlib import closing
 
 # Zato
 from zato.common.broker_message import OUTGOING
+from zato.common.odb.model import GenericConn as ModelGenericConn
+from zato.common.util.sql import get_instance_by_id
 from zato.server.service.internal import AdminService, AdminSIO
 
 # ################################################################################################################################
@@ -21,15 +25,19 @@ class Execute(AdminService):
     """
     class SimpleIO(AdminSIO):
         input_required = 'id', 'data', 'log_level'
-        output_optional = 'response_time', 'stdout', 'stderr'
+        output_optional = 'response_time', 'stdout', 'stderr', 'command_no'
+        response_elem = None
 
     def handle(self):
         msg = self.request.input.deepcopy()
         msg['action'] = OUTGOING.SFTP_EXECUTE.value
         msg['cid'] = self.cid
 
-        out = self.server.connector_sftp.invoke_sftp_connector(msg)
-        self.response.payload = out.text
+        with closing(self.odb.session()) as session:
+            instance = get_instance_by_id(session, ModelGenericConn, self.request.input.id)
+            conn = self.out.sftp[instance.name].conn
+            response = conn.execute(self.request.input.data, self.request.input.log_level)
+            self.response.payload = response.to_dict()
 
 # ################################################################################################################################
 # ################################################################################################################################