bib2GBT.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import json\n",
    "import re"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "bibFile = \"test/7-1.bib\"\n",
    "\n",
    "with open(bibFile, \"r\", encoding='utf-8') as f:\n",
    "    lines = f.readlines() \n",
    "\n",
    "lines_multibib = [[]]\n",
    "lines_singlebib = []\n",
    "bib_count = 0\n",
    "\n",
    "for l in lines:\n",
    "    if l == \"\\n\": continue\n",
    "    if l.find('@') != -1:\n",
    "        if bib_count % 2 == 1:\n",
    "            lines_multibib.insert(bib_count - 1, lines_singlebib[:])\n",
    "            lines_singlebib.clear()\n",
    "        bib_count += 1\n",
    "    \n",
    "    lines_singlebib.append(l)\n",
    "\n",
    "lines_multibib.insert(bib_count - 1, lines_singlebib[:])\n",
    "lines_multibib.pop(bib_count)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def getBibJson(lines: list) -> dict:\n",
    "\n",
    "    jsonLines =[]\n",
    "    jsonLines.append('{\\r\\n')\n",
    "\n",
    "\n",
    "    for l in lines[1: -1]:\n",
    "        if (l.find('\"') != -1) & (l.find('\"') != l.rfind('\"')):\n",
    "            index_1 = l.find(\"=\")\n",
    "            index_2 = l.find('\"')\n",
    "            index_3 = l.rfind('\"')\n",
    "            l = '\"' + l[0: index_1].replace(' ', '') + '\":' + l[index_2: index_3] + '\",\\r\\n'   \n",
    "\n",
    "        else:\n",
    "            #cnki\n",
    "            if ((l.find('=') != -1) &\n",
    "                (l.find('\"') == -1) &\n",
    "                (l.find('{') == -1) &\n",
    "                (l.find('}') == -1)):\n",
    "\n",
    "                index_1 = l.find(\"=\")\n",
    "                index_2 = l.find(',')\n",
    "                l = '\"' + l[0: index_1].replace(' ', '') + '\":\"' + l[index_1 + 1: index_2].strip() + '\",\\r\\n'\n",
    "            \n",
    "            else:\n",
    "                l = l.replace('{', '\"', 1)[::-1].replace('}', '\"', 1)[::-1]  # replace the outermost layer only\n",
    "                index = l.find(\"=\")\n",
    "                if(index != -1): l = '\"' + l[0: index].replace(' ', '') + '\":' + l[index + 1: -1] + \"\\r\\n\"\n",
    "\n",
    "        jsonLines.append(l)\n",
    "\n",
    "    jsonLines.append('}')\n",
    "\n",
    "    index = jsonLines[-2].rfind('\"')\n",
    "    jsonLines[-2] = jsonLines[-2][0: index] + jsonLines[-2][index: -1].replace(',', '') + \"\\r\\n\"\n",
    "    \n",
    "    bibJson = json.loads(\"\".join(jsonLines).replace('\\r', '').replace('\\n', '').replace('\\t', '').replace('\\\\', ''))\n",
    "\n",
    "    return bibJson"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def getAuthor(bibJson):\n",
    "    if bibJson[\"author\"].find('&') != -1:  # cnki\n",
    "        _author = bibJson[\"author\"].replace(' and ',\",\").replace(' & ',\",\").replace('.',\"\").upper()\n",
    "    else:\n",
    "        _author = bibJson[\"author\"].replace(',',\"\").replace(' and ',\",\").replace('.',\"\").upper()\n",
    "\n",
    "    _author = _author.replace(', ',\",\").replace(',',\", \")\n",
    "\n",
    "    _author = re.sub(r\"\\{[^{}]*\\}\", \"\", _author)\n",
    "    \n",
    "    if(_author.count(\",\") >= 3):\n",
    "        index_1 = _author.find(',')\n",
    "        index_2 = _author[index_1 + 1: -1].find(',')\n",
    "        index_3 = _author[index_1 + 1: -1][index_2 + 1: -1].find(',')\n",
    "\n",
    "        etc = \", et al\"\n",
    "\n",
    "        if('\\u4e00' <= _author[0: index_1] <= '\\u9fff'):\n",
    "            etc = \",等\"\n",
    "        \n",
    "        _author = _author[0: index_1 + index_2 + index_3 + 2] + etc\n",
    "\n",
    "    return _author"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "def type_D(bibJson, reftype, bibtype):\n",
    "    fileBGT = getAuthor(bibJson) + \". \" + bibJson[\"title\"] + reftype[bibtype] + \". \" + bibJson[\"school\"] + \",\" + bibJson[\"year\"] + \".\"\n",
    "    return fileBGT"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "def type_J(bibJson, reftype, bibtype):\n",
    "    \n",
    "    if \"year\" in bibJson: _year = \",\" + bibJson[\"year\"] + ','\n",
    "    else: _year = \"\"\n",
    "\n",
    "    if \"volume\" in bibJson: _volume = bibJson[\"volume\"]\n",
    "    else: _volume = \"\"\n",
    "\n",
    "    if \"number\" in bibJson: _number = '(' + bibJson[\"number\"] + \")\"\n",
    "    else: _number = \"\"\n",
    "\n",
    "    if \"pages\" in bibJson: _pages = ':' + bibJson[\"pages\"]\n",
    "    else: _pages = \"\"\n",
    "\n",
    "    # cnki\n",
    "    if (\"number\" in bibJson) & (\"pages\" not in bibJson):\n",
    "        _number = \"\"\n",
    "        _pages = ':' + bibJson[\"number\"]\n",
    "\n",
    "    fileBGT = (getAuthor(bibJson) + \". \" + \n",
    "               bibJson[\"title\"] + \n",
    "               reftype[bibtype] + \". \" + \n",
    "               bibJson[\"journal\"] +\n",
    "               _year + _volume + _number + _pages + \".\"\n",
    "               )\n",
    "\n",
    "    return fileBGT"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "def type_C(bibJson, reftype, bibtype):\n",
    "    \n",
    "    if \"year\" in bibJson: _year = \",\" + bibJson[\"year\"] + ','\n",
    "    else: _year = \"\"\n",
    "\n",
    "    if \"pages\" in bibJson: _pages = ':' + bibJson[\"pages\"]\n",
    "    else: _pages = \"\"\n",
    "\n",
    "    fileBGT = (getAuthor(bibJson) + \". \" + \n",
    "               bibJson[\"title\"] + \n",
    "               reftype[bibtype] + \". //\" + \n",
    "               bibJson[\"booktitle\"] +\n",
    "               _year + _pages + \".\"\n",
    "               )\n",
    "\n",
    "    return fileBGT"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "def type_N(bibJson, reftype, bibtype):\n",
    "    \n",
    "    fileBGT = (getAuthor(bibJson) + \". \" + \n",
    "               bibJson[\"title\"] + \n",
    "               reftype[bibtype] + \". \" + \n",
    "               bibJson[\"institution\"] + \",\" + \n",
    "               bibJson[\"year\"] + \".\"\n",
    "               )\n",
    "\n",
    "    return fileBGT"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "reftype = {\"article\": \"[J]\",\n",
    "            \"mastersthesis\": \"[D]\",\n",
    "            \"phdthesis\": \"[D]\",\n",
    "            \"inproceedings\": \"[C]\",\n",
    "            \"conference\": \"[C]\",\n",
    "            \"book\": \"[M]\",\n",
    "            \"booklet\": \"[M]\",\n",
    "            \"techreport\": \"[N]\",\n",
    "            \"misc\": \"[P]\",\n",
    "            \"manual\": \"[P]\"}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "fileBGT = []\n",
    "\n",
    "for _lines in lines_multibib:\n",
    "\n",
    "    index_1 = _lines[0].find('@')\n",
    "    index_2 = _lines[0].find('{')\n",
    "\n",
    "    if index_2 == -1: index_2 = _lines[0].find('(')\n",
    "\n",
    "    bibtype = _lines[0][index_1 + 1: index_2].replace(' ', '').lower()\n",
    "\n",
    "    # print(bibtype)\n",
    "\n",
    "    index_3 = _lines[-1].rfind(')')\n",
    "\n",
    "    if index_3 != -1:\n",
    "        _lines[-1] = _lines[-1][0: index_3]\n",
    "        _lines.append('}')\n",
    "    \n",
    "    bibJson = getBibJson(_lines)\n",
    "\n",
    "    match reftype[bibtype]:\n",
    "        case \"[J]\":\n",
    "            fileBGT.append(type_J(bibJson, reftype, bibtype))\n",
    "        case \"[D]\":\n",
    "            fileBGT.append(type_D(bibJson, reftype, bibtype))\n",
    "        case \"[C]\":\n",
    "            fileBGT.append(type_C(bibJson, reftype, bibtype))\n",
    "        case \"[N]\":\n",
    "            fileBGT.append(type_N(bibJson, reftype, bibtype))\n",
    "        case _:\n",
    "            fileBGT.append(\"** 类型不支持 **\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[1] 干春晖,郑若谷,余典范. 中国产业结构变迁对经济增长和波动的影响[J]. 经济研究,2011,46(5):4-16,31\n",
      "[2] 孙志军,薛磊,许阳明,等. 深度学习研究综述[J]. 计算机应用研究,2012,29:2806-2810\n"
     ]
    }
   ],
   "source": [
    "_count = 1\n",
    "for l in fileBGT:\n",
    "    if fileBGT.__len__() == 1: print(l)\n",
    "    else:\n",
    "        print(f'[{_count}]', l)\n",
    "        _count += 1"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}