Многопоточность Python3. Как завершить выполнение скрипта, когда приходит команда для этого?

Question

[CeBePHblY]

Изучаю питон. Для обучения пишу клиент-сервер. К северу подключается много клиентов они с ним общаются. Но вот к серверу подключился админ и он посылает команду на сервер "adminexit", чтобы полностью завершить работу сервера. Попробовал сделать, но не получается, скрипт не завершается. Подскажите как сделать, чтобы работало.

#!/usr/bin/env python3
import socket
from threading import Thread, current_thread

host = "localhost"
port = 1605

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((host, port))
sock.listen(5)
print("ready")
running=True

def clientthread(conn):
	while True:
		global running
		buf = conn.recv(1024)
		buf = buf.rstrip()
		if buf.decode('utf8') == "exit":
			conn.send(b"bye")
			break
		elif buf.decode('utf8') == "get":
			conn.send(b"data")
		elif buf.decode('utf8') == "adminexit":
			conn.send(b"EXIT!")
			running=False
			break
		else:
			conn.send(b"listen")

while True:	
	if running==True:
		conn, addr = sock.accept()
		print("client connected with address " + addr[0])
		conn.send(b"hello!")
		ht = Thread(target=clientthread, args=(conn,))
		ht.start()
	else:
		break
		
conn.close()
sock.close

Answer 1

[nirvimel]

import os
os._exit(0)

Comments

[CeBePHblY]

как я понял, sys.exit() надо поставить после строчки running=False, это завершает поток правильно? но мне кажется, что проблема еще в строке conn, addr = sock.accept(), так как сервер все время слушает порт, пока клиент не подключится и висит на строчке conn, addr = sock.accept(), при приходе команды "adminexit", значение переменной running меняется, но сервер то дальше conn, addr = sock.accept() не выполняется., только после подключения нового клиента происходит проверка значения running, и сервер закрывается. вот как это обойти?

[nirvimel]

CeBePHblY: Нет, можно написать:

if buf.decode('utf8') == "adminexit":
     sys.exit()

Это прихлопнет весь процесс со всеми потоками.

[CeBePHblY]

#!/usr/bin/env python3
import socket
from threading import Thread, current_thread
import sys

host = "localhost"
port = 1605

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((host, port))
sock.listen(5)
print("ready")

def clientthread(conn):
	while True:
		buf = conn.recv(1024)
		buf = buf.rstrip()
		if buf.decode('utf8') == "exit":
			conn.send(b"bye")
			break
		elif buf.decode('utf8') == "get":
			conn.send(b"data")
		elif buf.decode('utf8') == "adminexit":
			conn.send(b"EXIT!")
			sys.exit()
		else:
			conn.send(b"listen")
	
while True:	
	conn, addr = sock.accept()
	print("client connected with address " + addr[0])
	conn.send(b"hello!")
	ht = Thread(target=clientthread, args=(conn,))
	ht.start()
		
conn.close()
sock.close

все равно сервер не отключается по команде

[nirvimel]

CeBePHblY: Чем команду оправляете?

[CeBePHblY]

а продолжает слушать порт

[CeBePHblY]

nirvimel: в клиенте пишу "adminexit"

[nirvimel]

CeBePHblY: Все верно sys.exit() не может прервать основной поток, застрявший в вызове sock.accept()

Тогда используем более мощное средство - os._exit(0)

[CeBePHblY]

nirvimel: спасибо, работает. а может правильнее будет
conn, addr = sock.accept()
print("client connected with address " + addr[0])
conn.send(b"hello!")
ht = Thread(target=clientthread, args=(conn,))
ht.start()
вынести в отдельный поток? и все время проверять running, как только оно false, clientthread завершится по sys.exit(), и завершить поток с кодом выше?

[CeBePHblY]

вот так вроде работает:

#!/usr/bin/env python3
import socket
from threading import Thread, current_thread
import sys

host = "localhost"
port = 1605

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((host, port))
sock.listen(5)
print("ready")
running=True

def clientthread(conn):
while True:
global running
buf = conn.recv(1024)
buf = buf.rstrip()
if buf.decode('utf8') == "exit":
conn.send(b"bye")
break
elif buf.decode('utf8') == "get":
conn.send(b"data")
elif buf.decode('utf8') == "adminexit":
conn.send(b"EXIT!")
running=False
conn.close()
sys.exit()
else:
conn.send(b"listen")

def connectthread(sock):
while True:
conn, addr = sock.accept()
print("client connected with address " + addr[0])
conn.send(b"hello!")
ht = Thread(target=clientthread, args=(conn,))
ht.start()

