56 lines
1.1 KiB
Python
56 lines
1.1 KiB
Python
from ItemRelate import main as relate
|
|
|
|
import json
|
|
import os
|
|
import random
|
|
|
|
def batch():
|
|
scales = os.listdir("Scales")
|
|
items={}
|
|
for i in scales:
|
|
with open("Scales/"+i,"r") as scale:
|
|
tmp = json.load(scale)
|
|
for i in tmp["item"]:
|
|
items[i]=tmp["item"][i]
|
|
# print(items)
|
|
return items
|
|
|
|
def old_type(str):
|
|
with open(str,"r") as file:
|
|
scale=json.load(file)
|
|
new={"item":{}}
|
|
for i in scale:
|
|
new["item"][i["name"]]=i["label"]
|
|
# print(i["name"],i["label"])
|
|
with open(str,"w") as file:
|
|
file.write(json.dumps(new))
|
|
|
|
def data():
|
|
s=""
|
|
|
|
item = batch()
|
|
for i in item:
|
|
s+=i+','
|
|
s=s[:-1]+'\n'
|
|
|
|
for i in range(0,1000):
|
|
s += str(random.randint(0,4))
|
|
for j in range(1,20):
|
|
s += ',' + str(random.randint(0,4))
|
|
s+='\n'
|
|
|
|
with open("Temp/data.csv","w") as data:
|
|
data.write(s)
|
|
|
|
|
|
def main(force:bool = False):
|
|
if force or os.path.getsize("Temp/items.json") == 0:
|
|
que=relate(batch())
|
|
with open("Temp/items.json","w") as items:
|
|
items.write(json.dumps(que))
|
|
else:
|
|
with open("Temp/items.json","r") as items:
|
|
que = json.load(items)
|
|
data()
|
|
return que
|