Data loading over socket.

dsst/dsst_gtk3/client.py

@@ -0,0 +1,40 @@
+import pprint
+import socket
+from common import util, models
+try:
+    import cPickle as pickle
+except ImportError:
+    import pickle
+
+
+class Access:
+    def __init__(self, connection):
+        self.host = connection.get('host')
+        self.port = connection.get('port')
+        self.buffer = connection.get('buffer_size')
+        self.auth_key = connection.get('auth_key')
+        self.socket = socket.socket()
+
+    def send_request(self, action: str, *args):
+        request = {'auth_key': self.auth_key,
+                   'action': action,
+                   'args': args}
+        request = pickle.dumps(request)
+        try:
+            self.socket.connect((self.host, self.port))
+            util.send_msg(self.socket, request)
+            response = util.recv_msg(self.socket)
+            response = pickle.loads(response)
+            if not response.get('success'):
+                raise Exception(response.get('message'))
+        finally:
+            self.socket.close()
+        return response.get('data')
+
+if __name__ == '__main__':
+    access = Access({'host': 'europa', 'port': 12345, 'buffer_size': 1024, 'auth_key': 'a'})
+    action = 'load_seasons'
+    response = access.send_request(action)
+    pp = pprint.PrettyPrinter(indent=1)
+    for s in response:
+        pp.pprint(s.__dict__)