內(nèi)容
- 完成dict plugin的服務(wù)端啟動(dòng)socket監(jiān)聽
- 將Project使用setuptools進(jìn)行打包
- 生成客戶端程序save_word
- 使用argparse解析命令行參數(shù)
倉庫代碼中的day2分支 https://github.com/bymzy/remind-you/tree/day2
效果
使用python setup.py bdist_rpm生成rpm包替蛉，然后使用rpm -ivh remind_you-0.1-1.noarch.rpm安裝庐杨。python庫目錄中將會(huì)安裝remind_you包隶症，同時(shí)會(huì)在系統(tǒng)目錄中增加可執(zhí)行文件save_word句狼。
啟動(dòng)dict plugin服務(wù)端:
執(zhí)行save_word命令
可以看到十偶，客戶端執(zhí)行save_word命令后，服務(wù)端能夠收到客戶端的消息园细。
使用的庫
setuptools
使用setuptools對(duì)整個(gè)代碼打包(個(gè)人比較喜歡打成rpm包python setup.py bdist_rpm惦积，方便安裝刪除)。同時(shí)猛频，使用setuptools生成可執(zhí)行文件(save_word)狮崩。倉庫最上層目錄下的setup.py文件記錄了配置信息，代碼如下：
from setuptools import setup, find_packages
setup(
name = 'remind_you',
version = '0.1',
packages = find_packages(),
package_data = {
'':['*.conf']
},
data_files = [('/etc/', ['remind_you/plugins/scheduler.conf'])],
author = 'mingzhengying',
description = 'remind you something!!!',
entry_points = {
'console_scripts': [
'save_word = remind_you.cmds:save_word_cmd',
]
}
)
setuptools是distutils的加強(qiáng)版本鹿寻，上述配置文件中重要部分說明如下
- packages = find_packages() 表示在當(dāng)前setup.py同等目錄下尋找源文件
- package_data 部分表示將所有的包目錄下*.conf文件進(jìn)行打包
- data_files用來將配置文件睦柴，安裝到/etc目錄下
- console_scripts用來生成命令行執(zhí)行程序，程序的執(zhí)行入口是remind_you.cmds.save_word_cmd這個(gè)函數(shù)
socket毡熏， argparse
使用最簡(jiǎn)單的socket阻塞模型坦敌。dict plugin讀取配置文件，然后監(jiān)聽在特定的端口上∮剑客戶端命令運(yùn)行后同樣讀取配置文件杜顺，連接服務(wù)器端。
dict plugin服務(wù)端代碼如下：
def generate_request(_word, _trans):
trans = pack_int(_trans)
word = pack_str(_word)
totalLen = 4 + len(trans) + len(word)
return pack_int(totalLen) + trans + word
def recv_request(cl):
tmp = recv_len(cl, 4)
wordLen = unpack_int(tmp)
data = recv_len(cl, wordLen)
trans = unpack_int(data[0:4])
word = unpack_str(data[4:])
return word, trans
def run(args):
print 'my pid is: %u , %s ' % (os.getpid(), args)
port = args.get('port')
ip = args.get('ip')
dbPath = args.get('db')
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((ip, port))
sock.listen(5)
while True:
cl, addr = sock.accept()
word, trans = recv_request(cl)
print 'receive word %s , need trans %d: ' % (word, trans)
cl.close()
run方法作為dict plugin進(jìn)程的主方法蘸炸，會(huì)被主進(jìn)程啟動(dòng)躬络，然后監(jiān)聽在某個(gè)端口上。后續(xù)有客戶端連接后搭儒，從客戶端的輸入中解析出請(qǐng)求的word, trans穷当，分別表示單詞，和是否返回翻譯信息淹禾。目前的實(shí)現(xiàn)中馁菜，服務(wù)端僅僅會(huì)將得到的信息打印出來，然后斷開連接等待下個(gè)請(qǐng)求稀拐。
客戶端代碼如下：
cmds.py
from plugins import dict_cmd
def save_word_cmd():
dict_cmd.run()
setup.py配置中火邓，生成的客戶端命令save_word最終會(huì)調(diào)用save_word_cmd丹弱，進(jìn)而調(diào)用dict_cmd.run(）方法德撬。
dict_cmd.py
CONF = '/etc/scheduler.conf'
PLUGIN_NAME = 'dict'
def get_ip_port():
with open(CONF) as f:
data = f.read()
di = json.loads(data)
for plugin in di.get('plugins', []):
if plugin.get('name') == PLUGIN_NAME:
return plugin.get('args').get('ip'), plugin.get('args').get('port')
return None, None
def parse_cmd_args():
args_parser = argparse.ArgumentParser()
args_parser.add_argument("-w", "--word", type=str, required=True);
args_parser.add_argument('-t', '--trans', action='store_true', default=False);
args = args_parser.parse_args()
#use vars to get a dict
#print vars(args), type(vars(args))
return args.word, args.trans
def run():
ip, port = get_ip_port()
word, trans = parse_cmd_args()
s = connect_ip_port(ip, port)
if s is None:
print 'connect to server %s:%d failed!!!' % (ip, port)
sys.exit(1)
req = generate_request(word, trans)
send_data(s, req)
客戶端的save_word命令，調(diào)用dict_cmd.run()方法后會(huì)做如下事情：
- 讀取配置文件得到服務(wù)端的ip躲胳，port
- 解析命令行參數(shù)
- 生成請(qǐng)求req數(shù)據(jù)
- 調(diào)用send_data將req數(shù)據(jù)發(fā)送給服務(wù)端
send_data, connect_ip_port等函數(shù)在util.py中定義
