stiliyagi-bbs/backlink.py

import os
import re
from collections import defaultdict
import webtools as wt

threadpath = "./threads/"

def load_thread(th):
    th = int(th)
    thp = str(threadpath + str(th) + '.txt')
    if not os.path.exists(thp):
        return None
    with open(thp, 'r') as mt:
        mt = mt.read().splitlines()
    return mt
        
def do_backlink(th='0'):
    mt = load_thread(th)
    bld = defaultdict(list)
    for n, t in enumerate(mt):
        repl = re.findall(r'\>\>[1-9][00-99]*', t)
        repl = [r[8:] for r in repl if r]
        for r in repl:
            if int(r) >= len(mt) or str(n) in bld[r]:
                continue
            bld[r].append(str(n))

    for r in sorted([int(k) for k in bld.keys()]):
        r = str(r)
    return bld