def comment(p_comment, token, post_id):
# 两个表，两个游标
conn_c, cursor_c = mysql_pool.create_conn()
conn_p, cursor_p = mysql_pool.create_conn()
email = jwt.decode(token, config.key, algorithms=['HS256'])['user_email']
# print(email, str(post_id), "看这里啊")
email_crypto = rsa_key.encode_email(email, str(post_id))
# print(email_crypto)
# 查找对应总帖
sql = "SELECT * FROM POST WHERE ID={}".format(post_id)
# 如果查找父级为空
if not cursor_p.execute(sql):
conn_c.rollback()
mysql_pool.close_conn(conn_c, cursor_c)
mysql_pool.close_conn(conn_p, cursor_p)
return 2
# 在回帖表中插入回帖
sql = "INSERT INTO P_COMMENT VALUES(NULL, %s, (_binary %s), NOW(), %s)"
cursor_c.execute(sql, (p_comment, email_crypto, post_id))
sql = "UPDATE POST SET LAST_UPD=NOW() WHERE ID={}".format(post_id)
cursor_p.execute(sql)
conn_c.commit()
conn_p.commit()
mysql_pool.close_conn(conn_c, cursor_c)
mysql_pool.close_conn(conn_p, cursor_p)
return 1
def reply(c_comment, token, comment_id):
conn_po, cursor_po = mysql_pool.create_conn()
conn_c, cursor_c = mysql_pool.create_conn()
conn_p, cursor_p = mysql_pool.create_conn()
email = jwt.decode(token, config.key, algorithms=['HS256'])['user_email']
# 查找楼中楼回复的楼所对应的总帖 post_id
sql = "SELECT POST_ID FROM P_COMMENT WHERE ID={}".format(comment_id)
# 如果查找👨父级为空(不用回滚?)
if not cursor_p.execute(sql):
mysql_pool.close_conn(conn_po, cursor_po)
mysql_pool.close_conn(conn_c, cursor_c)
mysql_pool.close_conn(conn_p, cursor_p)
return 3
# 如果👴（总帖）or_p.fetchone()[0]
post_id = cursor_p.fetchone()[0]
if not post_id:
mysql_pool.close_conn(conn_po, cursor_po)
mysql_pool.close_conn(conn_c, cursor_c)
mysql_pool.close_conn(conn_p, cursor_p)
return 2
# 在楼中楼表中插入楼中楼
email_crypto = rsa_key.encode_email(email, str(post_id))
sql = "INSERT INTO C_COMMENT VALUES(NULL, %s, (_binary %s), NOW(), %s, %s)"
cursor_c.execute(sql, (c_comment, email_crypto, comment_id, post_id))
# 更新POST中对应总帖最后一次更新的时间
sql = "UPDATE POST SET LAST_UPD=NOW() WHERE ID={}".format(post_id)
cursor_po.execute(sql)
# 更新总帖的最后更新时间
conn_po.commit()
conn_c.commit()
conn_p.commit()
mysql_pool.close_conn(conn_po, cursor_po)
mysql_pool.close_conn(conn_c, cursor_c)
mysql_pool.close_conn(conn_p, cursor_p)
return 1
def delete(id, table_name, column_name):
conn, cursor = mysql_pool.create_conn()
sql = "DELETE FROM {0} WHERE {1} = {2}".format(table_name, column_name, id)
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
return True
def comment_elective(if_sign, test_form, if_hw, if_touch_fish, lecture,
comment, general):
conn, cursor = mysql_pool.create_conn()
sql = "INSERT INTO ELECTIVE VALUES(NULL, {}, {}, {}, {}, {}, '{}', NOW(), '{}')"\
.format(if_sign, test_form, if_hw, if_touch_fish, general, comment, lecture)
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
return True
def recreate_blacklist():
conn, cursor = mysql_pool.create_conn()
cursor.execute("DROP TABLE IF EXISTS BLACKLIST")
sql = """CREATE TABLE IF NOT EXISTS `BLACKLIST`(
`ID` INT UNSIGNED NOT NULL AUTO_INCREMENT,
`EMAIL_HASH` BIGINT(20),
PRIMARY KEY ( `ID`)
)ENGINE=InnoDB DEFAULT CHARSET=utf8;"""
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
def if_ban(token):
email = jwt.decode(token, config.key, algorithms=['HS256'])['user_email']
conn, cursor = mysql_pool.create_conn()
id_hash = hash(email[1:10])
sql = "SELECT * FROM BLACKLIST WHERE EMAIL_HASH = {}".format(id_hash)
if cursor.execute(sql):
conn.commit()
mysql_pool.close_conn(conn, cursor)
return True
else:
conn.commit()
mysql_pool.close_conn(conn, cursor)
return False
def post(c_post, token):
conn, cursor = mysql_pool.create_conn()
email = jwt.decode(token, config.key, algorithms=['HS256'])['user_email']
sql = "INSERT INTO POST VALUES(NULL, %s, NULL, NOW(), NOW())"
cursor.execute(sql, c_post)
id = cursor.lastrowid
email_crypto = rsa_key.encode_email(email, str(id))
# print(email, str(id), "看我这")
# _binary 表示这句为二进制
sql = "UPDATE POST SET EMAIL_CRYPTO=(_binary %s) WHERE ID=%s"
cursor.execute(sql, (email_crypto, id))
conn.commit()
mysql_pool.close_conn(conn, cursor)
return True
def recreate_post():
conn, cursor = mysql_pool.create_conn()
cursor.execute("DROP TABLE IF EXISTS POST")
sql = """CREATE TABLE IF NOT EXISTS `POST`(
`ID` INT UNSIGNED NOT NULL AUTO_INCREMENT,
`POST_CONTENT` VARCHAR(200) NOT NULL,
`EMAIL_CRYPTO` BLOB,
`TIME` TIMESTAMP,
`LAST_UPD` TIMESTAMP,
PRIMARY KEY ( `ID` )
)ENGINE=InnoDB DEFAULT CHARSET=utf8;"""
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
def ban(email):
conn, cursor = mysql_pool.create_conn()
id_hash = hash(email[1:10])
sql = "SELECT * FROM BLACKLIST WHERE EMAIL_HASH = {}".format(id_hash)
print(id_hash)
if cursor.execute(sql):
conn.commit()
mysql_pool.close_conn(conn, cursor)
return False
else:
sql = "INSERT INTO BLACKLIST VALUES(NULL, {})".format(id_hash)
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
return True
def get_post(post_id):
# 三个表，三个游标
conn_po, cursor_po = mysql_pool.create_conn()
conn_c, cursor_c = mysql_pool.create_conn()
conn_p, cursor_p = mysql_pool.create_conn()
post = []
# 查找对应post_id的post
sql = "SELECT * FROM POST WHERE ID = {}".format(post_id)
cursor_po.execute(sql)
post_content = cursor_po.fetchall()
# print("post_content is", post_content)
for c in post_content:
post.append(list(c))
# 查找对应post_id的comment
sql = "SELECT * FROM P_COMMENT WHERE POST_ID = {}".format(post_id)
cursor_p.execute(sql)
comment_content = cursor_p.fetchall()
# print("comment_content is", comment_content)
for c in comment_content:
post.append(list(c))
# 查找对应post_id的reply
sql = "SELECT * FROM C_COMMENT WHERE POST_ID = {}".format(post_id)
cursor_c.execute(sql)
reply_content = cursor_c.fetchall()
# print("reply_content is:", reply_content)
for c in reply_content:
post.append(list(c))
conn_po.commit()
conn_c.commit()
conn_p.commit()
blobs = [p[2] for p in post]
blob_orders = id_order(blobs)
for p in post:
p[2] = blob_orders[p[2]]
mysql_pool.close_conn(conn_po, cursor_po)
return post
def unban(id_num):
conn, cursor = mysql_pool.create_conn()
id_hash = hash(id_num)
print(id_hash)
sql = "SELECT * FROM BLACKLIST WHERE EMAIL_HASH = {}".format(id_hash)
if cursor.execute(sql):
sql = "DELETE FROM BLACKLIST WHERE EMAIL_HASH = {}".format(id_hash)
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
return True
else:
conn.commit()
mysql_pool.close_conn(conn, cursor)
return False
def recreate_elective():
conn, cursor = mysql_pool.create_conn()
cursor.execute("DROP TABLE IF EXISTS ELECTIVE")
sql = """CREATE TABLE IF NOT EXISTS `ELECTIVE`(
`ID` INT UNSIGNED NOT NULL AUTO_INCREMENT,
`SIGN` INT UNSIGNED NOT NULL,
`FORM` INT UNSIGNED NOT NULL,
`HW` INT UNSIGNED NOT NULL,
`FISH` INT UNSIGNED NOT NULL,
`GENERAL` INT UNSIGNED NOT NULL,
`COMMENT` VARCHAR(200) NOT NULL,
`TIME` TIMESTAMP,
`LECTURE` VARCHAR(40) NOT NULL,
PRIMARY KEY ( `ID` )
)ENGINE=InnoDB DEFAULT CHARSET=utf8;"""
cursor.execute(sql)
conn.commit()
mysql_pool.close_conn(conn, cursor)
def get_recent20_post():
conn_p, cursor_p = mysql_pool.create_conn()
sql = "SELECT ID, POST_CONTENT, LAST_UPD FROM POST ORDER BY LAST_UPD DESC"
cursor_p.execute(sql)
posts = cursor_p.fetchmany(20)
# for post in posts:
# print(post)
post_list = []
for post in posts:
post_dict = {}
post_dict.update({
"id": post[0],
"content": post[1],
"last_upd": post[2]
})
post_list.append(post_dict)
# print("len", len(post_list))
conn_p.commit()
mysql_pool.close_conn(conn_p, cursor_p)
return post_list