41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
import hashlib
|
||
|
|
|
||
|
|
from app.model import *
|
||
|
|
# from app.Login import md5
|
||
|
|
from pony import orm
|
||
|
|
|
||
|
|
def get_users():
|
||
|
|
with db_session:
|
||
|
|
return User.select().order_by(User.name)[:]
|
||
|
|
|
||
|
|
def get_user_by_username(username):
|
||
|
|
with db_session:
|
||
|
|
user = User.select(name=username)
|
||
|
|
if user:
|
||
|
|
return user.first()
|
||
|
|
|
||
|
|
def insert_user(username, password, start_date, expiry_date):
|
||
|
|
with db_session:
|
||
|
|
user = User(name=username, password=password, start_date=start_date, expiry_date=expiry_date)
|
||
|
|
orm.commit()
|
||
|
|
|
||
|
|
def update_password_by_username(username, password="123456"):
|
||
|
|
with db_session:
|
||
|
|
user = User.select(name=username)
|
||
|
|
if user:
|
||
|
|
user.first().password = md5(username + password)
|
||
|
|
|
||
|
|
def update_expiry_time_by_username(username, expiry_time="20230323"):
|
||
|
|
with db_session:
|
||
|
|
user = User.select(name=username)
|
||
|
|
if user:
|
||
|
|
user.first().expiry_date = expiry_time
|
||
|
|
|
||
|
|
def md5(s):
|
||
|
|
'''
|
||
|
|
MD5摘要
|
||
|
|
:param str: 字符串
|
||
|
|
:return: 经MD5以后的字符串
|
||
|
|
'''
|
||
|
|
h = hashlib.md5(s.encode(encoding='utf-8'))
|
||
|
|
return h.hexdigest()
|