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()