From 0af780f1accda67fccef140d4233177de07bc94d Mon Sep 17 00:00:00 2001
From: mxr612 <work@mxr612.top>
Date: Tue, 14 May 2024 17:18:57 +0800
Subject: [PATCH] move file load to new file, add csv data maker.

---
 file_load.py | 55 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 main.py      | 33 +++----------------------------
 2 files changed, 58 insertions(+), 30 deletions(-)
 create mode 100644 file_load.py

diff --git a/file_load.py b/file_load.py
new file mode 100644
index 0000000..3c0814f
--- /dev/null
+++ b/file_load.py
@@ -0,0 +1,55 @@
+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
diff --git a/main.py b/main.py
index 5c6a2f2..34ce745 100644
--- a/main.py
+++ b/main.py
@@ -1,37 +1,10 @@
-from ItemRelate import main as relate
+from file_load import main as init
+
 import json
 import os
 
-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))
-
-# old_type("Scales/ceE.json")
-
-if 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)
+que = init()
 
 for i in que:
 	print(i)