Added network connect/disconnect API.

Author: rtnpro

ircb/web/app.py

@@ -9,7 +9,8 @@
 
 from ircb.config import settings
 from ircb.web.user import SigninView, SignoutView
-from ircb.web.network import NetworkListView, NetworkView
+from ircb.web.network import (NetworkListView, NetworkView,
+                              NetworkConnectionView)
 
 logging.config.dictConfig(settings.LOGGING_CONF)
 
@@ -39,6 +40,9 @@ def init(loop):
                          name='networks')
     app.router.add_route('*', '/api/v1/network/{id}', NetworkView,
                          name='network')
+    app.router.add_route('PUT', '/api/v1/network/{id}/{action}',
+                         NetworkConnectionView,
+                         name='network_connection')
     srv = yield from loop.create_server(
         app.make_handler(), '0.0.0.0', 10001)
     return srv

ircb/web/network.py

@@ -96,3 +96,30 @@ def put(self):
             raise web.HTTPNotFound()
         resp = yield from self._create_or_update(data, username, networks[0])
         return resp
+
+
+class NetworkConnectionView(View):
+    store = NetworkStore
+
+    @auth_required
+    @asyncio.coroutine
+    def put(self):
+        network_id = self.request.match_info['id']
+        action = self.request.match_info['action']
+        if action not in ('connect', 'disconnect'):
+            raise web.HTTPNotFound()
+        username = yield from get_auth(self.request)
+        user = yield from UserStore.get(
+            dict(query=('username', username)))
+        networks = yield from NetworkStore.get(
+            dict(query={'user_id': user.id, 'id': network_id})
+        )
+        if not networks:
+            raise web.HTTPNotFound()
+        network = networks[0]
+        network = yield from NetworkStore.update(
+            dict(filter=('id', network.id),
+                 update={'status': '0' if action == 'connect' else '2'})
+        )
+        return web.Response(body=self.serialize(network).encode(),
+                            content_type='application/json')