delete user ok

src/admin.py

@@ -36,16 +36,17 @@ def adduser():
         userlimi = request.form['userlimit']
         error = None
         if not userlimi.isdecimal():
-            error = '用户空间必须是整数KB'
+            error = '用户空间必须是整数GB'
         elif not validateEmail(usermail):
             error = 'email格式不合法'
-
+        userlimi = int(userlimi)
+        userlimi *= 1024 * 1024 # userlimit is stored and measured by KB
         if error is None:
             db = get_db()
             try:
                 db.execute(
                     'insert into user(`user_name`, `user_mail`, `user_passwd`, `user_limit`) values (?,?,?,?)',
-                    (username, generate_password_hash(password), usermail, userlimi,)
+                    (username, usermail, generate_password_hash(password), userlimi,)
                 )
                 db.commit()
             except sqlite3.IntegrityError as _e:
@@ -57,4 +58,32 @@ def adduser():
             if error is None:
                 return redirect(url_for('admin.index'))
         flash(error)
-    return render_template("admin/adduser.html")
+    return render_template("admin/adduser.html")
+
+@bp.route('/removeuser', methods=("GET",))
+@admin_login_required
+def removeuser():
+    error = "请求错误"
+    if request.method == 'GET':
+        error = None
+        uid_to_del = request.args.get("uid")
+        if uid_to_del is not None and uid_to_del.isdecimal:
+            db = get_db()
+            rowcnt = 0
+            try:
+                rowcnt = db.execute(
+                    "delete from user where user.user_id=?", (uid_to_del, )
+                ).rowcount
+                db.commit()
+            except sqlite3.IntegrityError as _e:
+                error = "用户未做好被删除的准备：%s" % (_e)
+                db.rollback()
+            except sqlite3.Error as _e:
+                error = "删除发生未知错误: %s" %(_e)
+                db.rollback()
+            if error is None:
+                if rowcnt != 0:
+                    error = "删除uid为%s的用户成功！" % uid_to_del
+                else:
+                    error = "由于uid不存在，可能没有被删除，请再次检查删除结果"
+    return render_template("admin/removeuser.html", remove_result=error)

src/templates/admin/adduser.html

@@ -13,7 +13,7 @@
       <input name="usermail" id="usermail" required><br>
       <label for="password">密码</label>
       <input type="password" name="password" id="password" required><br>
-      <label for="userlimit">用户空间大小</label>
+      <label for="userlimit">用户空间大小(GB)</label>
       <input name="userlimit" id="userlimit" required>
       <input type="submit" value="新建用户">
     </form>

src/templates/admin/index.html

@@ -10,7 +10,9 @@
     <ul>
       {% for user in users%}
         <li>
-          UID:{{ user['user_id'] }}, Name: {{ user['user_name'] }}
+          UID:{{ user['user_id'] }}, 用户名: {{ user['user_name'] }}, 
+          邮箱: {{ user['user_mail'] }}, 用户空间大小: {{ user['user_limit'] / 1024 / 1024 }}GB
+          <a href={{"/admin/removeuser?uid=" ~ user['user_id']}}>删除</a>
         </li>
       {% endfor %}
     </ul>

src/templates/admin/removeuser.html

@@ -0,0 +1,12 @@
+{% extends 'base.html' %}
+
+{% block header %}
+  <h1>{% block title %}管理主页{% endblock %}</h1>
+{% endblock %}
+{% block content %}
+  <h2>删除用户结果</h2>
+  <div>
+    {{remove_result}}
+  </div>
+  <a href="/admin/">返回</a>
+{% endblock %}