def hash_password(password):
"""
Hashes a password using pbkdf2_sha512
:param password: The sha512 password from the login/register form
:return: A sha512 -> pbkdf2_sha512 encrypted password
"""
return pbkdf2_sha512.encrypt(password)
def encrypt_password(password):
"""
It receives the password from the form-size and encrypts it with pbkdf2
:param password: sha512 password from the site
:return: A sha512 pbkdf2 encrypted password
"""
return pbkdf2_sha512.encrypt(password)
def hash_password(password):
"""
:param password:
:return:
"""
return pbkdf2_sha512.encrypt(password)
def post(self):
uid = self.get_argument("uid", "")
level = int(self.get_argument('level', 0))
jinbi = int(self.get_argument('jinbi', 0))
money = int(self.get_argument('money', 0))
phone = self.get_argument('phone', "")
id_code = self.get_argument('id_code', "")
username = self.get_argument("username", "")
real_name = self.get_argument("real_name", "")
pwd = self.get_argument("pwd", "")
alipay = self.get_argument("alipay", "")
# user = self.db.user.find_one({"$or": [{'uid': uid}, {"phone": phone}, {'username': username}]})
user = self.db.user.find_one({'uid': uid})
if user:
user = {
'phone': phone,
'username': username,
'real_name': real_name,
'level': level,
'jinbi': jinbi,
'money': money,
'id_code': id_code,
'alipay': alipay,
}
if pwd:
password_hash = pbkdf2_sha512.encrypt(pwd)
user['pwd'] = password_hash
self.db.user.update({"uid": uid}, {"$set": user})
return self.render("ok.html", myuser=self.user, url="/admin/userlist", tip=u"资料修改成功")
else:
# baseurl = "http://cdn.zi-han.net/im/temp/"
user = {
'uid': phone,
'pwd': pwd,
'phone': phone,
'username': username,
'level': level,
'jinbi': jinbi,
'money': money,
'id_code': id_code,
'alipay': alipay,
'regtime': time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time())),
'safe_pwd': pwd,
'is_check': False,
'is_active': True,
'reward': 0,
}
# print 'user%s'%user
self.logging.info(('register user %s %s' % (user['uid'], user['pwd'])))
res = self.application.auth.register(user)
if not res:
print "register error"
self.render("ok.html", myuser=self.user, url="/admin/adduser", tip=u"注册失败")
self.redirect('/admin/userlist')
def import_user_profile(conn):
if not conn:
conn = MongoClient('127.0.0.1')
srcdb = conn.get_database('kline')
# srcdb.authenticate("admin","rss123")
dstdb = conn.get_database('playkline')
# dstdb.authenticate("admin","rss123")
dstdb.tb_user_profile.remove()
users = srcdb.user.find({"status": "online"}, {"_id": 0})
for v in users:
logins = []
login = {}
logininfo = v.get('login', None)
# print logininfo
if logininfo:
if isinstance(logininfo, list):
logins = logininfo
else:
login['ip'] = logininfo.get('ip', 0)
login['time'] = logininfo.get('time', 0)
login['bonus_state'] = logininfo.get('reeceive_state', 0)
logins.append(login)
playtimes = int(v.get('gametimes', 0))
wintimes = float(v.get('wintimes'))
if playtimes < 1:
winrate = 0
else:
winrate = round((wintimes / float(playtimes) * 100), 1)
passwd = pbkdf2_sha512.encrypt("rss123")
u = {
"userid": v.get("uid"),
"nick": v.get("nick", v.get('uid')),
"avatar": v.get("avatar"),
"score": v.get("integral"),
"gold": v.get("gold"),
"playtimes": v.get("gametimes", 0),
"wintimes": v.get("wintimes", 0),
"passwd": passwd,
"mobile": v.get("phone"),
"email": v.get("email", ""),
"login": logins,
"regtime": v.get("regtime"),
"status": v.get("status"),
"settings": v.get("setting"),
"logindays": v.get("logindays", 0),
"winrate": winrate
}
print u
dstdb.tb_user_profile.insert(u)
def update_admin_pwd():
"""更新管理员密码"""
db = database.database.getDB()
correct = False
while not correct:
name = raw_input("name:")
passwd = raw_input("password:"******"confirm password:"******"userid": name},
{"$set": {
"passwd": password_hash
}})
else:
print "password not correct,please input again"
def create_user():
"""创建用户"""
correct = False
db = database.database.getDB()
while 1:
phone = raw_input("phone: ")
exist_user = db.user.find_one({"phone": phone})
if exist_user:
print "the phone %s is existed" % phone
else:
break
user = []
nick = raw_input("name:")
while not correct:
passwd = raw_input("password:"******"confirm password:"******"userid", pymongo.DESCENDING).limit(1)
userid = id_auto_increment(last, "userid")
user = {
'userid': userid,
'nick': nick, # 用户名
'avatar': 1, # 头像
'gold': 2800, # 金币数
'integral': 100, # 积分
'gametimes': 0, # 比赛次数
'wintimes': 0, # 赢得次数
'logindays': 0, # 连续登录天数
'phone': phone,
'passwd': passwd,
'regtime': time.strftime('%Y-%m-%d',
time.localtime(time.time())), # 注册时间
}
else:
print "password not correct,please input again"
password_hash = pbkdf2_sha512.encrypt(user['passwd'])
user['passwd'] = password_hash
record = db.user.find_one({"userid": user['userid']})
if record is not None:
return False
db.user.insert(user)
def create_admin():
"""创建管理员账号"""
correct = False
db = database.database.getDB()
while 1:
username = raw_input("username: "******"uid": username})
if exist_user:
print "the user %s is existed" % (username)
else:
break
user = []
while not correct:
pwd = raw_input("password:"******"confirm password:"******"a admin user registered,name is %s") % (username))
user = {
"uid":"18672196620",
"username": username,
"pwd": pwd,
"phone":"18672196620",
"safe_pwd": pwd,
'jinbi':10000,
'money':100000,
"role": "superadmin",
'regtime': time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time())),
}
else:
print "password not correct,please input again"
password_hash = pbkdf2_sha512.encrypt(user['pwd'])
user['pwd'] = password_hash
record = db.user.find_one({"uid": user['uid']})
if record is not None:
logging.warn(("that user name %s had been registered") % (username))
return False
db.user.insert(user)
def update_user_pwd():
"""更新用户密码"""
correct = False
db = database.database.getDB()
while 1:
username = raw_input("username: "******"userid": username})
if exist_user:
print "the user %s is existed" % (username)
break
while not correct:
passwd = raw_input("password:"******"confirm password:"******"userid": username},
{"$set": {
"passwd": password_hash
}})
else:
print "password not correct,please input again"
def import_robert_profile(conn):
if not conn:
conn = MongoClient('127.0.0.1')
srcdb = conn.get_database('playkline')
# srcdb.authenticate("admin","rss123")
dstdb = conn.get_database('playkline')
# dstdb.authenticate("admin","rss123")
dstdb.tb_robert_profile.remove()
mobiles = [
"13509673370", "13809673370", "13609673370", "13709673370",
"13309673370", "13409673370"
]
nicknames = ["rokia", "penny", "sophie", "wiley", "donny", "sonny"]
packages = srcdb.tb_user_profile.find({}, {"_id": 0})
for v in packages:
print v
u = v
u["mobile"] = mobiles.pop()
u["passwd"] = pbkdf2_sha512.encrypt("rss123")
u["nick"] = nicknames.pop()
u["userid"] = u["nick"]
dstdb.tb_robert_profile.insert(u)
def create_admin():
"""创建管理员账号"""
correct = False
db = database.database.getDB()
while 1:
admin_name = raw_input("admin name: ")
exist_admin = db.tb_system_user.find_one({"userid": admin_name})
if exist_admin:
print "the Admin %s is existed" % admin_name
else:
break
admin = []
while not correct:
passwd = raw_input("password:"******"confirm password:"******"A admin registered,name is %s" % admin_name)
last = db.tb_system_user.find()
userid = id_auto_increment(last, "userid")
admin = {
"userid": userid,
"name": admin_name,
"passwd": passwd,
"role": "superadmin",
'regtime': time.strftime('%Y-%m-%d',
time.localtime(time.time())),
}
else:
print "password not correct,please input again"
password_hash = pbkdf2_sha512.encrypt(admin['passwd'])
admin['passwd'] = password_hash
record = db.tb_system_user.find_one({"userid": admin['userid']})
if record is not None:
logging.warn("that Admin name %s had been registered" % admin_name)
return False
db.tb_system_user.insert(admin)
def create_user():
"""创建用户"""
correct = False
db = database.database.getDB()
while 1:
username = raw_input("username: "******"uid": username})
if exist_user:
print "the user %s is existed" % (username)
else:
break
user = []
email = raw_input("email: ")
phone = raw_input("phone: ")
while not correct:
pwd = raw_input("password:"******"confirm password:"******"a user registered,name is %s") % (username))
user = {
"uid": username,
"name": username,
"pwd": pwd,
"email": email,
"phone": phone,
'regtime': time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time())),
}
else:
print "password not correct,please input again"
password_hash = pbkdf2_sha512.encrypt(user['pwd'])
user['pwd'] = password_hash
record = db.user.find_one({"uid": user['uid']})
if record is not None:
logging.warn(("that user name %s had been registered") % (username))
return False
db.user.insert(user)
def encrypt_password(password: str) -> str:
return pbkdf2_sha512.encrypt(password, rounds=10000, salt_size=10)
def get_encrypted_password(password):
return pbkdf2_sha512.encrypt(password, rounds=200000, salt_size=16)
def hash_password(password: str) -> str:
return pbkdf2_sha512.encrypt(password)
def encrypt(password):
'''encrypt a plain password'''
enc = pbkdf2_sha512.encrypt(password, rounds=12000)
enc = enc.replace(PBKDF2SHA512_HEADER, CUSTOM_HEADER, 1)
return enc
def set_password(self, new_password):
self.password_hash = pbkdf2_sha512.encrypt(new_password)
