import os
import sys

from thrift.server.TNonblockingServer import TNonblockingServer
from thrift.transport import TSocket

sys.path.append('/root/PixivSearch')
from PixivSearch.util import Util
from PixivSearch.dao.Comment import xml, downloadXml
from PixivSearch.thrift.QueryComment import QueryComment
from PixivSearch.thrift.QueryComment.QueryComment import Iface

commentPath = '/root/PixivSearch/PixivSearch/thrift/tmpFile/comment'


class Server(Iface):
    def commentSumList(self, cids):
        result = []
        for cid in cids:
            comment_selector = xml('https://comment.bilibili.com/{cid}.xml'.format(cid=cid))
            length = len(comment_selector.xpath('//i//d/text()'))
            print('cid:{cid},弹幕数：{length}'.format(cid=cid, length=length))
            result.append(length)
        return result

    def commentSum(self, cid):
        return self.commentSumList([cid])[0];

    def download(self, cids, fileName):
        path = '{commentPath}/{fileName}'.format(commentPath=commentPath, fileName=fileName)
        for cid in cids:
            downloadXml(path, cid,
                        histroy=False)
        zipFile = '{path}/{fileName}.zip'.format(path=path,fileName=fileName)
        print(zipFile)
        Util.zip(path, zipFile)
        if os.path.isfile(zipFile):
            print('压缩包成功生成到{zipFile}'.format(zipFile=zipFile))
            return zipFile
        else:
            return None


if __name__ == '__main__':
    socket = TSocket.TServerSocket(port=2233)
    processor = QueryComment.Processor(Server())
    server = TNonblockingServer(processor, socket)
    server.serve()