1. 程式人生 > 實用技巧 >python常駐任務接收外界引數

python常駐任務接收外界引數

實現一個簡單的常駐任務是容易的,寫一個while True就可以了,但是同時要能接收外界傳來的引數,就稍微有點複雜.

我第一個想到的是網路程式設計中的server/client模式

server作為常駐程序,可以接受client傳來的各種引數進行處理

服務端:server.py

import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.bind(('127.0.0.1', 10001)) s.listen(5) def do_some(): pass def do_other(): pass def tcplink(sock, addr):
print('Accept new connection from %s:%s...' % addr) sock.send(b'Welcome!') while True: data = sock.recv(1024) time.sleep(1) if not data or data.decode('utf-8') == 'exit': break data_str = data.decode('utf-8') data_list = data_str.split(',')
if len(data_list) == 3: do_some() else: do_other() sock.send(('processing, %s!' % s3).encode('utf-8')) sock.close() if __name__ == '__main__': while True: # 接受一個新連線: data, addr = s.accept() # 建立新執行緒來處理TCP連線: t = threading.Thread(target=tcplink, args=(data, addr)) t.start()

服務端實現完了,下面我們來實現客戶端client.py,客戶端支援傳入引數或從檔案讀取引數,來向客戶端傳送請求

import socket
import sys

if __name__ == '__main__':
    param = sys.argv[1]
    c_list = []
    if param.endswith('txt'):
        with open(param, 'r') as file:
            for line in file.readlines():
                c_list.append(line.strip().encode('utf8'))
    else:
        c = param.encode('utf8')
        c_list.append(c)
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.connect(('127.0.0.1', 10001))
    print(s.recv(1024).decode('utf-8'))
    for data in c_list:
        s.send(data)
        print(s.recv(1024).decode('utf-8'))
    s.send(b'exit')
    s.close()

這樣通過兩個指令碼,一個實現了常駐程序,一個實現了傳入引數,就可以不斷接受外界請求