diff --git a/appPublic/base64_to_file.py b/appPublic/base64_to_file.py new file mode 100644 index 0000000..c2d80e9 --- /dev/null +++ b/appPublic/base64_to_file.py @@ -0,0 +1,60 @@ +import os +import re +import base64 +from appPublic.uniqueID import getID + +MIME_EXT = { + # 图片 + "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", +} + +def getFilenameFromBase64(base64String): + match = re.match(r"data:(.*?);base64,(.*)", base64String) + if not match: + raise ValueError("不是合法的 base64 Data URL") + mime_type, b64_data = match.groups() + ext = MIME_EXT.get(mime_type, mime_type.split("/")[-1]) + name = getID() + fname = f'{name}.{ext}' + return fname + +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 + with open(output_path, 'wb') as file: + file.write(binary_data) diff --git a/appPublic/folderUtils.py b/appPublic/folderUtils.py index 38197f5..806bd68 100755 --- a/appPublic/folderUtils.py +++ b/appPublic/folderUtils.py @@ -48,22 +48,22 @@ def timestamp2datatiemStr(ts): """ def findAllDrives(): - Drives=[] - # print "Searching for drives..." - drives=win32api.GetLogicalDriveStrings().split(":") - for i in drives: - # print "i=",i,":" - dr=i[-1].lower() - if dr.isalpha(): - dr+=":\\" - inf=None - try: - inf=win32api.GetVolumeInformation(dr) - except: - pass # Removable drive, not ready - # You'll still get the drive letter, but inf will be None - Drives.append([dr,inf]) - return Drives + Drives=[] + # print "Searching for drives..." + drives=win32api.GetLogicalDriveStrings().split(":") + for i in drives: + # print "i=",i,":" + dr=i[-1].lower() + if dr.isalpha(): + dr+=":\\" + inf=None + try: + inf=win32api.GetVolumeInformation(dr) + except: + pass # Removable drive, not ready + # You'll still get the drive letter, but inf will be None + Drives.append([dr,inf]) + return Drives """ ## list all folder name under folder named by path @@ -77,19 +77,19 @@ def listFolder(path, rescursive=False) : yield full_name def listFile(folder,suffixs=[],rescursive=False): - subffixs = [ i.lower() for i in suffixs ] - for f in os.listdir(folder): - p = os.path.join(folder,f) - if rescursive and os.path.isdir(p): - for p1 in listFile(p,suffixs=suffixs,rescursive=True): - yield p1 - if os.path.isfile(p): - e = p.lower() - if suffixs == [] : - yield p - for s in subffixs: - if e.endswith(s): - yield p + subffixs = [ i.lower() for i in suffixs ] + for f in os.listdir(folder): + p = os.path.join(folder,f) + if rescursive and os.path.isdir(p): + for p1 in listFile(p,suffixs=suffixs,rescursive=True): + yield p1 + if os.path.isfile(p): + e = p.lower() + if suffixs == [] : + yield p + for s in subffixs: + if e.endswith(s): + yield p def folderInfo(root,uri=''): relpath = uri @@ -137,23 +137,12 @@ def rmdir_recursive(dir): os.rmdir(dir) def _mkdir(newdir) : - """works the way a good mkdir should :) - - already exists, silently complete - - regular file in the way, raise an exception - - parent directory(ies) does not exist, make them as well - """ - if os.path.isdir(newdir): - pass - elif os.path.isfile(newdir): - raise OSError("a file with the same name as the desired " \ - "dir, '%s', already exists." % newdir) - else: - head, tail = os.path.split(newdir) - if head and not os.path.isdir(head): - _mkdir(head) - #print "_mkdir %s" % repr(newdir) - if tail: - os.mkdir(newdir) + """works the way a good mkdir should :) + - already exists, silently complete + - regular file in the way, raise an exception + - parent directory(ies) does not exist, make them as well + """ + return os.makedirs(newdir, exist_ok=True) def _copyfile(fp,dir) : fs = open(fp,'rb')