Spaces:
Runtime error
Runtime error
File size: 4,625 Bytes
efac804 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 |
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from fastapi import FastAPI, APIRouter
import base64, json, zipfile, uvicorn
from strgen import StringGenerator
from pydantic import BaseModel
from pathlib import Path
from io import BytesIO
from PIL import Image
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
router = APIRouter()
processing = False
class TextRequest(BaseModel): text: str
@router.post('/save')
async def save(req_data: TextRequest):
data = json.loads(req_data.text)
zip_buffer = BytesIO()
with zipfile.ZipFile(zip_buffer, 'w') as zip_file:
zip_file.writestr('data.xml', data[0].encode('utf-8'))
zip_file.writestr('data.txt', data[1].encode('utf-8'))
for i in range(2, len(data)):
file_data = base64.b64decode(data[i].split(',', 1)[1])
file_name = f'{i - 1}.{data[i].split(";")[0].split("/")[1]}'
zip_file.writestr(file_name, file_data)
zip_buffer.seek(0)
return base64.b64encode(zip_buffer.getvalue()).decode('utf-8')
@router.post('/load')
async def load(zip_data: TextRequest):
zip_buffer = BytesIO(base64.b64decode(zip_data.text))
data = []
with zipfile.ZipFile(zip_buffer, 'r') as zip_file:
data.append(zip_file.open('data.xml').read().decode('utf-8'))
data.append(zip_file.open('data.txt').read().decode('utf-8'))
for file_info in zip_file.infolist():
if file_info.filename.startswith('data.'): continue
with zip_file.open(file_info) as f:
ext = file_info.filename.split('.')[-1]
url = base64.b64encode(f.read()).decode('utf-8')
data.append(f'data:image/{ext};base64,{url}')
return data
@router.get('/check')
async def check():
global processing
return 'ng' if processing else 'ok'
@router.post('/sb3')
async def sb3(req_data: TextRequest):
global processing
if processing: return ''
processing = True
data = json.loads(req_data.text)
with zipfile.ZipFile(Path('public').joinpath('mmp4.zip').resolve(), 'r') as template_zip:
with template_zip.open('project.json') as f:
project = json.loads(f.read().decode('utf-8'))
# 新しいZIPファイルを作成
zip_buffer = BytesIO()
with zipfile.ZipFile(zip_buffer, 'w') as zip_file:
variables = project['targets'][0]['variables']
tiles = project['targets'].index([v for v in project['targets'] if v['name'] == 'Tiles'][0])
project['targets'][0]['variables'][[v for v in variables if variables[v][0] == 'MODコード'][0]][1] = data[0]
names = StringGenerator('[a-f\\d]{32}').render_list(len(data) - 1, unique=True)
for i in range(1, len(data)):
name = f'{names[i - 1]}.png'
image_data = base64.b64decode(data[i].split(',', 1)[1])
# 画像をリサイズ
img = Image.open(BytesIO(image_data))
img_resized = img.resize((80, 80))
img_buffer = BytesIO()
img_resized.save(img_buffer, format='PNG')
img_buffer.seek(0)
# リサイズした画像をZIPに追加
zip_file.writestr(name, img_buffer.getvalue())
project['targets'][tiles]['costumes'].append({
"name": str(i),
"bitmapResolution": 2,
"dataFormat": "png",
"assetId": names[i - 1],
"md5ext": f"{names[i - 1]}.png",
"rotationCenterX": 40,
"rotationCenterY": 40
})
# 更新されたproject.jsonを書き込む
zip_file.writestr('project.json', json.dumps(project).encode('utf-8'))
# mmp4.zipの他のファイルもコピー
with zipfile.ZipFile(Path('public').joinpath('mmp4.zip').resolve(), 'r') as template_zip:
for item in template_zip.infolist():
if item.filename != 'project.json':
zip_file.writestr(item.filename, template_zip.read(item.filename))
# ZIPファイルのバイナリデータをBase64エンコード
zip_buffer.seek(0)
sb3_base64 = base64.b64encode(zip_buffer.getvalue()).decode('utf-8')
processing = False
return sb3_base64
app.include_router(router, prefix='/api')
if __name__ == '__main__':
app.mount('/', StaticFiles(directory=Path('public'), html=True), name='public')
uvicorn.run(app, host='0.0.0.0', port=7860) |