From 063263d924fb03b8cb95652f5d09e47398acf331 Mon Sep 17 00:00:00 2001
From: Mr Chen <midoks@163.com>
Date: Sun, 3 Nov 2024 18:45:36 +0800
Subject: [PATCH] update

---
 panel_tools.py         |  6 ++--
 web/thisdb/__init__.py |  3 +-
 web/thisdb/user.py     | 62 ++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 67 insertions(+), 4 deletions(-)
 create mode 100644 web/thisdb/user.py

diff --git a/panel_tools.py b/panel_tools.py
index 89208170a..c8491bbcc 100755
--- a/panel_tools.py
+++ b/panel_tools.py
@@ -256,14 +256,14 @@ def set_panel_username(username=None):
             print("|-错误，不能使用过于简单的用户名")
             return
 
-        model.setUserByRoot(name=username)
+        thisdb.setUserByRoot(name=username)
         print("|-新用户名: %s" % username)
         return
 
-    info = model.getUserByRoot()
+    info = thisdb.getUserByRoot()
     if info['name'] == 'admin':
         username = mw.getRandomString(8).lower()
-        model.setUserByRoot(name=username)
+        thisdb.setUserByRoot(name=username)
     print('|-用户名: ' + info['name'])
 
 
diff --git a/web/thisdb/__init__.py b/web/thisdb/__init__.py
index ebaf86032..8892e9129 100644
--- a/web/thisdb/__init__.py
+++ b/web/thisdb/__init__.py
@@ -8,4 +8,5 @@
 # Author: midoks <midoks@163.com>
 # ---------------------------------------------------------------------------------
 
-from .option import *
\ No newline at end of file
+from .option import *
+from .user import *
\ No newline at end of file
diff --git a/web/thisdb/user.py b/web/thisdb/user.py
new file mode 100644
index 000000000..0270a9021
--- /dev/null
+++ b/web/thisdb/user.py
@@ -0,0 +1,62 @@
+# coding:utf-8
+
+# ---------------------------------------------------------------------------------
+# MW-Linux面板
+# ---------------------------------------------------------------------------------
+# copyright (c) 2018-∞(https://github.com/midoks/mdserver-web) All rights reserved.
+# ---------------------------------------------------------------------------------
+# Author: midoks <midoks@163.com>
+# ---------------------------------------------------------------------------------
+
+import core.mw as mw
+
+
+def getUserById(id,
+) -> None:
+    '''
+    获取用户信息通过用户名
+    '''
+    users_field = 'id,name,password,login_ip,login_time,phone,email,add_time,update_time'
+    item =  mw.M('users').field(users_field).where('id=?', (1,)).select()
+    if len(item) == 0:
+        return None
+    row = {}
+    row['id'] = item[0]['id']
+    row['name'] = item[0]['name']
+    row['password'] = item[0]['password']
+    row['login_ip'] = item[0]['login_ip']
+    row['login_time'] = item[0]['login_time']
+    row['phone'] = item[0]['phone']
+    row['email'] = item[0]['email']
+    row['add_time'] = item[0]['add_time']
+    row['update_time'] = item[0]['update_time']
+    return row
+
+
+def getUserByRoot() -> None:
+    '''
+    获取用户信息通过用户名
+    '''
+    return getUserById(1)
+
+def setUserByRoot(
+    name: str | None = None,
+    password: str | None = None,
+) -> bool:
+    '''
+    设置配置的值
+    :name -> str 名称 (必填)
+    :value -> object值 (必填)
+    :type -> str 类型 (可选|默认common)
+    '''
+    data = {}
+    if name is not None:
+        mw.M('users').where('id=?', (1,)).setField('username', name)
+
+    if password is not None:
+        pwd = mw.md5(password)
+        mw.M('users').where('id=?', (1,)).setField('password', pwd)
+
+    if not data:
+        return False
+    return True
\ No newline at end of file