forked from zhoufeng1989/blog
-
Notifications
You must be signed in to change notification settings - Fork 0
/
user.py
43 lines (35 loc) · 1.23 KB
/
user.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import web
from db import db
from util import gen_password
def check_register(username, email):
sql = (
r'select count(*) as count from users where username = $username '
r'or email = $email')
result = db.query(sql, vars={'username': username, 'email': email})
return result[0]['count'] == 0
def register_user(username, password, email):
password, salt = gen_password(password)
return db.insert(
'users', **{
'username': username, 'password': password,
'email': email, 'salt': salt,
'create_time': web.SQLLiteral("unix_timestamp(NOW())")}
)
def check_login(username, password):
user = get_user_by_name(username)
print user
sql = (
r'select count(*) as count from users where username = $username '
r'and password = md5(concat($salt, md5($password)))')
result = db.query(
sql, vars={'username': username, 'password': password,
'salt': user['salt']})
return result[0]['count'] != 0
def get_user_by_name(username):
sql = (
r'select * from users where username = $username ')
result = db.query(sql, vars={'username': username})
if result:
return result[0]
else:
return {}