Skip to content
Snippets Groups Projects
Commit e6537d38 authored by Bo-Chun Chen's avatar Bo-Chun Chen
Browse files

Add database settings

parent 38251670
No related branches found
No related tags found
5 merge requests!147Merge previous default branch feat-cod-rmq into main,!85kill nginx process running under user from login node,!78Feat cod rmq,!62Feat notify user errmsg,!63Feat db handle
......@@ -3,6 +3,7 @@ import sys
import copy
import json
import signal
import dataset
import rc_util
import smtplib
from rc_rmq import RCRMQ
......@@ -15,6 +16,9 @@ timeout = 30
args = rc_util.get_args()
logger = rc_util.get_logger(args)
db = dataset.connect(f'sqlite:///.agent_db/user_reg.db')
table = db['users']
record = {
'uid': -1,
'gid': -1,
......@@ -77,6 +81,29 @@ def notify_admin(username, user_record):
logger.debug(f'User report sent to: {mail_cfg.Admin_email}')
def insert_db(username, msg):
# Search username in db
record = table.find_one(username=username)
if not record:
# SQL insert
table.insert({
'username': username,
'uid': msg.get('uid', -1),
'gid': msg.get('gid', -1),
'email': msg.get('email', ''),
'reason': msg.get('reason', ''),
'fullname': msg.get('fullname', ''),
'sent': None,
'last_update': datetime.now()
})
def update_db(username, data):
obj = { 'username': username, **data }
table.update(obj, ['username'])
def task_manager(ch, method, properties, body):
msg = json.loads(body)
username = method.routing_key.split('.')[1]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment