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



def change():
    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/ccc.json", 'w', encoding='utf-8')
    with codecs.open("test_data/bbb.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")