kopia lustrzana https://github.com/torrinworx/Blend_My_NFTs
73 wiersze
2.4 KiB
Python
73 wiersze
2.4 KiB
Python
# Purpose:
|
|
# This file sorts the NFT DNA from NFTRecord.json and exports it to a given number of Batch#.json files set by nftsPerBatch
|
|
# in config.py.
|
|
|
|
import bpy
|
|
import os
|
|
import json
|
|
import random
|
|
|
|
|
|
def makeBatches(nftName, maxNFTs, nftsPerBatch, save_path, batch_json_save_path):
|
|
Blend_My_NFTs_Output = os.path.join(save_path, "Blend_My_NFTs Output", "NFT_Data")
|
|
NFTRecord_save_path = os.path.join(Blend_My_NFTs_Output, "NFTRecord.json")
|
|
|
|
DataDictionary = json.load(open(NFTRecord_save_path))
|
|
|
|
numNFTsGenerated = DataDictionary["numNFTsGenerated"]
|
|
hierarchy = DataDictionary["hierarchy"]
|
|
DNAList = DataDictionary["DNAList"]
|
|
|
|
numBatches = maxNFTs / nftsPerBatch
|
|
|
|
print(f"To generate batches of {nftsPerBatch} DNA sequences per batch, with a total of {numNFTsGenerated}"
|
|
f" possible NFT DNA sequences, the number of batches generated will be {numBatches}")
|
|
|
|
# Clears the Batch Data folder of Batches:
|
|
batchList = os.listdir(batch_json_save_path)
|
|
|
|
if batchList:
|
|
for i in batchList:
|
|
batch = os.path.join(batch_json_save_path, i)
|
|
if os.path.exists(batch):
|
|
os.remove(
|
|
os.path.join(batch_json_save_path, i)
|
|
)
|
|
|
|
i = 0
|
|
while i < numBatches:
|
|
batchDictionary = {}
|
|
BatchDNAList = []
|
|
|
|
j = 0
|
|
while (j < nftsPerBatch) and (DNAList):
|
|
oneDNA = random.choice(DNAList)
|
|
BatchDNAList.append(oneDNA)
|
|
DNAList.remove(oneDNA)
|
|
j += 1
|
|
|
|
batchDictionary["NFTs_in_Batch"] = int(len(BatchDNAList))
|
|
batchDictionary["hierarchy"] = hierarchy
|
|
batchDictionary["BatchDNAList"] = BatchDNAList
|
|
|
|
batchDictionaryObject = json.dumps(batchDictionary, indent=1, ensure_ascii=True)
|
|
|
|
with open(os.path.join(batch_json_save_path, ("Batch{}.json".format(i + 1))), "w") as outfile:
|
|
outfile.write(batchDictionaryObject)
|
|
|
|
i += 1
|
|
|
|
if len(DNAList) > 0:
|
|
print(f"One batch could not be filled completely and will contain {len(DNAList)} NFTs.")
|
|
|
|
incompleteBatch = {"NFTs_in_Batch": int(len(DNAList)), "hierarchy": hierarchy, "BatchDNAList": DNAList}
|
|
|
|
incompleteBatch = json.dumps(incompleteBatch, indent=1, ensure_ascii=True)
|
|
|
|
with open(os.path.join(batch_json_save_path, ("Batch{}.json".format(i + 1))), "w") as outfile2:
|
|
outfile2.write(incompleteBatch)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
makeBatches()
|