タイトル通りだけど、ほんとに単純なソケット通信をやってみる。本当は通信をする際には例外処理や終端文字指定、スレッド処理などたくさんやらなければならないことがあるのだが、それは少しずつ実装することにして、まず基本からやろう。

今回はサーバ側とクライアント側を想定して二つのプログラムを作った。同じPC内でポート番号15000を使って通信をする。クライアント側はタイプされた文字列を次々にサーバ側に送り、サーバ側は受け取った文字列を表示するだけである。その内容は以下のようにしてみた。

simpleserver.py

import socket
host = ''
port = 15000
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind((host,port))
s.listen(1)
print 'Waiting for connections...'
clientsock, clientaddr = s.accept()
while 1:
rcvmsg = clientsock.recv(1024)
print 'You received -> %s' % (rcvmsg)
if rcvmsg == '':
break
clientsock.close()

simpleclient.py

import socket
host = 'localhost'
port = 15000
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((host,port))
while 1:
print 'Type message you want to send...'
msg = raw_input()
if msg == '':
s.close()
break
s.sendall(msg)

listen(1)というのは、処理がマルチスレッドではないとして、いくつまでクライアントを待たすことを許すのか、という設定らしい。今回は１つだけ許すということになる。recv(1024)というのはソケットからデータを10KBだけ読み取るということ（今回は問題になりませんが、もしこのプログラムで10KBを超える文字列を送信すると、上手くいかないはず）。

サーバ側を先に実行し、クライアント側を次に実行する。幾つかの文字列を入力した実行結果は以下。

実行結果（クライアント側）

C:\Python24>python.exe simpleclient.py
Type message you want to send...
Hello world!
Type message you want to send...
I love you!
Type message you want to send...
Python is a great language.
Type message you want to send...

実行結果（サーバ側）

C:\Python24>python.exe simpleserver.py
Waiting for connections...
You received -> Hello world!
You received -> I love you!
You received -> Python is a great language.
You received -> 

特に問題なく送れているようです。今後はこれを基に色々と処理を追加していこうかな。

そう言えば、まだディクショナリを使ったことがなかった。いまいち使うシーンが想像できなかったので使ってなかったんだけど、無理やりcsvファイルを扱うことに対して使ってみます。

以下のモジュールreadcsv.pyはファイルemp.csvを読み込み、時給×労働時間をデータごとに計算するという簡単なプログラム。

readcsv.py

def readcsv(filename):
L = [line[:-1] for line in file(filename).readlines()]
return map((lambda s: s.split(',')),L)
csvlist = readcsv('emp.csv')
for line in csvlist[1:]:
empdict = dict(zip(csvlist[0],line))
payment = int(empdict['WAGE']) * int(empdict['HOURS'])
print empdict['NAME'],'->',payment

まずreadcsv関数の中では、csvファイルの各行を行末の１文字（改行記号）を抜いた形で取得し、それぞれの行をカンマで分割してリストにする。返り値は「各行が要素となっているリスト、が要素となっているリスト」になる。ここでは練習のために、リスト内包表記を使ってみた。

実行部では、先ほど作ったリストの１番目の要素（ヘッダー）、と２番目以降の要素を

empdict = dict(zip(csvlist[0],line))

という部分でディクショナリ化する。あとはそれぞれの要素にヘッダーの名前でアクセス出来るようになるので、それを使って給与計算を行い結果を出力している。この方法であれば、CSVファイルのカラムの順番などにも影響されない処理が組めるはず。

emp.csv

NAME,WAGE,HOURS
John,1000,8
Tom,1200,9
Jack,1100,8
Nick,1200,9
Shelly,1200,7
Bill,1000,8
David,1100,9
Ron,1000,8
Kate,1000,9
Mary,1100,9

実行結果。

John -> 8000
Tom -> 10800
Jack -> 8800
Nick -> 10800
Shelly -> 8400
Bill -> 8000
David -> 9900
Ron -> 8000
Kate -> 9000
Mary -> 9900

これと同じ方法で、DBからSQLにて取ってきたデータの処理も出来るはずだ。

今は初めてのPython 第2版の第５部「モジュール」を勉強中なので、あまりコードを書く例がない。Pythonはリロードなどによりダイナミックにシステムの修正や改善を行えそうです。モジュールをディレクトリに分けるときは__init__.py作らなければいけないというのは忘れないようにしないと。

そろそろモジュール分けが必要になるくらいの、少し大きめのプログラムを組んでみてもいいかも。

この本もこの後はオブジェクト指向や例外処理の話になってくるので、Python特有と思われる部分の学習は終わりに近づいてきているのかな。そろそろ次のレベルに進みたいな。

