urllib.urlopenはステータス404でも例外を発生してくれない。
urllib2.urlopenは404で例外を発生してくれるけれど、そのままだと環境変数のプロキシ設定を参照するようで、ちょっと困る場合があった。
というわけで、urllib2.urlopenでプロキシを設定|参照しないようにする方法。

#!/usr/bin/env python
"""Python urllib2 proxy usage sample
"""
import urllib2

#今回はプロキシ設定を空にしておく
#proxies = {'http': 'http://www.example.com:3128/'}
proxies = {}

#プロキシハンドラーを作成して
handler = urllib2.ProxyHandler(proxies)

#プロキシハンドラーを指定してURL Openerを作成して
opener = urllib2.build_opener(handler)

#作成したURL Openerをインストールしてから
urllib2.install_opener(opener)

#普通にurlopenすると例外が発生してくれる
try:
    u = urllib2.urlopen('http://www.example.com/404NotFound.html')
    print u.read()
except IOError,error:
    print error

Gist:227549

Pythonカテゴリ作ってみた。

MoinMoinを仕事で使いたい。今すでに使っている社内の認証機構を使ってユーザ管理を楽するためにはどうすればいいか？ Pythonのちょっとしたスクリプトを書けばなんとかなるようだ。

ということで調べてみたら、

1. wikiconfig.pyのConfigクラス内でカスタムスクリプトを使うように設定する。
2. ログイン用のカスタムスクリプトを書いて、MoinMoin/auth/以下に置く。

の２つの手順でできることがわかった。MoinMoin-1.6で動くことを確認した。

以下、やってみて上手くいった手順を記録しておく。

この例では、

• カスタムスクリプト内にユーザ情報を保持している。この部分を必要な認証機構を呼び出すようにすれば使えるはず。
• メールアドレスと別名もMoinMoinのプロファイル外で管理していることにして、カスタムスクリプト内で設定管理している。

wikiconfig.pyの編集

class Config(DefaultConfig):
    """MoinMoinのカスタム認証スクリプトmyauth.loginを使うようにwikiconfig.py
　　　 で認証方法を指定。
    """
    from MoinMoin.auth import moin_session,myauth
    auth = [myauth.login, moin_session]
    #MoinMoin内のPreferenceを自動的に作成する
    user_autocreate = True

Gist:232443

カスタム認証スクリプト MoinMoin/auth/myauth.py

# -*- coding: utf-8 -*-
"""MoinMoin custom user authentication sample
"""
from MoinMoin import user
#User Database
_USER_MAP = {'user1,pass1' : ('user1@example.com',u'user1'),
    'user2,pass2' : ('user2@example.com','user2')}

def login(request, **kw):
	username = kw.get('name')
	password = kw.get('password')
	login = kw.get('login')
	user_obj = kw.get('user_obj')

	if not login:
		return user_obj, True

	u = None
	user_info = _USER_MAP.get("%s,%s" % (username,password),False)

	if user_info:
		u = user.User(request,
		name=username,
		auth_username=username,
		password=password,
		auth_method='myauth',
		auth_attribs=('name', 'auth_username', 'password', 'email', 'aliasname', ))
		u.email = user_info[0]
		u.aliasname = user_info[1]
		u.create_or_update(True)
		request.log("Login OK. user=%s, email=%s, aliasname=%s." % (username,user_info[0],user_info[1]))

	return u, True

Gist:227564

python-ldapを使ってみた。

ついでに、以前から試そうと思って手をつけていなかった、ActiveDirectoryのLDAP認証も。試しに、bindしてみるだけ。平文でパスワードが流れるのでこのままではちょっと使えない。

#!/usr/bin/env python
# -*- coding: utf-8-*-
# python-ldapを使ってActiveDirectoryで認証してみる
# http://python-ldap.sourceforge.net/

def ADAuth(username,password,host,port=389):
    """ActiveDirectoryのドメイコントローラにユーザ名とパスワードでbindしてみる。
    bindできれば認証OK。認証NGなら例外が起きる。
    """
    import ldap
    url = "ldap://%s:%d" % (host,port)
    l = ldap.initialize(url)
    l.simple_bind_s(username,password)
    l.unbind_s()

if __name__=='__main__':
    if len(sys.argv)<3:
        print "Usage: %s username password" % (sys.argv[0])
    else:
        try:
            userid = sys.argv[1]
            password = sys.argv[2]
            ADAuth(userid,password,"adserver.example.com")
            print "OK"
        except:
            print "NG"

Gist:227565