def UpdateMyInfo(self, pwss, user_name, email, identifyNum):
error = None
updateQuery = "UPDATE User SET password = %s, name = %s, email = %s\
idNumber = %s WHERE person_id = %s"
if "" in [pwss, user_name, email, identifyNum]:
error = "No data"
else:
self.selectquery = u"SELECT EXISTS (" + UserInfo.userIdCheck + u")"
curs.execute(self.selectquery, (self.user_id, ))
userid_ = curs.fetchone()
if userid_[0] == 0:
error = "No ID"
else:
flash("Info Changed!")
curs.execute(updateQuery, (
pwss,
user_name,
email,
identifyNum,
self.user_id,
))
conn.commit()
return error
def Register(self, user_id, pwss, user_name, email, identifyNum):
error = None
if "" in [user_id, pwss, user_name, email, identifyNum]:
error = "Filed is Empty!"
else:
self.selectquery = u"SELECT EXISTS (" + UserInfo.checkEmail + u")"
print(email)
curs.execute(self.selectquery, (email, ))
email_ = curs.fetchone()
self.selectquery = u"SELECT EXISTS (" + UserInfo.checkName + u")"
curs.execute(self.selectquery, (user_id, ))
name_ = curs.fetchone()
if email_[0] == 0 and name_[0] == 0:
flash('Account Created!')
curs.execute(UserInfo.InsertQuery, (
user_id,
pwss,
user_name,
email,
identifyNum,
))
conn.commit()
else:
error = "Already Exist"
return error
def AdminRemove(self, target_user_id):
error = None
RemoveAdmin = u"UPDATE User SET auth = 1 WHERE person_id like %s"
if target_user_id == "":
error = "Invalid Input!"
else:
self.selectquery = u"SELECT EXISTS (" + self.userIdCheck + u")"
curs.execute(self.selectquery, (target_user_id, ))
userid_ = curs.fetchone()
if userid_[0] == 0:
error = "No data"
else:
flash("Auth Changed!")
curs.execute(RemoveAdmin, (target_user_id, ))
conn.commit()
return error
def BoardRemove(self, board_name):
error = None
RemoveBoard = u"DELETE FROM Board WHERE board_name like %s"
if board_name == "":
error = "Invalid Input!"
else:
self.selectquery = u"SELECT EXISTS (" + self.boardCheck + u")"
curs.execute(self.selectquery, (target_board_name, ))
board_name_ = curs.fetchone()
if board_name_[0] == 0:
error = "No data"
else:
flash("Board Removed!")
curs.execute(RemoveBoard, (board_name))
conn.commit()
return error
def BoardCreate(self, board_name):
error = None
# write_auth 1 means that everyone can write a post in this board
# write_auth 0 menas that only manager can write a post in this board
CreateBoard = u"INSERT INTO Board(board_name, write_auth) \
VALUES (%s, 1)"
if board_name == "":
error = "Invalid Input!"
else:
self.selectquery = u"SELECT EXISTS (" + self.boardCheck + u")"
curs.execute(self.selectquery, (target_board_name, ))
board_name_ = curs.fetchone()
if board_name_[0] == 0:
flash("Board Created!")
curs.execute(CreateBoard, (board_name, ))
conn.commit()
else:
error = "Already Exist"
return error
def AddBlackList(self, target_board_name, target_user_id):
error = None
InsertToBlackList = u"INSERT INTO `BlackList`(BID, user_id) \
VALUES((SELECT BID from Board where board_name like %s), %s)"
if "" in [target_board_name, target_user_id]:
error = "Invalid Input!"
else:
self.selectquery = u"SELECT EXISTS (" + self.userIdCheck + u")"
curs.execute(self.selectquery, (target_user_id, ))
userid_ = curs.fetchone()
self.selectquery = u"SELECT EXISTS (" + self.boardCheck + u")"
curs.execute(self.selectquery, (target_board_name, ))
board_name_ = curs.fetchone()
self.selectquery = u"SELECT EXISTS (" + self.blacklistCheck + u")"
curs.execute(self.selectquery, (
target_user_id,
target_board_name,
))
blacklist_ = curs.fetchone()
if userid_[0] == 0 and board_name_[0] == 0:
error = "Invalid"
elif blacklist_[0] != 0:
error = "Already Exist"
else:
flash("Added to BlackList")
curs.execute(InsertToBlackList, (
target_board_name,
target_user_id,
))
conn.commit()
return error
def ModifyPost(self, title, contents, target_user_id):
error = None
ModifyQuery = "UPDATE `Post` SET `title` = %s,\
contents = %s WHERE person_id = %s"
curs.execute(ModifyQuery, (
title,
contents,
target_user_id,
))
if conn.commit() is False:
error = "error"
else:
error = "update your this post"
return error
def RemoveBlackList(self, target_board_name, target_user_id):
error = None
RemoveFromBlackList = u"DELETE FROM BlackList WHERE user_id like %s \
and BID = (SELECT BID from Board where board_name like %s)"
if "" in [target_board_name, target_user_id]:
error = "Invalid Input!"
else:
self.selectquery = u"SELECT EXISTS (" + self.userIdCheck + u")"
curs.execute(self.selectquery, (target_user_id, ))
userid_ = curs.fetchone()
self.selectquery = u"SELECT EXISTS (" + self.boardCheck + u")"
curs.execute(self.selectquery, (target_board_name, ))
board_name_ = curs.fetchone()
self.selectquery = u"SELECT EXISTS (" + self.blacklistCheck + u")"
curs.execute(self.selectquery, (
target_user_id,
target_board_name,
))
blacklist_ = curs.fetchone()
if userid_[0] == 0 and board_name_[0] == 0:
error = "Invalid"
elif blacklist_[0] == 0:
error = "No data"
else:
flash("Successfully Removed!")
curs.execute(RemoveFromBlackList, (
target_user_id,
target_board_name,
))
conn.commit()
return error
def AddPost(board_name, title, contents, writer):
error = None
InsertPost = "INSERT INTO %s (`title`, `contents`, `writer`)\
VALUES (%s, %s, %s, );"
curs.execute(InsertPost, (
board_name,
title,
contents,
writer,
))
if conn.commit() is False:
error = False
else:
error = "Success Insert!!"
return error