mirror of
https://github.com/imarkoff/Marble-shell-theme.git
synced 2025-09-18 17:27:55 -07:00
Fixed GDM installation, reworked label_files.py, fixed theme removing
This commit is contained in:
51
scripts/utils/files_labeler.py
Normal file
51
scripts/utils/files_labeler.py
Normal file
@@ -0,0 +1,51 @@
|
||||
import os
|
||||
|
||||
type LabeledFileGroup = tuple[str, str]
|
||||
|
||||
class FilesLabeler:
|
||||
def __init__(self, directory: str, *args: str):
|
||||
"""
|
||||
Initialize the working directory and files to change
|
||||
"""
|
||||
self.directory = directory
|
||||
self.files = args
|
||||
|
||||
def append_label(self, label: str):
|
||||
"""
|
||||
Append a label to all files in the directory
|
||||
and update references in the files
|
||||
"""
|
||||
labeled_files = self._label_files(label)
|
||||
self._update_references(labeled_files)
|
||||
|
||||
def _label_files(self, label: str) -> list[LabeledFileGroup]:
|
||||
labeled_files = []
|
||||
for filename in os.listdir(self.directory):
|
||||
if label in filename: continue
|
||||
|
||||
name, extension = os.path.splitext(filename)
|
||||
new_filename = f"{name}-{label}{extension}"
|
||||
|
||||
old_filepath = os.path.join(self.directory, filename)
|
||||
new_filepath = os.path.join(self.directory, new_filename)
|
||||
os.rename(old_filepath, new_filepath)
|
||||
|
||||
labeled_files.append((filename, new_filename))
|
||||
return labeled_files
|
||||
|
||||
def _update_references(self, labeled_files: list[LabeledFileGroup]):
|
||||
for file_path in self.files:
|
||||
with open(file_path, 'r') as file:
|
||||
file_content = file.read()
|
||||
|
||||
file_content = self._update_references_in_file(file_content, labeled_files)
|
||||
|
||||
with open(file_path, 'w') as file:
|
||||
file.write(file_content)
|
||||
|
||||
@staticmethod
|
||||
def _update_references_in_file(file_content: str, labeled_files: list[LabeledFileGroup]) -> str:
|
||||
replaced_content = file_content
|
||||
for old_name, new_name in labeled_files:
|
||||
replaced_content = replaced_content.replace(old_name, new_name)
|
||||
return replaced_content
|
Reference in New Issue
Block a user