52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
|
from db.pgsql import PostgreSQL
|
|||
|
|
from db.system_config_dao import SystemConfigDAO
|
|||
|
|
|
|||
|
|
def check_user_login(user_id: str, password: str) -> bool:
|
|||
|
|
db = PostgreSQL()
|
|||
|
|
try:
|
|||
|
|
db.connect()
|
|||
|
|
sql = "SELECT 1 FROM sys_user WHERE user_id=%s AND password=md5(%s)"
|
|||
|
|
result = db.execute_query(sql, (user_id, password))
|
|||
|
|
return bool(len(result) > 0)
|
|||
|
|
finally:
|
|||
|
|
db.disconnect()
|
|||
|
|
|
|||
|
|
def get_user_info(user_id: str):
|
|||
|
|
db = PostgreSQL()
|
|||
|
|
try:
|
|||
|
|
db.connect()
|
|||
|
|
sql = """
|
|||
|
|
SELECT user_id, nick_name AS user_name, position_id,
|
|||
|
|
f_getname('GETCORPEXP', '', '', org_id) AS corp_name,
|
|||
|
|
org_id AS corp_id, default_org_id
|
|||
|
|
FROM sys_user
|
|||
|
|
WHERE user_id = %s
|
|||
|
|
"""
|
|||
|
|
result = db.execute_query(sql, (user_id,))
|
|||
|
|
if result:
|
|||
|
|
user_name = result[0]['user_name']
|
|||
|
|
corp_name = result[0]['corp_name']
|
|||
|
|
corp_id = result[0]['corp_id']
|
|||
|
|
position_id = result[0]['position_id']
|
|||
|
|
default_org_id = result[0]['default_org_id']
|
|||
|
|
|
|||
|
|
# 如果有默认账套ID，使用默认账套ID替代登录账套ID
|
|||
|
|
if default_org_id:
|
|||
|
|
corp_id = default_org_id
|
|||
|
|
# 查询默认账套的名称
|
|||
|
|
corp_name_sql = """
|
|||
|
|
SELECT corp_exp
|
|||
|
|
FROM sys_org
|
|||
|
|
WHERE org_id = %s
|
|||
|
|
"""
|
|||
|
|
corp_result = db.execute_query(corp_name_sql, (corp_id,))
|
|||
|
|
if corp_result:
|
|||
|
|
corp_name = corp_result[0]['corp_exp']
|
|||
|
|
|
|||
|
|
# 将用户的当前账套ID保存到内存中
|
|||
|
|
SystemConfigDAO.save_user_selected_corp(user_id, corp_id)
|
|||
|
|
|
|||
|
|
return user_name, corp_name, corp_id, position_id
|
|||
|
|
return None, None, None, None
|
|||
|
|
finally:
|
|||
|
|
db.disconnect()
|