66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
import os
|
|
import base64
|
|
from appPuyblic.uniqueID import getID
|
|
|
|
function getFileExtensionFromBase64(base64String) {
|
|
// Extract MIME type from data URL
|
|
const mimeType = base64String.match(/data:([a-zA-Z0-9]+\/[a-zA-Z0-9-.+]+).*,.*/);
|
|
|
|
if (mimeType && mimeType[1]) {
|
|
const mime = mimeType[1];
|
|
const mimeToExtension = {
|
|
# 图片
|
|
"image/jpeg": "jpg",
|
|
"image/png": "png",
|
|
"image/gif": "gif",
|
|
"image/webp": "webp",
|
|
"image/bmp": "bmp",
|
|
"image/svg+xml": "svg",
|
|
"image/x-icon": "ico",
|
|
"image/tiff": "tiff",
|
|
|
|
# 音频
|
|
"audio/mpeg": "mp3",
|
|
"audio/wav": "wav",
|
|
"audio/ogg": "ogg",
|
|
"audio/webm": "weba",
|
|
"audio/aac": "aac",
|
|
"audio/flac": "flac",
|
|
"audio/mp4": "m4a",
|
|
"audio/3gpp": "3gp",
|
|
|
|
# 视频
|
|
"video/mp4": "mp4",
|
|
"video/webm": "webm",
|
|
"video/ogg": "ogv",
|
|
"video/x-msvideo": "avi",
|
|
"video/quicktime": "mov",
|
|
"video/x-matroska": "mkv",
|
|
"video/3gpp": "3gp",
|
|
"video/x-flv": "flv",
|
|
};
|
|
|
|
return mimeToExtension.get(mime, '');
|
|
}
|
|
|
|
return '';
|
|
}
|
|
def base64_to_file(base64_string, output_path):
|
|
# Remove data URL prefix if present (e.g., "data:image/png;base64,")
|
|
if ',' in base64_string:
|
|
header, base64_data = base64_string.split(',', 1)
|
|
else:
|
|
base64_data = base64_string
|
|
|
|
# Decode Base64 string
|
|
binary_data = base64.b64decode(base64_data)
|
|
|
|
# Write binary data to file
|
|
name = getID()
|
|
ext = getFileExtensionFromBase64(base64_string)
|
|
fname = f'{name}{ext}'
|
|
fp = os.path.join(output_path, fname)
|
|
with open(fp, 'wb') as file:
|
|
file.write(binary_data)
|
|
|