Xdesk/lib/chat.py

#! /usr/bin/python3
# -*- coding: utf-8 -*-

"""
This file is part of grandPA.

grandPA is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

grandPA is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with grandPA.  If not, see <http://www.gnu.org/licenses/>.

(c) 2012 micha.rathfelder@gmail.com
"""

import socket
import sys 
import time


class tcp_sender(object):
    def __init__(self,port=50000):
        self.port = port
        self.connect()
        
    def connect(self,client_name="unkown"):
        self.xip = "127.0.0.1" #raw_input("IP-Adresse: ")
        self.xs = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            self.xs.connect((self.xip, self.port)) #50000))
        except ConnectionRefusedError as e:
            print("ConnectionRefusedError: ", "ERR: {0} ".format(e.args) ,end="")
            print("Server nicht ereichbar/unterbrochen")
            time.sleep(1)
            self.connect()
        print("connected !")

    def send(self,nachricht):

        try:
            self.xs.send(bytes(nachricht+";","utf-8") )
        except socket.error as e:
            self.connect()
    def close(self):
        self.xs.close()



def dummyCB(msg):
    print("dummy_CB",msg)


def cmd(cb=dummyCB,port=50000):
    import socket
    import select

    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    #self.xs.getsockopt(socket.AF_INET, socket.SO_REUSEADDR )

    while 1:
        try:
            server.bind(("", port))
            break
        except Exception as e:
            print("except",e)
            print( "bind error")
            time.sleep(1)

    
    
    server.listen(1)

    clients = []
    clients2 = [""]*300
    try:
        while True:
            lesen, schreiben, oob = select.select([server] + clients,
                                                  [], [])

            for sock in lesen:
                if sock is server:
                    client, addr = server.accept()
                    client.setblocking(0)
                    clients.append(client)
                    print("+++ Client %s verbunden" % addr[0])
                    #sock.send("hi du")
                else:
                    msg=b''
                    try:
                        xmsg = sock.recv(1024)#5120)
                    except BlockingIOError as e:
                        pass#print( "exception",e)
                    try:
                        while xmsg: 
                            msg += xmsg
                            xmsg = sock.recv(1024)#5120)
                            xmsg = xmsg.replace(b";",b"")
                        #print(msg)
                    except BlockingIOError as e:
                        pass#print( "exception",e)

                    if not msg:
                        continue

                        
                    nachricht = msg
                    #print(msg)
                    nachricht = str(nachricht,"utf-8")
                    nachricht = nachricht.replace(";","")
                    nachrichten = nachricht.strip().replace("EOB","")
                    if "client_name:" in nachrichten:
                        if sock in clients:
                            client_nr = clients.index(sock)
                            clients2[client_nr] = nachrichten
                    if sock in clients:
                        client_nr = clients.index(sock)
                        #print(clients2[client_nr])
                    ip = sock.getpeername()[0]
                    #print(">>>", ip, nachrichten.split(";"))
                    if nachrichten:
                        tstamp = time.strftime("%H:%M:%S")

                        #print("from:",client_nr,">>>", tstamp , ip, nachrichten.split(";"))
                        for xx,nachricht in enumerate(nachrichten.split(";")):
                            cmd = nachricht #.split(" ")
                            #print(xx,cmd)
                            cb({"c":client_nr,"cmd":cmd})


                    else:
                        print("+++ Verbindung zu %s beendet" % ip)
                        sock.close()
                        if sock in clients:
                            client_nr = clients.index(sock)
                            clients2[client_nr] = ""
                        clients.remove(sock)
    except KeyboardInterrupt:
        print(" strg+c")
    finally:
        for c in clients:
            print(c,"close")
            c.close()
        server.close()
        print("server close")


if __name__ == "__main__":
    print( sys.argv ) 
    if sys.argv[1] == "server":
        cmd()
    elif sys.argv[1] == "client":
        c = tcp_sender()
        while 1:
            x = input(":: ")
            c.send(x)