bricknil.sockets

Complexity

Total Complexity

5

Size/Duplication

Total Lines	61
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	WebClient.__init__()	0	6	1
A	WebClient.run()	0	9	3

# Copyright 2019 Virantha N. Ekanayake 
# 
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
# http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Experimental socket updates of devices and hubs"""

import logging
import pprint
from curio import run, spawn,  sleep, Queue, tcp_server

#async def socket_server(web_out_queue, address):
    #sock = socket(AF_INET, SOCK_STREAM)
    #sock.setsockopt(SOL_SOCKET, SO_REUSEADDR,1)
    #sock.bind(address)
    #sock.listen(5)
    #print(f'Server listening at {address}')
    #async with sock:
        #while True:
            #client, addr = await sock.accept()
            #wc = WebClient(client, addr, web_out_queue)
            #await spawn(wc.run, daemon=True)

async def bricknil_socket_server(web_out_queue, address):
    async def web_client_connected(client, addr):
        print('connection from', addr)
        wc = WebClient(client, addr, web_out_queue)
        await wc.run()

    task = await spawn(tcp_server, '', 25000, web_client_connected, daemon=True)


class WebClient:

    def __init__(self, client, addr, in_queue):
        assert in_queue is not None
        self.in_queue = in_queue
        self.client = client
        self.addr = addr
        print(f'Web client {client} connected from {addr}')
        

    async def run(self):

        async with self.client:
            while True:
                msg = await self.in_queue.get()
                #print(f'Webclient queue got: {msg}')
                await self.in_queue.task_done()
                await self.client.sendall(msg)
        print('connection closed')