以前書いた「文章内の単語数を調べる関数」も、文字列のsplit関数とfilter関数の組み合わせで簡単に書き直せた。ちなみにsplit関数とは、文字列を指定した区切り文字で分割し、それぞれをリストの要素として返す関数である。今回は与えられた文字列を空白で分割するようにしている。

def countwords(s):
return len(filter((lambda c: c != ''),s.split(' ')))

これを応用して、今度はあるファイル内にある単語数を調べる関数を作成してみる。引数はファイルパス。

def countwords(s):
return len(filter((lambda c: c != '' and c != '\n'),s.split(' ')))
def countwordsinfile(file):
L = map(countwords,open(file).readlines())
return reduce((lambda x,y: x + y),L)

countwords関数を少しいじったのは、例えば改行だけの行があったときに1文字と認識されてしまう現象が起きたため。open(file).readlines()はファイル内の全ての行をリストの形で返してくれる便利な関数。map関数でそれぞれに行に対してcountwords関数で単語数を調べ、最後にreduce関数で各要素を足していく。

読むファイルは以下。改行コードとEOFは表現のしようがなかったので、とりあえず手書きで加えておきました。

news.txt

The declaration came after the deaths of at least seven Palestinian civilians on a Gaza beach\n
from an apparently errant Israeli artillery shell.\n
\n
Delphi expanded its buyout deal to include all U.A.W. members, signaling a stepped-up effort\n
to avoid a strike.\n
[EOF]

実行結果。

>>> countwordsinfile('news.txt')
41

一つの単語がcondi- tion と二行に亘る場合などに対応できてませんが、とりあえず出来ましたね。

う、時間とネタがないので、今日はシーザー暗号でも作ってみます。暗号化する関数encryptcaesarと、復号する関数decryptcaesarを以下のように定義する。

caesar.py

def encryptcaesar(str,shift=1):
L = list(str)
return ''.join(map(chr,map((lambda s: ord(s) + shift),L)))
def decryptcaesar(str,shift=1):
return encryptcaesar(str,shift*-1)

文字列をリストに直すlist関数と、ある文字のASCIIコードを返すord関数、そしてmap関数を用いると一行で書けてしまいます。

実行結果。

>>> str = encryptcaesar('I love my wife!')
>>> str
'J!mpwf!nz!xjgf"'
>>> decryptcaesar(str)
'I love my wife!'
>>> str = encryptcaesar('Python is great',3)
>>> str
'S|wkrq#lv#juhdw'
>>> decryptcaesar(str,3)
'Python is great'
>>> 

いまどきこんなの暗号とは言えないけど、まあ基本ってことで。

楽しくなってきたんで続けます。

ゲームプログラムめも日記:平方根

会社のプログラム見ていたら、平方根を求める関数があったので、
Pythonで書いてみました。

（中略）

reduceを使えば、2行で書けることが判明しました…！

なんと便利な。早速真似して僕も実践。Newton法を用いてある数の平方根を求める関数mysqrtを作成する。繰り返しの回数が精度を決めるが、とりあえず10回を初期値としておく。

mysqrt.py

from __future__ import division
def mysqrt(x,loop=10):
return reduce((lambda a,x: a - (a**2 - x) / (2 * a)),[x]*loop)

実行結果。

>>> mysqrt(2)
1.4142135623730951
>>> mysqrt(3)
1.7320508075688772
>>> mysqrt(5)
2.2360679774997898
>>> mysqrt(10)
3.1622776601683791
>>> mysqrt(625)
25.0
>>> mysqrt(1024)
32.000000000000803

from __future__ import divisionをきちんと書かないと、割り算の結果が全て切り捨てられてしまうのでご注意を。今気付いたけど、これだと繰り返し9回が初期値か。まあいいや。

何年ぶりかに手書きでもニュートン法を計算してみましたが、プログラムのありがたさがよく分かります。

ちなみに以下のようにmathモジュールをインポートしても、sqrt関数が使えないんだけど何でなんだろう。マニュアル見てもこれで良さそうなのだが。

>>> import math
>>> sqrt(10)
Traceback (most recent call last):
File "<pyshell#56>", line 1, in -toplevel-
sqrt(10)
NameError: name 'sqrt' is not defined

追記：

ちなみに同じ関数をループ使って書くとこんな感じになった。やっぱりスマートじゃないな。

def mysqrt(x):
a = x
cnt = 0
while cnt < 10:
a = a - (a**2 - x) / (2 * a)
cnt += 1
return a