from admin.dboper import *


# 前台, 加载所有论文信息
def loadAllPaperInfo ():
    sql = "select * from paperinfo"
    results = mySQL_Query(sql)
    return results

# 前台, 查看一个论文信息
def loadPaperInfoByID (pid):
    sql = "select *  from PaperInfo where PaperID=%s" % pid
    results = mySQL_Query(sql)
    results = results[0]
    return results

# 前台, 加载所有论文信息
def loadAllPaperInfo_Show ():
    sql = "select PaperID, RTID, PTitle, PAbstract, PImage, PubType, Publisher, Pages, PubYear, Authors from paperinfo where PStatus=1"
    results = mySQL_Query(sql)
    results2 = []
    if len(results) > 0:
        for row in results:
            row2 = list(row)
            if len(row2[3])>220:
                row2[3] = row2[3][0:220]+"..."
            results2.append(row2)
    return results2



# 根据PID查询一个论文信息(前台)
def loadPaperwithRTNamebyID_Show (pid):
    sql = "select PaperID, RTNameCN, PTitle, PAbstract, PImage, PContent, PubType, Publisher, Pages, PubYear, Authors, PubLink, ArxivLink, CodeLink, PubStatus  from PaperInfo p, ResearchTopic rt where p.RTID = rt.RTID and p.PaperID=" + pid
    results = mySQL_Query(sql)
    if len(results) > 0:
        results = list(results[0])
        if results[14] == 1:
            results[14] = "已发表"
        elif results[14] == 2:
            results[14] = "已录用"
        elif results[14] == 3:
            results[14] = "已投稿"
    return results

# 直接加载论文信息
def loadPaperList_Top6 ():
    sql = "Select * From PaperInfo limit 0,6"
    results = mySQL_Query(sql)
    return results


# 查询所有论文信息
def loadallpapers ():
    sql = "select p.PaperID, rt.RTNameCN, p.PTitle, p.Publisher, p.PubYear, p.PubStatus  from paperinfo p, researchtopic rt where p.RTID = rt.RTID;"
    results = mySQL_Query(sql)
    results2 = []
    for row in results:
        row = list(row)
        row[2] = row[2][0:10] + "..."  # 标题
        row[3] = row[3][0:10] + "..."  # 作者

        if (row[5] == 1):
            row[5] = "已发表"
        elif (row[5] == 2):
            row[5] = "已录用"
        else:
            row[5] = "已投稿"
        results2.append(row)
    return results2


# PaperID, RTID, PTitle, PAbstract, PImage, PContent, PubType, Publisher, Pages, PubYear, Authors, PubLink, ArxivLink, CodeLink, PubStatus, PStatus
# 添加论文信息
def add_paperinfo (data):
    sql = "insert into paperinfo values (NULL, %s, '%s', '%s', '%s', '%s', %s, '%s', '%s', '%s', '%s', '%s', '%s', '%s', %s, %s);" % (
    data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8], data[9], data[10], data[11],
    data[12], data[13], data[14])
    i, msg = mySQL_Execute(sql)
    return i, msg


# 根据论文编号查询一条论文信息
def loadPaperbyID (pid):
    sql = "select *  from paperinfo where PaperID =" + pid
    results = mySQL_Query(sql)
    if len(results) > 0:
        results = results[0]
    return results


# 根据PID查询一个论文信息
def loadPaperwithRTNamebyID (pid):
    sql = "select PaperID, RTNameCN, PTitle, PAbstract, PImage, PContent, PubType, Publisher, Pages, PubYear, Authors, PubLink, ArxivLink, CodeLink, PubStatus, PStatus  from PaperInfo p, ResearchTopic rt where p.RTID = rt.RTID and p.PaperID=" + pid
    results = mySQL_Query(sql)
    if len(results) > 0:
        results = list(results[0])
        if results[14] == 1:
            results[14] = "已发表"
        elif results[14] == 2:
            results[14] = "已录用"
        elif results[14] == 3:
            results[14] = "已投稿"

        if results[15] == 1:
            results[15] = "显示"
        elif results[15] == 2:
            results[15] = "不显示"
    return results


# 添加论文信息
def update_paperinfo (data):
    sql = "update paperinfo set RTID=%s, PTitle='%s', PAbstract='%s', PImage='%s', PContent='%s', PubType=%s, Publisher='%s', Pages='%s', PubYear='%s', Authors='%s', PubLink='%s', ArxivLink='%s', CodeLink='%s', PubStatus=%s, PStatus=%s where PaperID=%s" % (
    data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8], data[9], data[10], data[11],
    data[12], data[13], data[14], data[15])
    #   print(sql)
    i, msg = mySQL_Execute(sql)
    return i, msg


# 删除一个论文信息
def del_Paper (pid):
    sql = "delete from PaperInfo where PaperID=%s;" % pid
    # print("sql语句:", sql)
    i, msg = mySQL_Execute(sql)
    return i, msg