def register(self):
username = self.username_lineEdit.text()
password = self.password_lineEdit.text()
confirm = self.confirm_password_lineEdit.text()
if '' in [username, password, confirm]:
msg_box(self, '提示', '关键信息不能为空!')
return
db = DBHelp()
count, res = db.query_super(table_name='user',
column_name='username',
condition=username)
if count != 0:
msg_box(self, '提示', '用户名已存在!')
return
if password != confirm:
msg_box(self, '错误', '两次输入密码不一致!')
return
user_info = [
get_uuid(), username,
get_md5(password), 1,
get_current_time(), 0,
get_current_time()
]
db.add_user(user_info)
db.db_commit()
db.instance = None
del db
msg_box(self, '提示', '注册成功!')
self.close()
def add(self):
book_name = self.book_name_lineEdit.text()
author = self.author_lineEdit.text()
publish_company = self.publish_company_lineEdit.text()
publish_date = self.publish_date_lineEdit.text()
store_num = self.store_num_lineEdit.text()
if '' in [book_name, author, publish_company, publish_date, store_num]:
msg_box(self, '错误', '请输入图书的关键信息!')
return
db = DBHelp()
count, res = db.query_super(table_name='book',
column_name='book_name',
condition=book_name)
if count:
msg_box(self, '错误', '已存在同名书籍!')
return
book_info = [
get_uuid(), book_name, author, publish_company, store_num, 0,
get_current_time(), publish_date
]
db.add_book(data=book_info)
db.db_commit()
db.instance = None
del db
self.add_book_don_signal.emit()
self.close()
msg_box(self, '提示', '添加新图书成功!')
def get_book_info(self):
db = DBHelp()
count, res = db.query_super(table_name='book',
column_name='booK_name',
condition=self.book_info)
self.current_book_info = list(res[0])
self.init_book_info_done_signal.emit()
db.instance = None
del db
def book_info_th(self):
if self.user_role == '管理员':
db = DBHelp()
count, res = db.query_all(table_name='borrow_info')
self.method_name(db, res)
else:
db = DBHelp()
count, res = db.query_super(table_name='borrow_info',
column_name='borrow_user',
condition=self.username)
self.method_name(db, res)
def login(self, username, password):
db = DBHelp()
count, res = db.query_super(table_name='user',
column_name='username',
condition=username)
if count == 0:
self.login_done_signal.emit(1)
return
if get_md5(password) != res[0][2]:
self.login_done_signal.emit(11)
return
self.role = res[0][3]
self.login_done_signal.emit(111)
def init_data(self):
try:
db = DBHelp()
count, res = db.query_super(table_name='borrow_info',
column_name='id',
condition=self.borrow_id)
self.origin_return_time = res[0][7]
self.origin_borrow_day = res[0][5]
self.book_name_label.setText('书籍《' + res[0][2] + '》续借')
self.return_date_lineEdit.setText(str(res[0][7]))
db.instance = None
del db
except:
self.renew_done_signal.emit(0)
def search_borrow_info(self):
if self.borrow_user_search_lineEdit.text() == '':
msg_box(self, '提示', '请输入需要搜索的内容!')
return
if self.user_role == '管理员':
search_type = self.comboBox.currentText()
search_content = self.borrow_user_search_lineEdit.text()
db = DBHelp()
count, res = db.query_super(
table_name='borrow_info',
column_name=SEARCH_CONTENT_MAP.get(search_type),
condition=search_content)
if count == 0:
msg_box(widget=self, title='提示', msg='未找到相关记录!')
return
self.get_data_from_database(db, res=res)
def login(self, username, password):
"""
登陆子线程用户验证
:param username: 需要验证的用户名
:param password: 匹配的密码
:return: 验证出错返回,并发射相应TAG的信号
"""
db = DBHelp()
count, res = db.query_super(table_name='user',
column_name='username',
condition=username)
if count == 0:
self.login_done_signal.emit(1)
return
if get_md5(password) != res[0][2]:
self.login_done_signal.emit(11)
return
self.role = res[0][3]
self.login_done_signal.emit(111)
def btn_slot(self, tag):
if tag == 'add':
self.add_book_win = AddBookWindow()
self.add_book_win.add_book_don_signal.connect(self.add_book_done)
self.add_book_win.show()
if tag == 'search':
search_type = self.search_comboBox.currentText()
search_content = self.book_search_content_lineEdit.text()
if search_content == '':
msg_box(self, '提示', '请输入搜索内容~')
return
db = DBHelp()
count, res = db.query_super(table_name='book', column_name=SEARCH_CONTENT_MAP.get(search_type),
condition=search_content)
if count == 0:
msg_box(self, '提示', '您所搜索的图书不存在!')
return
self.show_book([count, res])
if tag == 'refresh':
self.get_book_info()