import rispy
import pandas as pd
import io

class RisFileException(Exception):
    pass


class RisFile:
    '''
    ris文件解析器, 使用rispy模块
    '''
    def __init__(self, file):
        self.file = file
        self.fHandle = None


    def _fetch_info(self, kwd: list):
        collected = []
        for entry in rispy.load(self.fHanlde):
            rec = {}
            for key in kwd:
                if not key in self.keywords:
                    raise RisFileException(f'Not valid info that can be parsed from ris file, all keywords: {self.keywords}')
                rec[key] = entry[key]
            collected.append(rec)
        return pd.DataFrame(collected)


    def parse_info(self, kwd: list):
        '''
        解析给定区域的数值, 如果字段不存在则抛出错误
        '''
        if isinstance(self.file, str):
            with open(self.file, 'r') as self.fHanlde:
                return self._fetch_info(kwd)
        elif isinstance(self.file, io.StringIO):
            self.fHanlde = self.file
            return self._fetch_info(kwd)


    @property
    def keywords(self):
        '''
        调用rispy给出可解析的所有字段
        '''
        return set(rispy.TAG_KEY_MAPPING.values())


if __name__ == "__main__":
    risFile = RisFile(file='/home/silen/git_proj/ReviewGPT/test/G1/Paper035')
    print(risFile.keywords)
    print(risFile.parse_info(kwd=['doi', 'title', 'abstract']))