while True:
if running==True:
ht2 = Thread(target=connectthread, args=(sock,))
ht2.start()
else:
sock.close
sys.exit()
print("IS EXIT!")

будет ли это лучше?

[nirvimel]

CeBePHblY:

while True:
if running==True:
ht2 = Thread(target=connectthread, args=(sock,))
ht2.start()

У вас тут в цикле порождение бесконечного количества потоков!

[nirvimel]

CeBePHblY: Оформляйте код через <code> </code> а то невозможно читать.

[CeBePHblY]

nirvimel: переделал вот так:

#!/usr/bin/env python3
import socket
from threading import Thread, current_thread

host = "localhost"
port = 1605

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((host, port))
sock.listen(5)
print("ready")
running=True

def clientthread(conn):
	while True:
		global running
		buf = conn.recv(1024)
		buf = buf.rstrip()
		if buf.decode('utf8') == "exit":
			conn.send(b"bye")
			break
		elif buf.decode('utf8') == "get":
			conn.send(b"data")
		elif buf.decode('utf8') == "adminexit":
			conn.send(b"IS EXIT!")
			running=False
			conn.close()
			break
		else:
			conn.send(b"listen")

def connectthread(sock):	
	while True:	
		conn, addr = sock.accept()
		print("client connected with address " + addr[0])
		conn.send(b"hello!")
		ht = Thread(target=clientthread, args=(conn,))
		ht.start()
		
ht2 = Thread(target=connectthread, args=(sock,))
ht2.start()

while True:
        if running==False:
                sock.close
                print("IS EXIT!")
                break

все отлично работает, но теперь сервер включается, выводит "ready", но при подключении клиентов не выводит строку "client connected with address...."(при этом окно сервера как бы зависает, его нельзя передвинуть(OS windows)), при этом отвечает клиентам на команды и завершатся при посылки команды "adminexit". не смертельно, но хотелось бы разобраться почему и исправить это.

[nirvimel]

CeBePHblY:
Чтобы код не зависал в вызове sock.accept(), можно включить таймаут. В этом случае процесс завершится не позднее чем через 1 секунду (в моем примере) после установки running = False.

Надеюсь, вы понимаете смысл приведенного кода, как и почему он завершается и почему без этого он завершаться не мог.

import socket
from threading import Thread

host = "localhost"
port = 1605

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

sock.settimeout(1)  # 1 second I/O timeout

sock.bind((host, port))
sock.listen(5)
print("ready")
running = True


def clientthread(conn):
    global running
    while True:
        buf = conn.recv(1024)
        buf = buf.rstrip()
        # for debug
        print ('received : ' + buf)
        if buf.decode('utf8') == "exit":
            conn.send(b"bye")
            conn.close()
            break
        elif buf.decode('utf8') == "get":
            conn.send(b"data")
        elif buf.decode('utf8') == "adminexit":
            conn.send(b"EXIT!")
            conn.close()
            running = False
            break
        else:
            conn.send(b"listen")


while running:
    try:
        conn, addr = sock.accept()
        print("client connected with address " + addr[0])
        conn.send(b"hello!")
        ht = Thread(target=clientthread, args=(conn,))
        ht.start()
    except socket.timeout:
        pass

sock.close()

[CeBePHblY]

nirvimel: запустил ваш код, отправил с клиента adminexit - сервер продолжает работать, при чем подключает после этого новых клиентов и с ними общается как ни в чем не бывало. кстати мой код выше, точно так же ведет себя...

[nirvimel]

CeBePHblY: Странно, сейчас только проверял - все работает.
У вас случайно в системе не осталось предыдущих запущенных экземпляров на том же порту?
Какая версия интерпретатора?
Что пишет в консоли при отправке "adminexit"?

Да, там еще одна мелкая ошибка:print (b'received : ' + buf) в этой строке без b не будет работать на Python 3.

[CeBePHblY]

Вот запускаю ваш сервер и своего клиента

[nirvimel]

CeBePHblY: Я на Windows не тестировал, возможно там чуть по-другому работают сокеты.
Возможно, это поможет, если завернуть conn.recv() в try...except socket.timeout:

try:
            buf = conn.recv(1024)
        except socket.timeout:
            continue

Если вылетает с ошибкой, то дайте Traceback с текстом ошибки.

Answer 2

[epolyak]

Должно помочь

....
ht = Thread(target=clientthread, args=(conn,))
ht.daemon = True
ht.start()
.....