init

common/utils.py

@@ -0,0 +1,131 @@
+# ----------------------------------------
+# - mode: python - 
+# - author: helloplhm-qwq - 
+# - name: utils.py - 
+# - project: lx-music-api-server - 
+# - license: MIT - 
+# ----------------------------------------
+# This file is part of the "lx-music-api-server" project.
+# Do not edit except you know what you are doing.
+
+import platform
+import binascii
+import base64
+import zlib
+import re
+import ujson as json
+from urllib.parse import quote
+from hashlib import md5 as _md5
+from flask import Response
+
+def to_base64(data_bytes):
+    encoded_data = base64.b64encode(data_bytes)
+    return encoded_data.decode('utf-8')
+
+def to_hex(data_bytes):
+    hex_encoded = binascii.hexlify(data_bytes)
+    return hex_encoded.decode('utf-8')
+
+def from_base64(data):
+    decoded_data = base64.b64decode(data)
+    return decoded_data
+
+def from_hex(data):
+    decoded_data = binascii.unhexlify(data.decode('utf-8'))
+    return decoded_data
+
+def inflate_raw_sync(data):
+    decompress_obj = zlib.decompressobj(-zlib.MAX_WBITS)
+    decompressed_data = decompress_obj.decompress(data) + decompress_obj.flush()
+    return decompressed_data
+
+def require(module):
+    index = 0
+    module_array = module.split('.')
+    for m in module_array:
+        if index == 0:
+            _module = __import__(m)
+            index += 1
+        else:
+            _module = getattr(_module, m)
+            index += 1
+    return _module
+
+def sanitize_filename(filename):
+    if platform.system() == 'Windows' or platform.system() == 'Cygwin':
+        # Windows不合法文件名字符
+        illegal_chars = r'[<>:"/\\|?*\x00-\x1f]'
+    else:
+        # 不合法文件名字符
+        illegal_chars = r'[/\x00-\x1f]'
+    # 将不合法字符替换为下划线
+    return re.sub(illegal_chars, '_', filename)
+
+def md5(s: str):
+    # 计算md5
+    # print(s)
+    return _md5(s.encode("utf-8")).hexdigest()
+
+def readfile(path, mode = "text"):
+    try:
+        fileObj = open(path, "rb")
+    except FileNotFoundError:
+        return "file not found"
+    content = fileObj.read()
+    if mode == "base64":
+        return to_base64(content)
+    elif mode == "hex":
+        return to_hex(content)
+    elif mode == "text":
+        return content.decode("utf-8")
+    else:
+        return "unsupported mode"
+
+def unique_list(list_in):
+    unique_list = []
+    [unique_list.append(x) for x in list_in if x not in unique_list]
+    return unique_list
+
+def format_dict_json(dic):
+    return Response(json.dumps(dic, indent=2, ensure_ascii=False), mimetype = "application/json")
+
+def encodeURIComponent(component):
+    return quote(component)
+
+def sort_dict(dictionary):
+    sorted_items = sorted(dictionary.items())
+    sorted_dict = {k: v for k, v in sorted_items}
+    return sorted_dict
+
+def merge_dict(dict1, dict2):
+    merged_dict = dict2.copy()
+    merged_dict.update(dict1)
+    return merged_dict
+
+class jsobject(dict):
+    def __init__(self, d):
+        super().__init__(d)
+        self._raw = d
+        for key, value in d.items():
+            if isinstance(value, dict):
+                setattr(self, key, jsobject(value))
+            else:
+                setattr(self, key, value)
+
+    def __setattr__(self, key, value):
+        super().__setattr__(key, value)
+        if key != "_raw":
+            self._raw[key] = value
+
+    def to_dict(self):
+        result = {}
+        for key, value in self.items():
+            if isinstance(value, jsobject):
+                result[key] = value.to_dict()
+            else:
+                result[key] = value
+        return result
+
+    def __getattr__(self, UNUSED):
+        return None
+