mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-01-08 20:13:14 -08:00
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
|
|
def sanitize_name(name):
|
|
"""Remove bad chars from the filename.
|
|
|
|
>>> sanitize_name('a/b/c')
|
|
'a-b-c'
|
|
>>> sanitize_name('abc')
|
|
'abc'
|
|
>>> sanitize_name('a"b')
|
|
'ab'
|
|
>>> sanitize_name('.a.b..')
|
|
'a.b'
|
|
"""
|
|
name = re.sub(r'[\\/*]', '-', name)
|
|
name = re.sub(r'[:\'<>|?]', '', name)
|
|
# remove leading/trailing periods and spaces
|
|
name = name.strip(' .')
|
|
return name
|
|
|
|
|
|
def clean_file_name(filename):
|
|
"""Clean up nzb name by removing any . and _ characters and trailing hyphens.
|
|
|
|
Is basically equivalent to replacing all _ and . with a
|
|
space, but handles decimal numbers in string, for example:
|
|
"""
|
|
filename = re.sub(r'(\D)\.(?!\s)(\D)', r'\1 \2', filename)
|
|
# if it ends in a year then don't keep the dot
|
|
filename = re.sub(r'(\d)\.(\d{4})', r'\1 \2', filename)
|
|
filename = re.sub(r'(\D)\.(?!\s)', r'\1 ', filename)
|
|
filename = re.sub(r'\.(?!\s)(\D)', r' \1', filename)
|
|
filename = filename.replace('_', ' ')
|
|
filename = re.sub('-$', '', filename)
|
|
filename = re.sub(r'^\[.*]', '', filename)
|
|
return filename.strip()
|
|
|
|
|
|
def is_sample(input_name) -> bool:
|
|
# Ignore 'sample' in files
|
|
if re.search('(^|[\\W_])sample\\d*[\\W_]', input_name.lower()):
|
|
return True
|
|
return False
|