import json
import numpy as np
from random import choice
from tqdm import tqdm
import io
import codecs
import sys
import re


schemas=[]
charid=[]
i=0
with io.open('Relation/origin_data/all_50_schemas',"r", encoding='utf-8') as fr:
        for line in fr:
            dic = json.loads(line.encode('utf-8'))
            schemas.append(dic)
            charid.append(dic['predicate'])

json_file = codecs.open("test_data/test_demo.res", 'w', encoding='utf-8')
with codecs.open("test_data/test_p.json", 'r', 'utf-8') as fr:
    for line in fr:
        try:
            dic = json.loads(line.encode('utf-8'))
        except ValueError:
            continue
        print('text:', dic['text'])

        for schema in schemas:
            for spolist in dic['spo_list']:
                if schema['predicate'] == spolist['predicate']:
                    spolist['object_type']=schema['object_type']
                    spolist['subject_type'] = schema['subject_type']
        json.dump(dic, json_file, ensure_ascii=False)
        json_file.write("\n")