快捷導(dǎo)航

Python使用htpasswd實(shí)現(xiàn)基本認(rèn)證授權(quán)的例子

更新時(shí)間：2014年06月10日 09:31:24 作者：

這篇文章主要介紹了Python使用htpasswd實(shí)現(xiàn)基本認(rèn)證授權(quán)的例子,服務(wù)器使用的是mini_httpd,需要的朋友可以參考下

前面我講解了如何將樹(shù)莓派（Raspberry Pi）打造成無(wú)線(xiàn)路由，感覺(jué)每次通過(guò)命令ssh管理顯麻煩，于是自己動(dòng)手編寫(xiě)Web界面，主要是使用Python編寫(xiě)的CGI程序，這里用到了mini_httpd這款輕量的Web服務(wù)器，本來(lái)想裝nginx的，但是想想還是精簡(jiǎn)一些吧，畢竟資源有限，況且Web管理界面僅我一個(gè)人訪(fǎng)問(wèn)。

CGI應(yīng)用跑起來(lái)了，但問(wèn)題來(lái)了，如何實(shí)現(xiàn)普通路由的那種打開(kāi)頁(yè)面就彈出輸入用戶(hù)名密碼的對(duì)話(huà)框？

這里主要用到HTTP協(xié)議的一個(gè)知識(shí)，那就是HTTP基本認(rèn)證。

服務(wù)器端通過(guò)發(fā)送類(lèi)似下面的頭信息來(lái)實(shí)現(xiàn)需要認(rèn)證請(qǐng)求：

復(fù)制代碼代碼如下:

HTTP/1.0 401 Authorization Required
WWW-Authenticate: Basic realm="Secure Area"
Content-Type: text/html

針對(duì)上述要求，于是我在CGI中采用了如下的Python代碼：

復(fù)制代碼代碼如下:

def check_login():
    import base64

    if "Authorization" in os.environ:
    try:
        cred = base64.b64decode(os.environ['Authorization'].split(' ')[1])
        username, password = cred.split(":")
        if db_validate_user(username, password): # 這里匹配數(shù)據(jù)庫(kù)用戶(hù)名密碼
        return True
    except:
        pass

    print 'Status: 401 Unauthorized'
    print 'Pragma: no-cache'
    print 'Content-Type: text/html'
    print 'WWW-Authenticate: Basic realm=\"My Wireless Router\"'
    print
    print """
    <html>
    <head>
        <title>Not authenticated</title>
    </head>
    <body>
    <h1>Not authenticated.</h1>
    </body>
    </html>"""
    return False

# 調(diào)用
if not check_login():
    sys.exit(0)

但是實(shí)際操作下來(lái)后發(fā)現(xiàn)mini_httpd并不轉(zhuǎn)發(fā)來(lái)自用戶(hù)的Authorization的用戶(hù)名和密碼，也就是說(shuō)os.environ取不到這個(gè)頭信息，從而導(dǎo)致認(rèn)證失敗。

經(jīng)過(guò)網(wǎng)上搜索后得知mini_httpd原生支持通過(guò).htpasswd實(shí)現(xiàn)簡(jiǎn)單認(rèn)證的技術(shù)，也就是說(shuō)我們可以在需要授權(quán)訪(fǎng)問(wèn)的目錄下建立.htpasswd文件實(shí)現(xiàn)，當(dāng)然這個(gè)文件是有格式要求的，我們可以通過(guò)htpasswd命令來(lái)創(chuàng)建。這個(gè)命令一般Apache服務(wù)器軟件會(huì)自帶，不過(guò)mini_httpd也自帶了，所以你可以直接使用這個(gè)命令。

復(fù)制代碼代碼如下:

#        建立文件名 賬戶(hù)名 密碼
htpasswd -bc .htpasswd admin 123456

當(dāng)一個(gè)目錄下有.htpasswd文件時(shí)，mini_httpd就會(huì)彈出要求用戶(hù)名和密碼的對(duì)話(huà)框，輸入正確后才可以瀏覽，如果沒(méi)有這個(gè)文件則正常瀏覽。

因?yàn)槲业腸gi應(yīng)用是基于Python的，所以我希望Python能夠管理.htpasswd文件，幸好Python世界里有現(xiàn)成的庫(kù)，避免了我們重復(fù)造輪子，使用easy_install的安裝方式如下：

復(fù)制代碼代碼如下:

sudo easy_install htpasswd

官方文檔給出的例子如下，感覺(jué)操作挺方便的，大家可以試一試：

復(fù)制代碼代碼如下:

import htpasswd

with htpasswd.Basic("/path/to/user.db") as userdb:

    try:
        userdb.add("bob", "password")
    except htpasswd.basic.UserExists, e:
        print e
    try:
        userdb.change_password("alice", "newpassword")
    except htpasswd.basic.UserNotExists, e:
        print e

with htpasswd.Group("/path/to/group.db") as groupdb:

    try:
        groupdb.add_user("bob", "admins")
    except htpasswd.group.UserAlreadyInAGroup, e:
        print e
    try:
        groupdb.delete_user("alice", "managers")
    except htpasswd.group.UserNotInAGroup, e:
        print e