Hykilpikonna
38 lines
940 B
Python
38 lines
940 B
Python
import base64
|
|
|
|
FILENAME_BLACKLIST = [
|
|
# Unix and Windows
|
|
"/",
|
|
|
|
# Windows only
|
|
"<", ">", ":", '"', "\\", "|", "?", "*", "\0",
|
|
"CON", "PRN", "AUX", "NUL",
|
|
"COM1", "COM2", "COM3", "COM4", "COM5", "COM6", "COM7", "COM8", "COM9",
|
|
"LPT1", "LPT2", "LPT3", "LPT4", "LPT5", "LPT6", "LPT7", "LPT8", "LPT9",
|
|
|
|
# Just for extra safety
|
|
"~"
|
|
]
|
|
|
|
FILENAME_REPLACE = {c: f"%{base64.b64encode(c.encode()).decode().replace('=', '')}" for c in FILENAME_BLACKLIST}
|
|
|
|
|
|
def escape_filename(fn: str) -> str:
|
|
fn = fn.replace("%", "[ PeRcEnT EsCaPe owo ]")
|
|
|
|
for c, r in FILENAME_REPLACE.items():
|
|
fn = fn.replace(c, r)
|
|
|
|
fn = fn.replace("[ PeRcEnT EsCaPe owo ]", "%%")
|
|
return fn
|
|
|
|
|
|
def unescape_filename(fn: str) -> str:
|
|
fn = fn.replace("%%", "[ PeRcEnT EsCaPe owo ]")
|
|
|
|
for c, r in FILENAME_REPLACE.items():
|
|
fn = fn.replace(r, c)
|
|
|
|
fn = fn.replace("[ PeRcEnT EsCaPe owo ]", "%")
|
|
return fn
|