在PostgreSQL下实现用户认证

问题

  • 如何利用PostgreSQL数据库实现一个用户认证系统?

解法

  • 用户认证系统有很多功能。在这个例子中,将展示如何在PostgreSQL数据库环境下一步一步完成一个用户认证系统

必需

  • 因为要用到make模板和postgreSQL数据库,所以要: import web from web.contrib.template import render_mako import pg

第一步:创建数据库

首先,为创建一个用户表。虽然这个表结构非常简单,但对于大部分项目来说都足够用了。

CREATE TABLE example_users
(
  id serial NOT NULL,
  user character varying(80) NOT NULL,
  pass character varying(80) NOT NULL,
  email character varying(100) NOT NULL,
  privilege integer NOT NULL DEFAULT 0,
  CONSTRAINT utilisateur_pkey PRIMARY KEY (id)
)

第二步:确定网址

登录和注销对应两个网址:

  • "Login" 对应登录页

  • "Reset" 对应注销页

urls = (
    '/login', 'login',
    '/reset', 'reset',
     )

第三步:判断用户是否登录

要判断用户是否已登录,是非常简单的,只要有个变量记录用户登录的状态即可。在login/reset类中使用这段代码:

def logged():
    if session.login==1:
        return True
    else:
        return False

第四步:简单的权限管理

我把我的用户划为四类:管理员,用户,读者(已登录),访客(未登录)。根据example_users表中定义的不同权限,选择不同的模板路径。

def create_render(privilege):
    if logged():
        if privilege==0:
            render = render_mako(
                directories=['templates/reader'],
                input_encoding='utf-8',
                output_encoding='utf-8',
                )
        elif privilege==1:
            render = render_mako(
                directories=['templates/user'],
                input_encoding='utf-8',
                output_encoding='utf-8',
                )
        elif privilege==2:
            render = render_mako(
                directories=['templates/admin'],
                input_encoding='utf-8',
                output_encoding='utf-8',
                )
    else:
        render = render_mako(
            directories=['templates/communs'],
            input_encoding='utf-8',
            output_encoding='utf-8',
            )
    return render

第五:登录(Login)和注销(Reset)的python类

现在,让我们用个轻松的方法来解决: - 如果你已登录,就直接重定向到login_double.html模板文件 - 否则,还是到login.html。

class login:
    def GET(self):
        if logged():
            render = create_render(session.privilege)
            return "%s" % (
                render.login_double()               )
        else:
            render = create_render(session.privilege)
            return "%s" % (
                render.login()
                )
  • 好了。现在写POST()方法。从.html文件中,我们得到表单提交的变量值(见login.html),并根据变量值得到example_users表中对应的user数据
  • 如果登录通过了,就重定向到login_ok.html。
  • 如果没通过,就重定向到login_error.html。
 def POST(self):
        user, passwd = web.input().user, web.input().passwd
        ident = db.query("select * from example_users where user = '%s'" % (user)).getresult()
        try:
            if passwd==ident[0][2]:
                session.login=1
                session.privilege=ident[0][4]
                render = create_render(session.privilege)
                return "%s" % (
                        render.login_ok()
                        )
            else:
                session.login=0
                session.privilege=0
                render = create_render(session.privilege)
                return "%s" % (
                    render.login_error()
                    )
        except:
            session.login=0
            session.privilege=0
            render = create_render(session.privilege)
            return "%s" % (
                render.login_error()
                )

对于reset方法,只要清除用户session,再重定向到logout.html模板页即可。

class reset:
    def GET(self):
        session.login=0
        session.kill()
        render = create_render(session.privilege)
        return "%s" % (
            render.logout()
            )

6th: 第六步:HTML模板帮助

嗯,我认为没有人想看这个,但我喜欢把所有的信息都提供出来。最重要的就是login.html。

<FORM action=/login method=POST>
    <table id="login">
        <tr>
            <td>User: </td>
            <td><input type=text name='user'></td>
        </tr>
        <tr>
            <td>Password: </td>
            <td><input type="password" name=passwd></td>
        </tr>
        <tr>
            <td></td>
            <td><input type=submit value=LOGIN></td>
        </tr>
    </table>
</form>

第七:问题或疑问?

  • 邮件:您可以联想我,我的邮箱是guillaume(at)process-evolution(dot)fr
  • IRC:#webpy on irc.freenode.net (pseudo: Ephedrax)
  • 翻译:我是法国人,我的英文不好...你可以修改我的文档(译注:哈哈,谦虚啥,你那是没见过wrongway的山东英文...)