happyz
/
stable-diffusion-for-fun
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Manage] adds capability to delete jobs based on uuid

This commit is contained in:
HappyZ 2023-05-08 18:23:40 -07:00
parent a609258a0c
commit 662883d3f1
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
manage_db.py
View File

@ -128,14 +128,19 @@ def delete_user(c, username):
c.execute("DELETE FROM users WHERE username=?", (username,)) c.execute("DELETE FROM users WHERE username=?", (username,))
def delete_job(c, uuid): def delete_jobs(c, job_uuid="", username=""):
"""Delete the job with the given uuid, or ignore the operation if the uuid does not exist""" """Delete the job with the given uuid, or ignore the operation if the uuid does not exist"""
if username:
c.execute(
"DELETE FROM history WHERE apikey=(SELECT apikey FROM users WHERE username=?)",
(username,),
)
elif job_uuid:
c.execute( c.execute(
"DELETE FROM history WHERE uuid=?", "DELETE FROM history WHERE uuid=?",
(uuid,), (job_uuid,),
) )
def show_users(c, username="", details=False): def show_users(c, username="", details=False):
"""Print all users in the users table if username is not specified, """Print all users in the users table if username is not specified,
or only the user with the given username otherwise""" or only the user with the given username otherwise"""
@ -219,7 +224,9 @@ def manage(args):
if args.delete_type == "user": if args.delete_type == "user":
delete_user(c, args.username) delete_user(c, args.username)
elif args.delete_type == "job": elif args.delete_type == "job":
delete_job(c, args.job_id) delete_jobs(c, job_uuid=args.job_id)
elif args.delete_type == "jobs":
delete_jobs(c, username=args.username)
elif args.action == "list": elif args.action == "list":
show_users(c, args.username, args.details) show_users(c, args.username, args.details)
@ -284,6 +291,8 @@ def main():
user_parser.add_argument("username") user_parser.add_argument("username")
job_parser = delete_subparsers.add_parser("job") job_parser = delete_subparsers.add_parser("job")
job_parser.add_argument("job_id") job_parser.add_argument("job_id")
jobs_parser = delete_subparsers.add_parser("jobs")
jobs_parser.add_argument("username")
# Sub-parser for the "list" action # Sub-parser for the "list" action
list_parser = subparsers.add_parser("list") list_parser = subparsers.add_parser("list")